From bd77aaba1bb51cc10ec654201f3f65717b46d2e0 Mon Sep 17 00:00:00 2001 From: Yukio Siraichi Date: Thu, 25 Jan 2024 08:02:19 -0300 Subject: [PATCH] Fix walrus operator precedence. (#6374) --- benchmarks/torchbench_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py index 6b809f8ac9f..de697324e8d 100644 --- 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,