• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1set(COMMON_GIT_REPOSITORY "https://gitee.com/mirrors_triton-inference-server/common.git")
2set(COMMON_GIT_TAG "1df32b982a6ed11ead3271a55b04bf6e7abc1cf9")
3set(COMMON_SHA256 "1df32b982a6ed11ead3271a55b04bf6e7abc1cf9")  #unused
4mindspore_add_pkg(triton_common
5        LIBS tritonasyncworkqueue
6        GIT_REPOSITORY ${COMMON_GIT_REPOSITORY}
7        GIT_TAG ${COMMON_GIT_TAG}
8        SHA256 ${COMMON_SHA256}
9        PATCHES ${TOP_DIR}/third_party/patch/triton/triton_common.patch001
10        CMAKE_OPTION
11        -DCMAKE_BUILD_TYPE:STRING=Release
12        -DTRITON_COMMON_ENABLE_JSON:BOOL=off
13        )
14
15set(CORE_GIT_REPOSITORY "https://gitee.com/mirrors_triton-inference-server/core.git")
16set(CORE_GIT_TAG "9714cd666d512c4a2c687d028cd4ebf3d354a9dc")
17set(CORE_SHA256 "9714cd666d512c4a2c687d028cd4ebf3d354a9dc")  #unused
18mindspore_add_pkg(triton_core
19        LIBS tritonserver
20        GIT_REPOSITORY ${CORE_GIT_REPOSITORY}
21        GIT_TAG ${CORE_GIT_TAG}
22        SHA256 ${CORE_SHA256}
23        LIB_SUFFIXES_PATH stubs
24        PATCHES ${TOP_DIR}/third_party/patch/triton/triton_core.patch001
25        CMAKE_OPTION
26        -DCMAKE_BUILD_TYPE:STRING=Release
27        )
28
29set(BACKEND_GIT_REPOSITORY "https://gitee.com/mirrors_triton-inference-server/backend.git")
30set(BACKEND_GIT_TAG "2f3f44eba661af9b4fcd62160eafd9012db32652")
31set(BACKEND_SHA256 "2f3f44eba661af9b4fcd62160eafd9012db32652")  #unused
32mindspore_add_pkg(triton_backend
33        LIBS tritonbackendutils
34        GIT_REPOSITORY ${BACKEND_GIT_REPOSITORY}
35        GIT_TAG ${BACKEND_GIT_TAG}
36        SHA256 ${BACKEND_SHA256}
37        PATCHES ${TOP_DIR}/third_party/patch/triton/triton_backend.patch001
38        CMAKE_OPTION
39        -DCMAKE_BUILD_TYPE:STRING=Release
40        -DTRITON_ENABLE_GPU:BOOL=${TRITON_ENABLE_GPU}
41        -DTRITON_RAPID_JSON_PATH:STRING=${TRITON_RAPID_JSON_PATH}
42        -DTRITON_CORE_PATH=${triton_core_INC}
43        -DTRITON_COMMON_PATH=${triton_common_INC}
44        )
45
46include_directories(${triton_backend_INC})
47include_directories(${triton_common_INC})
48add_library(mindspore::tritonasyncworkqueue ALIAS triton_common::tritonasyncworkqueue)
49add_library(mindspore::tritonserver ALIAS triton_core::tritonserver)
50add_library(mindspore::tritonbackendutils ALIAS triton_backend::tritonbackendutils)
51