ci: run Stage A CUDA tests as stage-a-test-small-1-gpu on 5090 (#20988)

This commit is contained in:
Lianmin Zheng
2026-03-20 02:55:16 -07:00
committed by GitHub
parent fa89d152c0
commit c1da420799
12 changed files with 78 additions and 100 deletions
+3 -3
View File
@@ -24,7 +24,7 @@ HW_MAPPING = {
PER_COMMIT_SUITES = {
HWBackend.CPU: ["stage-a-cpu-only"],
HWBackend.AMD: [
"stage-a-test-1-amd",
"stage-a-test-small-1-gpu-amd",
"stage-b-test-small-1-gpu-amd",
"stage-b-test-small-1-gpu-amd-nondeterministic",
"stage-b-test-small-1-gpu-amd-mi35x",
@@ -35,7 +35,7 @@ PER_COMMIT_SUITES = {
"stage-c-test-large-8-gpu-amd-mi35x",
],
HWBackend.CUDA: [
"stage-a-test-1",
"stage-a-test-small-1-gpu",
"stage-b-test-small-1-gpu",
"stage-b-test-large-1-gpu",
"stage-b-test-large-2-gpu",
@@ -49,7 +49,7 @@ PER_COMMIT_SUITES = {
"stage-c-test-deepep-8-gpu-h200",
],
HWBackend.NPU: [
"stage-a-test-1",
"stage-a-test-small-1-gpu",
"stage-b-test-1-npu-a2",
"stage-b-test-2-npu-a2",
"stage-b-test-4-npu-a3",