From 4cc0cb6f76a608622a29d9ea8f4d415697e21364 Mon Sep 17 00:00:00 2001 From: Andreas Karatzas Date: Thu, 3 Sep 2026 04:55:10 -0500 Subject: [PATCH] [CI][AMD] Avoid expandable segments in LoRA TP tests (#55094) Signed-off-by: Andreas Karatzas --- .buildkite/test_areas/lora.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.buildkite/test_areas/lora.yaml b/.buildkite/test_areas/lora.yaml index 573cc3fd5c..d63fb567ea 100644 --- a/.buildkite/test_areas/lora.yaml +++ b/.buildkite/test_areas/lora.yaml @@ -70,7 +70,7 @@ steps: - tests/lora commands: - export VLLM_WORKER_MULTIPROC_METHOD=spawn - - export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True + # expandable_segments breaks custom all-reduce IPC on ROCm. - case "$$BUILDKITE_PARALLEL_JOB" in 0|1|2|3) ;; *) echo "unexpected BUILDKITE_PARALLEL_JOB=$$BUILDKITE_PARALLEL_JOB" && exit 1;; esac - test "$$BUILDKITE_PARALLEL_JOB" != "3" || pytest -v -s -x lora/test_chatglm3_tp.py - test "$$BUILDKITE_PARALLEL_JOB" != "1" || pytest -v -s -x lora/test_llama_tp.py