From e76be06550895b633d1d1aca90f0725152f92294 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C5=81ukasz=20Durejko?= Date: Mon, 26 May 2025 14:26:07 +0200 Subject: [PATCH] [Hardware][Intel-Gaudi] [CI/Build] Add tensor parallel size = 2 test to HPU CI (#18709) Signed-off-by: Lukasz Durejko --- .buildkite/scripts/hardware_ci/run-hpu-test.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.buildkite/scripts/hardware_ci/run-hpu-test.sh b/.buildkite/scripts/hardware_ci/run-hpu-test.sh index 95b6ac37f1857..c3b78d4712977 100644 --- a/.buildkite/scripts/hardware_ci/run-hpu-test.sh +++ b/.buildkite/scripts/hardware_ci/run-hpu-test.sh @@ -21,4 +21,6 @@ remove_docker_container # Run the image and launch offline inference docker run --runtime=habana --name=hpu-test --network=host -e HABANA_VISIBLE_DEVICES=all -e VLLM_SKIP_WARMUP=true --entrypoint="" hpu-test-env python3 examples/offline_inference/basic/generate.py --model facebook/opt-125m +docker run --runtime=habana --name=hpu-test --network=host -e HABANA_VISIBLE_DEVICES=all -e VLLM_SKIP_WARMUP=true --entrypoint="" hpu-test-env python3 examples/offline_inference/basic/generate.py --model facebook/opt-125m --tensor-parallel-size 2 + EXITCODE=$?