[CI][AMD] Avoid expandable segments in LoRA TP tests (#55094)
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com>
Andreas Karatzas committed
Sep 3, 2026 at 04:55 UTC
4cc0cb6f76a608622a29d9ea8f4d415697e21364
1 file changed
+1
-1
.buildkite/test_areas/lora.yaml
+1
-1
@@ -70,7 +70,7 @@ steps:
70
- tests/lora
71
commands:
72
- export VLLM_WORKER_MULTIPROC_METHOD=spawn
73
- - export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
73
+ # expandable_segments breaks custom all-reduce IPC on ROCm.
74
- case "$$BUILDKITE_PARALLEL_JOB" in 0|1|2|3) ;; *) echo "unexpected BUILDKITE_PARALLEL_JOB=$$BUILDKITE_PARALLEL_JOB" && exit 1;; esac
75
- test "$$BUILDKITE_PARALLEL_JOB" != "3" || pytest -v -s -x lora/test_chatglm3_tp.py
76
- test "$$BUILDKITE_PARALLEL_JOB" != "1" || pytest -v -s -x lora/test_llama_tp.py