Use torch.compile for scaling penalty (#3133)

2025-01-25 18:27:33 -08:00
parent da6f8081f6
commit 27acf63bbd
3 changed files with 14 additions and 29 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
@@ -1,6 +1,5 @@
 import argparse
 import itertools
-import time

 import torch
 import triton