Make PYTORCH_TEST_WITH_SLOW_GRADCHECK consistent with other test envvars
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75987
Approved by: https://github.com/soulitzer
diff --git a/.jenkins/pytorch/test.sh b/.jenkins/pytorch/test.sh
index a74a3e8..96c06f4 100755
--- a/.jenkins/pytorch/test.sh
+++ b/.jenkins/pytorch/test.sh
@@ -48,7 +48,7 @@
fi
if [[ "$BUILD_ENVIRONMENT" == *slow-gradcheck* ]]; then
- export PYTORCH_TEST_WITH_SLOW_GRADCHECK=ON
+ export PYTORCH_TEST_WITH_SLOW_GRADCHECK=1
fi
if [[ "$BUILD_ENVIRONMENT" == *cuda* ]]; then
diff --git a/torch/testing/_internal/common_utils.py b/torch/testing/_internal/common_utils.py
index f5f9f51..0b07291 100644
--- a/torch/testing/_internal/common_utils.py
+++ b/torch/testing/_internal/common_utils.py
@@ -2929,7 +2929,7 @@
"fast_mode": True,
}
- if os.environ.get('PYTORCH_TEST_WITH_SLOW_GRADCHECK', "0FF") == "ON":
+ if os.environ.get('PYTORCH_TEST_WITH_SLOW_GRADCHECK', "0") == "1":
default_values["fast_mode"] = False
for key, value in default_values.items():
@@ -2949,7 +2949,7 @@
"fast_mode": True,
}
- if os.environ.get('PYTORCH_TEST_WITH_SLOW_GRADCHECK', "0FF") == "ON":
+ if os.environ.get('PYTORCH_TEST_WITH_SLOW_GRADCHECK', "0") == "1":
default_values["fast_mode"] = False
for key, value in default_values.items():