From d28078ea9fafa171aa6bd11e4480c61b525e56bf Mon Sep 17 00:00:00 2001 From: Sean Kavanagh Date: Fri, 12 Jul 2024 13:53:03 -0400 Subject: [PATCH] Revert "Only throw `PYTORCH_JIT_USE_NNC_NOT_NVFUSER=0` warning when it's actually 0" This reverts commit 1509463f3ed1c72fac0fc6efe6e9d7ae09203edf. --- nequip/utils/_global_options.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nequip/utils/_global_options.py b/nequip/utils/_global_options.py index fd9be40f..3a08e55e 100644 --- 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!" )