Skip to content

Commit 310bdd9

Browse files
pcastonguaydc3671
authored andcommitted
fix: Fix triton backend build [nvbug 5396469] (#6098)
Signed-off-by: Patrice Castonguay <[email protected]>
1 parent a03c680 commit 310bdd9

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

tests/integration/defs/triton_server/test_triton.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -508,7 +508,7 @@ def test_cpp_unit_tests(tritonserver_test_root, test_name, llm_root):
508508

509509
run_shell_command(
510510
f"cd {llm_root}/triton_backend/inflight_batcher_llm/build && "
511-
f"cmake .. -DTRTLLM_DIR={llm_root} -DCMAKE_INSTALL_PREFIX=install/ -DBUILD_TESTS=ON -DUSE_CXX11_ABI=ON "
511+
f"cmake .. -DTRTLLM_DIR={llm_root} -DCMAKE_INSTALL_PREFIX=install/ -DBUILD_TESTS=ON -DUSE_CXX11_ABI=ON -DTRITON_COMMON_REPO_TAG=r25.05 -DTRITON_CORE_REPO_TAG=r25.05 -DTRITON_THIRD_PARTY_REPO_TAG=r25.05 -DTRITON_BACKEND_REPO_TAG=r25.05 "
512512
"&& make -j8 install", llm_root)
513513

514514
# Run the cpp unit tests

triton_backend/inflight_batcher_llm/scripts/build.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,8 @@ if [[ "$BUILD_UNIT_TESTS" == "true" ]]; then
5151
BUILD_TESTS_ARG="-DBUILD_TESTS=ON -DUSE_CXX11_ABI=ON"
5252
fi
5353

54-
cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install ${BUILD_TESTS_ARG} ..
54+
# TODO: Remove specifying Triton version after cmake version is upgraded to 3.31.8
55+
cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install ${BUILD_TESTS_ARG} -DTRITON_COMMON_REPO_TAG=r25.05 -DTRITON_CORE_REPO_TAG=r25.05 -DTRITON_THIRD_PARTY_REPO_TAG=r25.05 -DTRITON_BACKEND_REPO_TAG=r25.05 ..
5556
make install
5657

5758
mkdir -p /opt/tritonserver/backends/tensorrtllm

0 commit comments

Comments
 (0)