Skip to content

Commit b30a795

Browse files
committed
Update submodule url
1 parent e89ee44 commit b30a795

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

build.py

+11-6
Original file line numberDiff line numberDiff line change
@@ -1785,11 +1785,6 @@ def tensorrtllm_prebuild(cmake_script):
17851785
cmake_script.cmd("mv tensorrtllm/inflight_batcher_llm/cmake tensorrtllm")
17861786
cmake_script.cmd("mv tensorrtllm/inflight_batcher_llm/CMakeLists.txt tensorrtllm")
17871787

1788-
cmake_script.cmd(
1789-
"cd tensorrtllm_backend && git submodule set-url -- tensorrt_llm https://github.com/NVIDIA/TensorRT-LLM.git"
1790-
)
1791-
cmake_script.cmd("cd tensorrtllm_backend && git submodule sync")
1792-
17931788

17941789
def backend_build(
17951790
be,
@@ -1811,10 +1806,20 @@ def backend_build(
18111806
cmake_script.comment()
18121807
cmake_script.mkdir(build_dir)
18131808
cmake_script.cwd(build_dir)
1814-
cmake_script.gitclone(backend_repo(be), tag, be, github_organization)
18151809

18161810
if be == "tensorrtllm":
1811+
cmake_script.cmd(
1812+
"git clone --single-branch --depth=1 -b {} https://github.com/triton-inference-server/tensorrtllm_backend tensorrtllm".format(
1813+
tag
1814+
)
1815+
)
1816+
cmake_script.cmd(
1817+
"cd tensorrtllm_backend && git submodule set-url -- tensorrt_llm https://github.com/NVIDIA/TensorRT-LLM.git"
1818+
)
1819+
cmake_script.cmd("cd tensorrtllm_backend && git submodule sync")
18171820
tensorrtllm_prebuild(cmake_script)
1821+
else:
1822+
cmake_script.gitclone(backend_repo(be), tag, be, github_organization)
18181823

18191824
cmake_script.mkdir(repo_build_dir)
18201825
cmake_script.cwd(repo_build_dir)

0 commit comments

Comments
 (0)