We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c0211b0 commit 2f2b22dCopy full SHA for 2f2b22d
timm/utils/jit.py
@@ -35,8 +35,8 @@ def set_jit_fuser(fuser):
35
torch._C._jit_set_texpr_fuser_enabled(False)
36
elif fuser == "nvfuser" or fuser == "nvf":
37
os.environ['PYTORCH_NVFUSER_DISABLE_FALLBACK'] = '1'
38
- os.environ['PYTORCH_NVFUSER_DISABLE_FMA'] = '1'
39
- os.environ['PYTORCH_NVFUSER_JIT_OPT_LEVEL'] = '0'
+ #os.environ['PYTORCH_NVFUSER_DISABLE_FMA'] = '1'
+ #os.environ['PYTORCH_NVFUSER_JIT_OPT_LEVEL'] = '0'
40
41
torch._C._jit_set_profiling_executor(True)
42
torch._C._jit_set_profiling_mode(True)
0 commit comments