File tree Expand file tree Collapse file tree 3 files changed +1
-9
lines changed Expand file tree Collapse file tree 3 files changed +1
-9
lines changed Original file line number Diff line number Diff line change 55eval " $( ./conda/bin/conda shell.bash hook) "
66conda activate ./env
77
8- export PYTORCH_TEST_WITH_SLOW=' 1'
98python -m torch.utils.collect_env
109pytest --junitxml=test-results/junit.xml -v --durations 20
Original file line number Diff line number Diff line change @@ -8,6 +8,5 @@ conda activate ./env
88this_dir=" $( cd " $( dirname " ${BASH_SOURCE[0]} " ) " > /dev/null 2>&1 && pwd ) "
99source " $this_dir /set_cuda_envs.sh"
1010
11- export PYTORCH_TEST_WITH_SLOW=' 1'
1211python -m torch.utils.collect_env
1312pytest --junitxml=test-results/junit.xml -v --durations 20
Original file line number Diff line number Diff line change @@ -129,16 +129,10 @@ def get_export_import_copy(m):
129129 imported = torch .jit .load (path )
130130 return imported
131131
132- TEST_WITH_SLOW = os .getenv ("PYTORCH_TEST_WITH_SLOW" , "0" ) == "1"
133- if not TEST_WITH_SLOW or skip :
132+ if skip :
134133 # TorchScript is not enabled, skip these tests
135134 msg = (
136135 f"The check_jit_scriptable test for { nn_module .__class__ .__name__ } was skipped. "
137- "This test checks if the module's results in TorchScript "
138- "match eager and that it can be exported. To run these "
139- "tests make sure you set the environment variable "
140- "PYTORCH_TEST_WITH_SLOW=1 and that the test is not "
141- "manually skipped."
142136 )
143137 warnings .warn (msg , RuntimeWarning )
144138 return None
You can’t perform that action at this time.
0 commit comments