Skip to content
This repository was archived by the owner on Jun 22, 2025. It is now read-only.

Commit 0ce67c1

Browse files
authored
Remove enforce-eager to enable HPU graphs for better vLLM perf (opea-project#954)
* remove enforce-eager to enable HPU graphs Signed-off-by: Wang, Kai Lawrence <kai.lawrence.wang@intel.com> * Increase the llm max timeout in ci for fully warmup Signed-off-by: Wang, Kai Lawrence <kai.lawrence.wang@intel.com> --------- Signed-off-by: Wang, Kai Lawrence <kai.lawrence.wang@intel.com>
1 parent e8aa78b commit 0ce67c1

8 files changed

Lines changed: 10 additions & 10 deletions

File tree

comps/llms/faq-generation/vllm/langchain/docker_compose_llm.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ services:
2323
cap_add:
2424
- SYS_NICE
2525
ipc: host
26-
command: --enforce-eager --model $LLM_MODEL_ID --tensor-parallel-size 1 --host 0.0.0.0 --port 80
26+
command: --model $LLM_MODEL_ID --tensor-parallel-size 1 --host 0.0.0.0 --port 80
2727
llm:
2828
image: opea/llm-faqgen-vllm:latest
2929
container_name: llm-faqgen-server

comps/llms/summarization/vllm/langchain/docker_compose_llm.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ services:
2323
cap_add:
2424
- SYS_NICE
2525
ipc: host
26-
command: --enforce-eager --model $LLM_MODEL_ID --tensor-parallel-size 1 --host 0.0.0.0 --port 80
26+
command: --model $LLM_MODEL_ID --tensor-parallel-size 1 --host 0.0.0.0 --port 80
2727
llm:
2828
image: opea/llm-docsum-vllm:latest
2929
container_name: llm-docsum-vllm-server

comps/llms/text-generation/vllm/langchain/dependency/launch_vllm_service.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ volume=$PWD/data
3838

3939
# Build the Docker run command based on hardware mode
4040
if [ "$hw_mode" = "hpu" ]; then
41-
docker run -d --rm --runtime=habana --name="vllm-service" -p $port_number:80 -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --ipc=host -e HTTPS_PROXY=$https_proxy -e HTTP_PROXY=$https_proxy -e HF_TOKEN=${HF_TOKEN} opea/vllm-gaudi:latest --enforce-eager --model $model_name --tensor-parallel-size $parallel_number --host 0.0.0.0 --port 80 --block-size $block_size --max-num-seqs $max_num_seqs --max-seq_len-to-capture $max_seq_len_to_capture
41+
docker run -d --rm --runtime=habana --name="vllm-service" -p $port_number:80 -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --ipc=host -e HTTPS_PROXY=$https_proxy -e HTTP_PROXY=$https_proxy -e HF_TOKEN=${HF_TOKEN} opea/vllm-gaudi:latest --model $model_name --tensor-parallel-size $parallel_number --host 0.0.0.0 --port 80 --block-size $block_size --max-num-seqs $max_num_seqs --max-seq_len-to-capture $max_seq_len_to_capture
4242
else
4343
docker run -d --rm --name="vllm-service" -p $port_number:80 --network=host -v $volume:/data -e HTTPS_PROXY=$https_proxy -e HTTP_PROXY=$https_proxy -e HF_TOKEN=${HF_TOKEN} -e VLLM_CPU_KVCACHE_SPACE=40 opea/vllm-cpu:latest --model $model_name --host 0.0.0.0 --port 80
4444
fi

comps/llms/text-generation/vllm/langchain/docker_compose_llm.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ services:
2323
cap_add:
2424
- SYS_NICE
2525
ipc: host
26-
command: --enforce-eager --model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80
26+
command: --model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80
2727
llm:
2828
image: opea/llm-vllm:latest
2929
container_name: llm-vllm-gaudi-server

comps/llms/text-generation/vllm/llama_index/dependency/launch_vllm_service.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ volume=$PWD/data
3838

3939
# Build the Docker run command based on hardware mode
4040
if [ "$hw_mode" = "hpu" ]; then
41-
docker run -d --rm --runtime=habana --name="vllm-service" -p $port_number:80 -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --ipc=host -e HTTPS_PROXY=$https_proxy -e HTTP_PROXY=$https_proxy -e HF_TOKEN=${HUGGINGFACEHUB_API_TOKEN} opea/vllm-gaudi:latest --enforce-eager --model $model_name --tensor-parallel-size $parallel_number --host 0.0.0.0 --port 80 --block-size $block_size --max-num-seqs $max_num_seqs --max-seq_len-to-capture $max_seq_len_to_capture
41+
docker run -d --rm --runtime=habana --name="vllm-service" -p $port_number:80 -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --ipc=host -e HTTPS_PROXY=$https_proxy -e HTTP_PROXY=$https_proxy -e HF_TOKEN=${HUGGINGFACEHUB_API_TOKEN} opea/vllm-gaudi:latest --model $model_name --tensor-parallel-size $parallel_number --host 0.0.0.0 --port 80 --block-size $block_size --max-num-seqs $max_num_seqs --max-seq_len-to-capture $max_seq_len_to_capture
4242
else
4343
docker run -d --rm --name="vllm-service" -p $port_number:80 --network=host -v $volume:/data -e HTTPS_PROXY=$https_proxy -e HTTP_PROXY=$https_proxy -e HF_TOKEN=${HUGGINGFACEHUB_API_TOKEN} -e VLLM_CPU_KVCACHE_SPACE=40 opea/vllm-cpu:latest --model $model_name --host 0.0.0.0 --port 80
4444
fi

comps/llms/text-generation/vllm/llama_index/docker_compose_llm.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ services:
2323
cap_add:
2424
- SYS_NICE
2525
ipc: host
26-
command: --enforce-eager --model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80
26+
command: --model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80
2727
llm:
2828
image: opea/llm-vllm-llamaindex:latest
2929
container_name: llm-vllm-gaudi-server

tests/llms/test_llms_text-generation_vllm_langchain_on_intel_hpu.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ function start_service() {
4848
--ipc=host \
4949
-e HF_TOKEN=${HUGGINGFACEHUB_API_TOKEN} \
5050
opea/vllm-gaudi:comps \
51-
--enforce-eager --model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80 --block-size 128 --max-num-seqs 256 --max-seq_len-to-capture 2048
51+
--model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80 --block-size 128 --max-num-seqs 256 --max-seq_len-to-capture 2048
5252

5353
export vLLM_ENDPOINT="http://${ip_address}:${port_number}"
5454
docker run -d --rm \
@@ -62,7 +62,7 @@ function start_service() {
6262

6363
# check whether vllm ray is fully ready
6464
n=0
65-
until [[ "$n" -ge 120 ]] || [[ $ready == true ]]; do
65+
until [[ "$n" -ge 160 ]] || [[ $ready == true ]]; do
6666
docker logs test-comps-vllm-service > ${WORKPATH}/tests/test-comps-vllm-service.log
6767
n=$((n+1))
6868
if grep -q throughput ${WORKPATH}/tests/test-comps-vllm-service.log; then

tests/llms/test_llms_text-generation_vllm_llamaindex_on_intel_hpu.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ function start_service() {
4848
--ipc=host \
4949
-e HF_TOKEN=${HUGGINGFACEHUB_API_TOKEN} \
5050
opea/vllm-gaudi:comps \
51-
--enforce-eager --model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80 --block-size 128 --max-num-seqs 256 --max-seq_len-to-capture 2048
51+
--model $LLM_MODEL --tensor-parallel-size 1 --host 0.0.0.0 --port 80 --block-size 128 --max-num-seqs 256 --max-seq_len-to-capture 2048
5252

5353
export vLLM_ENDPOINT="http://${ip_address}:${port_number}"
5454
docker run -d --rm \
@@ -62,7 +62,7 @@ function start_service() {
6262

6363
# check whether vllm ray is fully ready
6464
n=0
65-
until [[ "$n" -ge 120 ]] || [[ $ready == true ]]; do
65+
until [[ "$n" -ge 160 ]] || [[ $ready == true ]]; do
6666
docker logs test-comps-vllm-service > ${WORKPATH}/tests/test-comps-vllm-service.log
6767
n=$((n+1))
6868
if grep -q throughput ${WORKPATH}/tests/test-comps-vllm-service.log; then

0 commit comments

Comments
 (0)