[Bugfix] Fix benchmark_moe.py (#23177)

Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
This commit is contained in:
Jee Jee Li
2025-08-19 21:39:40 +08:00
committed by GitHub
parent b87cb97a53
commit 4d9c61993a

View File

@@ -430,7 +430,6 @@ class BenchmarkWorker:
hidden_size,
topk,
dtype_str,
is_marlin=False,
)
else:
config = op_config[min(op_config.keys(), key=lambda x: abs(x - num_tokens))]