Revert "Only throw PYTORCH_JIT_USE_NNC_NOT_NVFUSER=0 warning when it's actually 0"

kavanase · kavanase · commit d28078ea9faf · 2024-07-12T13:53:03.000-04:00
This reverts commit 1509463.
diff --git a/nequip/utils/_global_options.py b/nequip/utils/_global_options.py
@@ -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!"
         )

Original file line number	Diff line number	Diff line change
`@@ -91,7 +91,7 @@ def _set_global_options(config, warn_on_override: bool = False) -> None:`
`91`	`91`	# - Disable nvfuser with env variable `export PYTORCH_JIT_ENABLE_NVFUSER=0`.
`92`	`92`	`#`
`93`	`93`	`k = "PYTORCH_JIT_USE_NNC_NOT_NVFUSER"`
`94`		`- if k in os.environ and int(os.environ["PYTORCH_JIT_USE_NNC_NOT_NVFUSER"]) == 0:`
	`94`	`+ if k in os.environ:`
`95`	`95`	`warnings.warn(`
`96`	`96`	`"Do NOT manually set PYTORCH_JIT_USE_NNC_NOT_NVFUSER=0 unless you know exactly what you're doing!"`
`97`	`97`	`)`