From 6fd77d57d324bbb43bbe4224808c417cc3824890 Mon Sep 17 00:00:00 2001 From: Bert Maher Date: Tue, 5 Nov 2024 15:51:18 -0500 Subject: [PATCH] `fast_flush` kwarg of `do_bench` is removed Differential Revision: D65444506 Pull Request resolved: https://github.com/pytorch/ao/pull/1222 --- benchmarks/quantized_training/benchmark_int8mm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/quantized_training/benchmark_int8mm.py b/benchmarks/quantized_training/benchmark_int8mm.py index 85892afa8..0e85cd831 100644 --- a/benchmarks/quantized_training/benchmark_int8mm.py +++ b/benchmarks/quantized_training/benchmark_int8mm.py @@ -6,7 +6,7 @@ def bench_f(f, *args): - return do_bench(lambda: f(*args), fast_flush=False, return_mode="median") + return do_bench(lambda: f(*args), return_mode="median") shapes = [(sz, sz, sz) for sz in [1024, 2048, 4096]]