Skip to content

Commit 560ea15

Browse files
committed
[Ascend] Fix input param of do_bench_npu
1 parent f7ae8e6 commit 560ea15

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

benchmark/base.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -310,8 +310,9 @@ def get_latency(self, op, *args, **kwargs):
310310
do_bench = triton.backends.ascend.testing.do_bench_npu
311311
latency = do_bench(
312312
fn,
313-
warmup=Config.warm_up,
314-
active=Config.repetition,
313+
# do_bench_npu requires iterations, rather than duration
314+
# warmup=Config.warm_up,
315+
# active=Config.repetition,
315316
)
316317
else:
317318
do_bench = triton.testing.do_bench

0 commit comments

Comments
 (0)