[diffusion] CI: add initial nvfp4 ci test for b200 (#21767)

Co-authored-by: Mick <mickjagger19@icloud.com>
2026-04-02 11:31:08 +08:00
parent 648632b6c4
commit 24997fe42c
8 changed files with 148 additions and 213 deletions
@@ -273,6 +273,7 @@ def handle_rerun_stage(
        "multimodal-gen-test-2-gpu",
        "multimodal-gen-component-accuracy-1-gpu",
        "multimodal-gen-component-accuracy-2-gpu",
+        "multimodal-gen-test-1-b200",
    ]

    # Valid AMD stage names that support target_stage