diff options
| author | Edward Z. Yang <ezyang@mit.edu> | 2018-03-18 00:45:28 -0400 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2018-03-18 00:45:28 -0400 |
| commit | 77042266ee6e2001d567d15a2daf9cb1d17255f5 (patch) | |
| tree | 8ce086536c973a723e3b6383864318ed644dbbab /.jenkins | |
| parent | 2f64e1cdf6bfec1e533b3687341bc30f19a9ce25 (diff) | |
| download | pytorch-77042266ee6e2001d567d15a2daf9cb1d17255f5.tar.gz pytorch-77042266ee6e2001d567d15a2daf9cb1d17255f5.tar.bz2 pytorch-77042266ee6e2001d567d15a2daf9cb1d17255f5.zip | |
Multi-gpu test. (#5854)
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Diffstat (limited to '.jenkins')
| -rw-r--r-- | .jenkins/enabled-configs.txt | 1 | ||||
| -rwxr-xr-x | .jenkins/multigpu-test.sh | 19 |
2 files changed, 20 insertions, 0 deletions
diff --git a/.jenkins/enabled-configs.txt b/.jenkins/enabled-configs.txt index 7d8d8fd566..24acd003ec 100644 --- a/.jenkins/enabled-configs.txt +++ b/.jenkins/enabled-configs.txt @@ -7,6 +7,7 @@ pytorch-linux-xenial-cuda8-cudnn6-py3-build pytorch-linux-xenial-cuda8-cudnn6-py3-test +pytorch-linux-xenial-cuda8-cudnn6-py3-multigpu-test pytorch-linux-xenial-cuda9-cudnn7-py2-build pytorch-linux-xenial-cuda9-cudnn7-py2-test pytorch-linux-xenial-cuda9-cudnn7-py3-build diff --git a/.jenkins/multigpu-test.sh b/.jenkins/multigpu-test.sh new file mode 100755 index 0000000000..477b7a26a0 --- /dev/null +++ b/.jenkins/multigpu-test.sh @@ -0,0 +1,19 @@ +#!/bin/bash + +COMPACT_JOB_NAME="${BUILD_ENVIRONMENT}-multigpu-test" +source "$(dirname "${BASH_SOURCE[0]}")/common.sh" + +# Required environment variable: $BUILD_ENVIRONMENT +# (This is set by default in the Docker images we build, so you don't +# need to set it yourself. + +export PATH=/opt/conda/bin:$PATH + +export LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs:$LD_LIBRARY_PATH +export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH + +echo "Testing pytorch" +export OMP_NUM_THREADS=4 +export MKL_NUM_THREADS=4 + +time python test/run_test.py --verbose -i distributed |