From f4d9f39a94f4dea478a489a7c9d31dedbad06bd5 Mon Sep 17 00:00:00 2001
From: Peter Goldsborough <psag@fb.com>
Date: Tue, 11 Sep 2018 20:27:25 -0700
Subject: [PATCH] Test libtorch on cuda

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/11554

Differential Revision: D9784772

Pulled By: goldsborough

fbshipit-source-id: c3e071695f56c1f427984f427b1f7722722947d3
---
 .jenkins/pytorch/common.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.jenkins/pytorch/common.sh b/.jenkins/pytorch/common.sh
index 5ce6ee01a46975..ca728df2b826c5 100644
--- 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