Skip to content

Commit 9a11187

Browse files
authored
Freeze the triton version in vllm-gaudi image to 3.1.0 (opea-project#1463)
The new triton version 3.2.0 can't work with vllm-gaudi. Freeze the triton version in vllm-gaudi image to 3.1.0. Issue create for vllm-fork: HabanaAI/vllm-fork#732 Signed-off-by: chensuyue <suyue.chen@intel.com>
1 parent ffce706 commit 9a11187

File tree

6 files changed

+6
-2
lines changed

6 files changed

+6
-2
lines changed

.github/workflows/_example-workflow.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,8 @@ jobs:
7878
cd vllm && git rev-parse HEAD && cd ../
7979
fi
8080
if [[ $(grep -c "vllm-gaudi:" ${docker_compose_path}) != 0 ]]; then
81-
git clone --depth 1 --branch v0.6.4.post2+Gaudi-1.19.0 https://github.com/HabanaAI/vllm-fork.git
81+
git clone --depth 1 --branch v0.6.4.post2+Gaudi-1.19.0 https://github.com/HabanaAI/vllm-fork.git
82+
sed -i 's/triton/triton==3.1.0/g' vllm-fork/requirements-hpu.txt
8283
fi
8384
git clone --depth 1 --branch ${{ inputs.opea_branch }} https://github.com/opea-project/GenAIComps.git
8485
cd GenAIComps && git rev-parse HEAD && cd ../

AgentQnA/tests/step1_build_images.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ function build_vllm_docker_image() {
4343
fi
4444
cd ./vllm-fork
4545
git checkout v0.6.4.post2+Gaudi-1.19.0
46+
sed -i 's/triton/triton==3.1.0/g' requirements-hpu.txt
4647
docker build --no-cache -f Dockerfile.hpu -t opea/vllm-gaudi:ci --shm-size=128g . --build-arg https_proxy=$https_proxy --build-arg http_proxy=$http_proxy
4748
if [ $? -ne 0 ]; then
4849
echo "opea/vllm-gaudi:ci failed"

AgentQnA/tests/test_compose_on_gaudi.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
#!/bin/bash
22
# Copyright (C) 2024 Intel Corporation
33
# SPDX-License-Identifier: Apache-2.0
4-
54
set -xe
65

76
WORKPATH=$(dirname "$PWD")

ChatQnA/tests/test_compose_guardrails_on_gaudi.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ function build_docker_images() {
2929
cd $WORKPATH/docker_image_build
3030
git clone --depth 1 --branch ${opea_branch} https://github.com/opea-project/GenAIComps.git
3131
git clone --depth 1 --branch v0.6.4.post2+Gaudi-1.19.0 https://github.com/HabanaAI/vllm-fork.git
32+
sed -i 's/triton/triton==3.1.0/g' vllm-fork/requirements-hpu.txt
3233

3334
echo "Build all the images with --no-cache, check docker_image_build.log for details..."
3435
service_list="chatqna-guardrails chatqna-ui dataprep retriever vllm-gaudi guardrails nginx"

ChatQnA/tests/test_compose_on_gaudi.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ function build_docker_images() {
2929
cd $WORKPATH/docker_image_build
3030
git clone --depth 1 --branch ${opea_branch} https://github.com/opea-project/GenAIComps.git
3131
git clone --depth 1 --branch v0.6.4.post2+Gaudi-1.19.0 https://github.com/HabanaAI/vllm-fork.git
32+
sed -i 's/triton/triton==3.1.0/g' vllm-fork/requirements-hpu.txt
3233

3334
echo "Build all the images with --no-cache, check docker_image_build.log for details..."
3435
service_list="chatqna chatqna-ui dataprep retriever vllm-gaudi nginx"

ChatQnA/tests/test_compose_without_rerank_on_gaudi.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ function build_docker_images() {
2929
cd $WORKPATH/docker_image_build
3030
git clone --depth 1 --branch ${opea_branch} https://github.com/opea-project/GenAIComps.git
3131
git clone --depth 1 --branch v0.6.4.post2+Gaudi-1.19.0 https://github.com/HabanaAI/vllm-fork.git
32+
sed -i 's/triton/triton==3.1.0/g' vllm-fork/requirements-hpu.txt
3233

3334
echo "Build all the images with --no-cache, check docker_image_build.log for details..."
3435
service_list="chatqna-without-rerank chatqna-ui dataprep retriever vllm-gaudi nginx"

0 commit comments

Comments
 (0)