diff --git a/userbenchmark/dynamo/dynamobench/torchbench.py b/userbenchmark/dynamo/dynamobench/torchbench.py index 2a9437e08b..bf497a2b19 100755 --- a/userbenchmark/dynamo/dynamobench/torchbench.py +++ b/userbenchmark/dynamo/dynamobench/torchbench.py @@ -432,7 +432,7 @@ def forward_and_backward_pass(self, mod, inputs, collect_outputs=True): cloned_inputs = clone_inputs(inputs) self.optimizer_zero_grad(mod) with self.autocast(**self.autocast_arg): - if isinstance(clone_inputs, dict): + if isinstance(cloned_inputs, dict): pred = mod(**cloned_inputs) else: pred = mod(*cloned_inputs)