File tree 1 file changed +11
-6
lines changed
1 file changed +11
-6
lines changed Original file line number Diff line number Diff line change @@ -1785,11 +1785,6 @@ def tensorrtllm_prebuild(cmake_script):
1785
1785
cmake_script .cmd ("mv tensorrtllm/inflight_batcher_llm/cmake tensorrtllm" )
1786
1786
cmake_script .cmd ("mv tensorrtllm/inflight_batcher_llm/CMakeLists.txt tensorrtllm" )
1787
1787
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
-
1793
1788
1794
1789
def backend_build (
1795
1790
be ,
@@ -1811,10 +1806,20 @@ def backend_build(
1811
1806
cmake_script .comment ()
1812
1807
cmake_script .mkdir (build_dir )
1813
1808
cmake_script .cwd (build_dir )
1814
- cmake_script .gitclone (backend_repo (be ), tag , be , github_organization )
1815
1809
1816
1810
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" )
1817
1820
tensorrtllm_prebuild (cmake_script )
1821
+ else :
1822
+ cmake_script .gitclone (backend_repo (be ), tag , be , github_organization )
1818
1823
1819
1824
cmake_script .mkdir (repo_build_dir )
1820
1825
cmake_script .cwd (repo_build_dir )
You can’t perform that action at this time.
0 commit comments