[diffusion] fix: fix accuracy bug caused by #14717 (#18296)

2026-02-05 20:36:18 +08:00
parent 8f8c1724ae
commit 4aa03d91fd
3 changed files with 6 additions and 6 deletions
--- a/python/sglang/jit_kernel/tests/test_fused_norm_scale_shift.py
+++ b/python/sglang/jit_kernel/tests/test_fused_norm_scale_shift.py
@@ -24,11 +24,11 @@ SHAPE_MAP = {
 }
 SHAPES = [
    # (B, S, F, D)
-    (1, 1024, 8, 3072),
-    (4, 512, 16, 3072),
    (1, 115200, 1, 3072),  # Hunyuan
    (1, 32760, 1, 1536),  # Wan
    (1, 6, 1, 3072),  # Qwen
+    (1, 1024, 8, 3072),
+    (4, 512, 16, 3072),
 ]
 DTYPES = [torch.float16, torch.bfloat16, torch.float32]
 NORM_TYPES = ["layer", "rms"]