Skip to content

Commit fed4bc1

Browse files
nv-kmcgill53kyle
andauthored
patching git repository parameterization from production branch 1 (#332)
Co-authored-by: kyle <kmcgill@kmcgill-ubuntu.nvidia.com>
1 parent 5d4a99c commit fed4bc1

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

CMakeLists.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,6 +182,7 @@ if(NOT TRITON_CORE_HEADERS_ONLY)
182182
option(TRITON_ENABLE_AZURE_STORAGE "Include Azure Storage Filesystem support in server" OFF)
183183

184184
# Repo tags
185+
set(TRITON_REPO_ORGANIZATION "https://github.com/triton-inference-server" CACHE STRING "Git repository to pull from")
185186
set(TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo")
186187
set(TRITON_THIRD_PARTY_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/third_party repo")
187188

@@ -212,7 +213,7 @@ if(NOT TRITON_CORE_HEADERS_ONLY)
212213
include(FetchContent)
213214
FetchContent_Declare(
214215
repo-third-party
215-
GIT_REPOSITORY https://github.com/triton-inference-server/third_party.git
216+
GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION}/third_party.git
216217
GIT_TAG ${TRITON_THIRD_PARTY_REPO_TAG}
217218
)
218219
FetchContent_MakeAvailable(repo-third-party)
@@ -325,6 +326,7 @@ if(NOT TRITON_CORE_HEADERS_ONLY)
325326
-Daws-checksums_DIR:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/aws-sdk-cpp/${LIB_DIR}/aws-checksums/cmake
326327
-DCMAKE_PREFIX_PATH:PATH=${CMAKE_PREFIX_PATH}
327328
-DCNMEM_PATH:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/cnmem
329+
-DTRITON_REPO_ORGANIZATION:STRING=${TRITON_REPO_ORGANIZATION}
328330
-DTRITON_COMMON_REPO_TAG:STRING=${TRITON_COMMON_REPO_TAG}
329331
-DTRITON_EXTRA_LIB_PATHS:PATH=${TRITON_EXTRA_LIB_PATHS}
330332
-DTRITON_ENABLE_NVTX:BOOL=${TRITON_ENABLE_NVTX}

src/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ include(FetchContent)
4242

4343
FetchContent_Declare(
4444
repo-common
45-
GIT_REPOSITORY https://github.com/triton-inference-server/common.git
45+
GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION}/common.git
4646
GIT_TAG ${TRITON_COMMON_REPO_TAG}
4747
)
4848

0 commit comments

Comments
 (0)