diff --git a/scripts/ci/utils/slash_command_handler.py b/scripts/ci/utils/slash_command_handler.py index b2c48d301..9e7f98f87 100644 --- a/scripts/ci/utils/slash_command_handler.py +++ b/scripts/ci/utils/slash_command_handler.py @@ -256,6 +256,7 @@ def handle_rerun_stage( "stage-b-test-small-1-gpu", "stage-b-test-large-1-gpu", "stage-b-test-large-2-gpu", + "stage-b-test-4-gpu-b200", "stage-c-test-4-gpu-h100", "stage-c-test-8-gpu-h200", "stage-c-test-8-gpu-h20", @@ -265,22 +266,6 @@ def handle_rerun_stage( "stage-c-test-deepep-8-gpu-h200", "multimodal-gen-test-1-gpu", "multimodal-gen-test-2-gpu", - "quantization-test", - "stage-b-test-4-gpu-b200", - "unit-test-backend-4-gpu", - "unit-test-backend-8-gpu-h200", - "unit-test-backend-8-gpu-h20", - "unit-test-backend-8-gpu-b200", - "performance-test-1-gpu-part-1", - "performance-test-1-gpu-part-2", - "performance-test-1-gpu-part-3", - "performance-test-2-gpu", - "accuracy-test-1-gpu", - "accuracy-test-2-gpu", - "unit-test-deepep-4-gpu", - "unit-test-deepep-8-gpu", - "unit-test-backend-4-gpu-b200", - "unit-test-backend-4-gpu-gb200", ] # Valid AMD stage names that support target_stage