Add A100 fused MoE kernel configs for Dpsk (#9677)

2025-08-27 11:49:48 +08:00
parent b9683be653
commit 8f7b1c31e8
2 changed files with 147 additions and 1 deletions
--- a/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py
+++ b/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py
@@ -336,7 +336,7 @@ class BenchmarkWorker:
                        block_shape,
                        num_iters=10,
                    )
-                except triton.runtime.autotuner.OutOfResources:
+                except (triton.runtime.autotuner.OutOfResources, RuntimeError):
                    # Some configurations may be invalid and fail to compile.
                    continue