update input list

pytorch · May 17, 2024 · 3ed01aa · 3ed01aa
1 parent e5bc96a
commit 3ed01aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)