Skip to content

Commit 00a95a2

Browse files
yiqingy0dominicshanshan
authored andcommitted
[https://nvbugs/5444624][fix] Fix LLM_ROOT in triton_backend build.sh (NVIDIA#6744)
Signed-off-by: Yiqing Yan <[email protected]> Signed-off-by: Wangshanshan <[email protected]>
1 parent b1cbde1 commit 00a95a2

File tree

1 file changed

+2
-1
lines changed
  • triton_backend/inflight_batcher_llm/scripts

1 file changed

+2
-1
lines changed

triton_backend/inflight_batcher_llm/scripts/build.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,8 @@ fi
5353

5454
# TODO: Remove specifying Triton version after cmake version is upgraded to 3.31.8
5555
# Get TRITON_SHORT_TAG from docker/Dockerfile.multi
56-
LLM_ROOT="$(dirname $0)/../../../.."
56+
LLM_ROOT=$BUILD_DIR/../../..
57+
LLM_ROOT=$(cd -- "$LLM_ROOT" && pwd)
5758
TRITON_SHORT_TAG=$("$LLM_ROOT/jenkins/scripts/get_triton_tag.sh" "$LLM_ROOT")
5859
cmake -DCMAKE_INSTALL_PREFIX:PATH=`pwd`/install ${BUILD_TESTS_ARG} -DTRITON_COMMON_REPO_TAG=${TRITON_SHORT_TAG} -DTRITON_CORE_REPO_TAG=${TRITON_SHORT_TAG} -DTRITON_THIRD_PARTY_REPO_TAG=${TRITON_SHORT_TAG} -DTRITON_BACKEND_REPO_TAG=${TRITON_SHORT_TAG} ..
5960
make install

0 commit comments

Comments
 (0)