Test libtorch on cuda

Summary: Pull Request resolved: pytorch#11554 Differential Revision: D9784772 Pulled By: goldsborough fbshipit-source-id: c3e071695f56c1f427984f427b1f7722722947d3
Tianshu-Bao · Sep 12, 2018 · f4d9f39 · f4d9f39
1 parent 35348da
commit f4d9f39
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.jenkins/pytorch/common.sh b/.jenkins/pytorch/common.sh
@@ -112,7 +112,8 @@ else
   exit 1
 fi
 
-if [[ "$BUILD_ENVIRONMENT" == *pytorch-linux-trusty-py3.6-gcc7* ]]; then
+if [[ "$BUILD_ENVIRONMENT" == *pytorch-linux-xenial-cuda9-cudnn7-py3 ]] || \
+   [[ "$BUILD_ENVIRONMENT" == *pytorch-linux-trusty-py3.6-gcc7* ]]; then
   BUILD_TEST_LIBTORCH=1
 else
   BUILD_TEST_LIBTORCH=0