[AMD] Fix dpsk-v32 accuracy issue on mi355 (#20840)

2026-03-18 17:06:15 +08:00
parent f0d7a3f427
commit c0a4408f78
1 changed files with 1 additions and 0 deletions
--- a/python/sglang/srt/layers/quantization/fp8_utils.py
+++ b/python/sglang/srt/layers/quantization/fp8_utils.py
@@ -68,6 +68,7 @@ def use_aiter_triton_gemm_w8a8_tuned_gfx950(n: int, k: int) -> bool:
        (4608, 7168),
        (512, 7168),
        (7168, 2048),
+        (7168, 2304),
        (7168, 16384),
        (7168, 256),
        (8192, 1024),