clean moe align block kernel code and add acc test (#3332)

2025-02-06 16:42:36 +08:00
parent 32de54ed1a
commit ad3499858e
4 changed files with 237 additions and 80 deletions
--- a/benchmark/kernels/fused_moe_triton/benchmark_deepseekv3_moe_align_blocks.py
+++ b/benchmark/kernels/fused_moe_triton/benchmark_deepseekv3_moe_align_blocks.py
@@ -310,4 +310,4 @@ if __name__ == "__main__":

    calculate_diff(batch_size=4, seq_len=1024)

-    benchmark.run(print_data=True, save_path=args.save_path)
+    benchmark.run(print_data=True)