Skip to content

Commit 4fa7daa

Browse files
nv-kmcgill53kyle
and
kyle
authored
patching git repository parameterization from production branch 1 (#124)
Co-authored-by: kyle <[email protected]>
1 parent 0931f9d commit 4fa7daa

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

CMakeLists.txt

+4-3
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ set(TRITON_PYTORCH_DOCKER_IMAGE "" CACHE STRING "Docker image containing the PyT
5555
set(TRITON_PYTORCH_INCLUDE_PATHS "" CACHE PATH "Paths to Torch includes")
5656
set(TRITON_PYTORCH_LIB_PATHS "" CACHE PATH "Paths to Torch libraries")
5757

58+
set(TRITON_REPO_ORGANIZATION "https://github.com/triton-inference-server" CACHE STRING "Git repository to pull from")
5859
set(TRITON_BACKEND_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/backend repo")
5960
set(TRITON_CORE_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/core repo")
6061
set(TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo")
@@ -95,19 +96,19 @@ include(FetchContent)
9596

9697
FetchContent_Declare(
9798
repo-common
98-
GIT_REPOSITORY https://github.com/triton-inference-server/common.git
99+
GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION}/common.git
99100
GIT_TAG ${TRITON_COMMON_REPO_TAG}
100101
GIT_SHALLOW ON
101102
)
102103
FetchContent_Declare(
103104
repo-core
104-
GIT_REPOSITORY https://github.com/triton-inference-server/core.git
105+
GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION}/core.git
105106
GIT_TAG ${TRITON_CORE_REPO_TAG}
106107
GIT_SHALLOW ON
107108
)
108109
FetchContent_Declare(
109110
repo-backend
110-
GIT_REPOSITORY https://github.com/triton-inference-server/backend.git
111+
GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION}/backend.git
111112
GIT_TAG ${TRITON_BACKEND_REPO_TAG}
112113
GIT_SHALLOW ON
113114
)

0 commit comments

Comments
 (0)