Skip to content

Commit fa3361e

Browse files
authored
TPRD-1536: Update 'build.py' to meet TensorRT-LLM repository structur… (#8214)
1 parent b8caffb commit fa3361e

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

build.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -622,7 +622,7 @@ def backend_cmake_args(images, components, be, install_dir, library_paths):
622622

623623
cargs += cmake_backend_extra_args(be)
624624
if be == "tensorrtllm":
625-
cargs.append("-S ../inflight_batcher_llm -B .")
625+
cargs.append("-S ../triton_backend/inflight_batcher_llm -B .")
626626

627627
else:
628628
cargs.append("..")
@@ -2081,7 +2081,16 @@ def backend_build(
20812081
cmake_script.comment()
20822082
cmake_script.mkdir(build_dir)
20832083
cmake_script.cwd(build_dir)
2084-
cmake_script.gitclone(backend_repo(be), tag, be, github_organization)
2084+
if be == "tensorrtllm":
2085+
github_organization = (
2086+
"https://github.com/NVIDIA"
2087+
if "triton-inference-server" in FLAGS.github_organization
2088+
else FLAGS.github_organization
2089+
)
2090+
repository_name = "TensorRT-LLM"
2091+
cmake_script.gitclone(repository_name, tag, be, github_organization)
2092+
else:
2093+
cmake_script.gitclone(backend_repo(be), tag, be, github_organization)
20852094

20862095
if be == "tensorrtllm":
20872096
tensorrtllm_prebuild(cmake_script)

0 commit comments

Comments
 (0)