[CI] Add Blackwell AsyncTP correctness test (#35871)

Signed-off-by: Stefano Castagnetta <scastagnetta@nvidia.com>
This commit is contained in:
Stefano Castagnetta
2026-03-04 20:41:21 +01:00
committed by GitHub
parent 417fd28fb1
commit d7166e74c1
2 changed files with 15 additions and 0 deletions

View File

@@ -36,6 +36,16 @@ steps:
- export VLLM_TEST_CLEAN_GPU_MEMORY=1
- pytest -v -s tests/compile/correctness_e2e/test_async_tp.py
- label: AsyncTP Correctness Tests (B200)
timeout_in_minutes: 50
working_dir: "/vllm-workspace/"
device: b200
optional: true
num_devices: 2
commands:
- export VLLM_TEST_CLEAN_GPU_MEMORY=1
- pytest -v -s tests/compile/correctness_e2e/test_async_tp.py
- label: Distributed Compile Unit Tests (2xH100)
timeout_in_minutes: 20
working_dir: "/vllm-workspace/"