File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -182,6 +182,7 @@ if(NOT TRITON_CORE_HEADERS_ONLY)
182
182
option (TRITON_ENABLE_AZURE_STORAGE "Include Azure Storage Filesystem support in server" OFF )
183
183
184
184
# Repo tags
185
+ set (TRITON_REPO_ORGANIZATION "https://github.com/triton-inference-server" CACHE STRING "Git repository to pull from" )
185
186
set (TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo" )
186
187
set (TRITON_THIRD_PARTY_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/third_party repo" )
187
188
@@ -212,7 +213,7 @@ if(NOT TRITON_CORE_HEADERS_ONLY)
212
213
include (FetchContent )
213
214
FetchContent_Declare (
214
215
repo-third-party
215
- GIT_REPOSITORY https://github.com/triton-inference-server /third_party.git
216
+ GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION} /third_party.git
216
217
GIT_TAG ${TRITON_THIRD_PARTY_REPO_TAG}
217
218
)
218
219
FetchContent_MakeAvailable (repo-third-party )
@@ -325,6 +326,7 @@ if(NOT TRITON_CORE_HEADERS_ONLY)
325
326
-Daws-checksums_DIR:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/aws-sdk-cpp/${LIB_DIR}/aws-checksums/cmake
326
327
-DCMAKE_PREFIX_PATH:PATH=${CMAKE_PREFIX_PATH}
327
328
-DCNMEM_PATH:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/cnmem
329
+ -DTRITON_REPO_ORGANIZATION:STRING=${TRITON_REPO_ORGANIZATION}
328
330
-DTRITON_COMMON_REPO_TAG:STRING=${TRITON_COMMON_REPO_TAG}
329
331
-DTRITON_EXTRA_LIB_PATHS:PATH=${TRITON_EXTRA_LIB_PATHS}
330
332
-DTRITON_ENABLE_NVTX:BOOL=${TRITON_ENABLE_NVTX}
Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ include(FetchContent)
42
42
43
43
FetchContent_Declare (
44
44
repo-common
45
- GIT_REPOSITORY https://github.com/triton-inference-server /common.git
45
+ GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION} /common.git
46
46
GIT_TAG ${TRITON_COMMON_REPO_TAG}
47
47
)
48
48
You can’t perform that action at this time.
0 commit comments