diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml index dc61e891dbfe..1d71b283ec08 100644 --- a/.github/workflows/pr-test.yml +++ b/.github/workflows/pr-test.yml @@ -583,6 +583,7 @@ jobs: fi python3 run_suite.py --hw cpu --suite stage-a-cpu-only $CONTINUE_ON_ERROR_FLAG + # Runs on 5090 (32GB, SM120) stage-b-test-small-1-gpu: needs: [check-changes, call-gate, stage-a-test-1, sgl-kernel-build-wheels] if: | @@ -635,7 +636,7 @@ jobs: fi python3 run_suite.py --hw cuda --suite stage-b-test-small-1-gpu --auto-partition-id ${{ matrix.partition }} --auto-partition-size 8 $CONTINUE_ON_ERROR_FLAG - # Runs on H200 (80GB, SM90) - tests that don't pass on 5090 (FA3, FP8, high VRAM, etc.) + # Runs on H100 (80GB, SM90) - tests that don't pass on 5090 (FA3, FP8, high VRAM, etc.) stage-b-test-large-1-gpu: needs: [check-changes, call-gate, stage-a-test-1, sgl-kernel-build-wheels] if: |