Fix walrus operator precedence. (#6374)

pytorch · Apr 22, 2024 · bd77aab · bd77aab
1 parent 24eb3c1
commit bd77aab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py
@@ -300,7 +300,7 @@ def load_benchmark(self):
     # torch.backends.__allow_nonbracketed_mutation_flag = True
 
     # torchbench uses `xla` as device instead of `tpu`
-    if device := self.benchmark_experiment.accelerator == 'tpu':
+    if (device := self.benchmark_experiment.accelerator) == 'tpu':
       device = str(self.benchmark_experiment.get_device())
     return benchmark_cls(
         test=self.benchmark_experiment.test,