Another fix

pytorch · May 30, 2024 · 22264db · 22264db
1 parent 0146901
commit 22264db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py
@@ -3974,9 +3974,9 @@ def run(runner, args, original_dir=None):
             assert "cuda" in args.devices, "Quantization requires CUDA device."
             assert args.bfloat16, "Quantization requires dtype bfloat16."
             try:
-                from .torchao_backend import setup_baseline, torchao_optimize_ctx
-            except ImportError:
                 from torchao_backend import setup_baseline, torchao_optimize_ctx
+            except ImportError:
+                from userbenchmark.dynamo.dynamobench.torchao_backend import setup_baseline, torchao_optimize_ctx
 
             setup_baseline()
             baseline_ctx = functools.partial(