File tree 3 files changed +1
-9
lines changed 3 files changed +1
-9
lines changed Original file line number Diff line number Diff line change 5
5
eval " $( ./conda/bin/conda shell.bash hook) "
6
6
conda activate ./env
7
7
8
- export PYTORCH_TEST_WITH_SLOW=' 1'
9
8
python -m torch.utils.collect_env
10
9
pytest --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
8
8
this_dir=" $( cd " $( dirname " ${BASH_SOURCE[0]} " ) " > /dev/null 2>&1 && pwd ) "
9
9
source " $this_dir /set_cuda_envs.sh"
10
10
11
- export PYTORCH_TEST_WITH_SLOW=' 1'
12
11
python -m torch.utils.collect_env
13
12
pytest --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):
129
129
imported = torch .jit .load (path )
130
130
return imported
131
131
132
- TEST_WITH_SLOW = os .getenv ("PYTORCH_TEST_WITH_SLOW" , "0" ) == "1"
133
- if not TEST_WITH_SLOW or skip :
132
+ if skip :
134
133
# TorchScript is not enabled, skip these tests
135
134
msg = (
136
135
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."
142
136
)
143
137
warnings .warn (msg , RuntimeWarning )
144
138
return None
You can’t perform that action at this time.
0 commit comments