From 38fcd765df2d14b0d8772bf6d0612560ce0e10b8 Mon Sep 17 00:00:00 2001 From: gameofdimension Date: Fri, 24 May 2024 16:27:18 +0000 Subject: [PATCH] update debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py. Signed-off-by: gameofdimension --- debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py | 1 + 1 file changed, 1 insertion(+) diff --git a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py index 856cb237ca7..2afb901c8b1 100644 --- a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py +++ b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py @@ -394,6 +394,7 @@ def _run_ut(parser=None): def run_ut_command(args): if not is_gpu: torch.npu.set_compile_mode(jit_compile=args.jit_compile) + torch.npu.conv.allow_hf32 = False used_device = current_device + ":" + str(args.device_id[0]) try: if is_gpu: -- Gitee