From 3ed01aa9f3e200a9744dd25b8864709674d274b6 Mon Sep 17 00:00:00 2001 From: zpcore Date: Fri, 17 May 2024 18:10:25 +0000 Subject: [PATCH] update input list --- benchmarks/experiment_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/experiment_runner.py b/benchmarks/experiment_runner.py index 00bb0f657159..951587ff3724 100644 --- a/benchmarks/experiment_runner.py +++ b/benchmarks/experiment_runner.py @@ -285,7 +285,7 @@ def run_once_and_gather_metrics(self, benchmark_experiment, benchmark_model, # Reset state and sync. reset_rng_state(benchmark_experiment) if benchmark_experiment.torch_xla2: - self._mark_step(benchmark_experiment, inputs) + self._mark_step(benchmark_experiment, inputs_list) else: self._mark_step(benchmark_experiment) self._synchronize(benchmark_experiment)