add nn_moe

a1b2eff7 · zhuwenwen · 4924f6b5 · a1b2eff7
Commit a1b2eff7 authored Dec 04, 2025 by zhuwenwen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

benchmarks/kernels/benchmark_moe.py benchmarks/kernels/benchmark_moe.py +1 -0

No files found.
--- a/benchmarks/kernels/benchmark_moe.py
+++ b/benchmarks/kernels/benchmark_moe.py
@@ -208,6 +208,7 @@ def benchmark_config(
                inplace=True,
                quant_config=quant_config,
                allow_deep_gemm=use_deep_gemm,
+                use_nn_moe=nn_moe,
            )
    # JIT compilation & warmup