Disable nvfuser fma / opt level overrides per #1244

3 years ago · 2f2b22d8c7
parent c0211b0bf7
commit 2f2b22d8c7
1 changed files with 2 additions and 2 deletions
--- 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)