File tree 1 file changed +4
-3
lines changed
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
55
55
set (TRITON_PYTORCH_INCLUDE_PATHS "" CACHE PATH "Paths to Torch includes" )
56
56
set (TRITON_PYTORCH_LIB_PATHS "" CACHE PATH "Paths to Torch libraries" )
57
57
58
+ set (TRITON_REPO_ORGANIZATION "https://github.com/triton-inference-server" CACHE STRING "Git repository to pull from" )
58
59
set (TRITON_BACKEND_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/backend repo" )
59
60
set (TRITON_CORE_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/core repo" )
60
61
set (TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo" )
@@ -95,19 +96,19 @@ include(FetchContent)
95
96
96
97
FetchContent_Declare(
97
98
repo-common
98
- GIT_REPOSITORY https://github.com/triton-inference-server /common.git
99
+ GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION} /common.git
99
100
GIT_TAG ${TRITON_COMMON_REPO_TAG}
100
101
GIT_SHALLOW ON
101
102
)
102
103
FetchContent_Declare(
103
104
repo-core
104
- GIT_REPOSITORY https://github.com/triton-inference-server /core.git
105
+ GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION} /core.git
105
106
GIT_TAG ${TRITON_CORE_REPO_TAG}
106
107
GIT_SHALLOW ON
107
108
)
108
109
FetchContent_Declare(
109
110
repo-backend
110
- GIT_REPOSITORY https://github.com/triton-inference-server /backend.git
111
+ GIT_REPOSITORY ${TRITON_REPO_ORGANIZATION} /backend.git
111
112
GIT_TAG ${TRITON_BACKEND_REPO_TAG}
112
113
GIT_SHALLOW ON
113
114
)
You can’t perform that action at this time.
0 commit comments