From 53fe8834ec1787f0fc02d4f55715b4f77af89e89 Mon Sep 17 00:00:00 2001 From: Rahul Sharma Date: Thu, 29 Jan 2026 06:46:47 +0000 Subject: [PATCH] fix gemm timing logic --- Ironwood/src/benchmark_gemm.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/Ironwood/src/benchmark_gemm.py b/Ironwood/src/benchmark_gemm.py index 99564490..b802ddc0 100644 --- a/Ironwood/src/benchmark_gemm.py +++ b/Ironwood/src/benchmark_gemm.py @@ -213,6 +213,8 @@ def data_generator(): return (lhs_device, rhs_device) # Run the benchmark + num_runs = 1 + ## Need to fix gemm timing logic to handle num_runs > 1 time_ms_list = iteration_timeit( jit_sharded_f, @@ -300,6 +302,9 @@ def data_generator(): return (lhs_device, rhs_device) + num_runs = 1 + ## Need to fix gemm timing logic to handle num_runs > 1 + # Run the benchmark time_ms_list = iteration_timeit( jit_sharded_f, @@ -402,6 +407,9 @@ def data_generator(): return (lhs_device, rhs_device, sf0_device, sf1_device) + num_runs = 1 + ## Need to fix gemm timing logic to handle num_runs > 1 + time_ms_list = iteration_timeit( jit_sharded_f, data_generator, @@ -513,6 +521,10 @@ def data_generator(): return (out_buffer_device, lhs_device, rhs_device, sf0_device, sf1_device) + + num_runs = 1 + ## Need to fix gemm timing logic to handle num_runs > 1 + time_ms_list = iteration_timeit( jit_sharded_f, data_generator,