From 2f2b22d8c7889174dbf11b92c2d72d8587f9164b Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 13 May 2022 09:26:16 -0700 Subject: [PATCH] Disable nvfuser fma / opt level overrides per #1244 --- timm/utils/jit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/utils/jit.py b/timm/utils/jit.py index 8ebfdbff..a32cbd40 100644 --- a/timm/utils/jit.py +++ b/timm/utils/jit.py @@ -35,8 +35,8 @@ def set_jit_fuser(fuser): torch._C._jit_set_texpr_fuser_enabled(False) elif fuser == "nvfuser" or fuser == "nvf": os.environ['PYTORCH_NVFUSER_DISABLE_FALLBACK'] = '1' - os.environ['PYTORCH_NVFUSER_DISABLE_FMA'] = '1' - os.environ['PYTORCH_NVFUSER_JIT_OPT_LEVEL'] = '0' + #os.environ['PYTORCH_NVFUSER_DISABLE_FMA'] = '1' + #os.environ['PYTORCH_NVFUSER_JIT_OPT_LEVEL'] = '0' torch._C._jit_set_texpr_fuser_enabled(False) torch._C._jit_set_profiling_executor(True) torch._C._jit_set_profiling_mode(True)