From 22264db97a2022017ce9ff368f958178bb76892f Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 30 May 2024 14:27:26 -0400 Subject: [PATCH] Another fix --- userbenchmark/dynamo/dynamobench/common.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/userbenchmark/dynamo/dynamobench/common.py b/userbenchmark/dynamo/dynamobench/common.py index 466e6b30d0..21e916e080 100644 --- 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(