Skip to content

Commit 2f2b22d

Browse files
committed
Disable nvfuser fma / opt level overrides per #1244
1 parent c0211b0 commit 2f2b22d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/utils/jit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,8 +35,8 @@ def set_jit_fuser(fuser):
3535
torch._C._jit_set_texpr_fuser_enabled(False)
3636
elif fuser == "nvfuser" or fuser == "nvf":
3737
os.environ['PYTORCH_NVFUSER_DISABLE_FALLBACK'] = '1'
38-
os.environ['PYTORCH_NVFUSER_DISABLE_FMA'] = '1'
39-
os.environ['PYTORCH_NVFUSER_JIT_OPT_LEVEL'] = '0'
38+
#os.environ['PYTORCH_NVFUSER_DISABLE_FMA'] = '1'
39+
#os.environ['PYTORCH_NVFUSER_JIT_OPT_LEVEL'] = '0'
4040
torch._C._jit_set_texpr_fuser_enabled(False)
4141
torch._C._jit_set_profiling_executor(True)
4242
torch._C._jit_set_profiling_mode(True)

0 commit comments

Comments
 (0)