File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -55,6 +55,7 @@ set(TRITON_PYTORCH_DOCKER_IMAGE "" CACHE STRING "Docker image containing the PyT
5555set (TRITON_PYTORCH_INCLUDE_PATHS "" CACHE PATH "Paths to Torch includes" )
5656set (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" )
5859set (TRITON_BACKEND_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/backend repo" )
5960set (TRITON_CORE_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/core repo" )
6061set (TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo" )
@@ -95,19 +96,19 @@ include(FetchContent)
9596
9697FetchContent_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)
102103FetchContent_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)
108109FetchContent_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)
You can’t perform that action at this time.
0 commit comments