Skip to content

Commit

Permalink
Revert "Only throw PYTORCH_JIT_USE_NNC_NOT_NVFUSER=0 warning when i…
Browse files Browse the repository at this point in the history
…t's actually 0"

This reverts commit 1509463.
  • Loading branch information
kavanase committed Jul 12, 2024
1 parent 6f0a53c commit d28078e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nequip/utils/_global_options.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def _set_global_options(config, warn_on_override: bool = False) -> None:
# - Disable nvfuser with env variable `export PYTORCH_JIT_ENABLE_NVFUSER=0`.
#
k = "PYTORCH_JIT_USE_NNC_NOT_NVFUSER"
if k in os.environ and int(os.environ["PYTORCH_JIT_USE_NNC_NOT_NVFUSER"]) == 0:
if k in os.environ:
warnings.warn(
"Do NOT manually set PYTORCH_JIT_USE_NNC_NOT_NVFUSER=0 unless you know exactly what you're doing!"
)
Expand Down

0 comments on commit d28078e

Please sign in to comment.