Test running a CUDA build on CPU machine. (#18242)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18242
ghimport-source-id: b949d312a48226a34f90304162e910acee7c95cd

Stack from [ghstack](https://github.com/ezyang/ghstack):
* **#18242 Test running a CUDA build on CPU machine.**
* #18362 Add ability to query if built with CUDA and MKL-DNN.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Differential Revision: D14584429

fbshipit-source-id: b54de5b33f0c795a7d9605d30576cdf9b74050fd

diff --git a/.circleci/cimodel/data/pytorch_build_definitions.py b/.circleci/cimodel/data/pytorch_build_definitions.py
index 48bfb67..78ce3d2 100644 (file)
--- a/.circleci/cimodel/data/pytorch_build_definitions.py
+++ b/.circleci/cimodel/data/pytorch_build_definitions.py
@@ -83,7 +83,7 @@ class Conf(object):
         if self.pyver:
             env_dict["PYTHON_VERSION"] = miniutils.quote(self.pyver)
 
-        if build_or_test == "test" and self.cuda_version:
+        if build_or_test == "test" and self.gpu_resource:
             env_dict["USE_CUDA_DOCKER_RUNTIME"] = miniutils.quote("1")
 
         d = {
@@ -143,6 +143,7 @@ def gen_dependent_configs(xenial_parent_config):
         (["NO_AVX2"], "medium"),
         (["NO_AVX", "NO_AVX2"], "medium"),
         (["slow"], "medium"),
+        (["nogpu"], None),
     ]
 
     configs = []

diff --git a/.circleci/config.yml b/.circleci/config.yml
index f9c23b4..a4bcd4b 100644 (file)
--- a/.circleci/config.yml
+++ b/.circleci/config.yml
@@ -1279,6 +1279,14 @@ jobs:
     resource_class: gpu.medium
     <<: *pytorch_linux_test_defaults
 
+  pytorch_linux_xenial_cuda8_cudnn7_py3_nogpu_test:
+    environment:
+      BUILD_ENVIRONMENT: pytorch-linux-xenial-cuda8-cudnn7-py3-nogpu-test
+      DOCKER_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-xenial-cuda8-cudnn7-py3:291"
+      PYTHON_VERSION: "3.6"
+    resource_class: large
+    <<: *pytorch_linux_test_defaults
+
   pytorch_linux_xenial_cuda9_cudnn7_py2_build:
     environment:
       BUILD_ENVIRONMENT: pytorch-linux-xenial-cuda9-cudnn7-py2-build
@@ -3102,6 +3110,9 @@ workflows:
       - pytorch_linux_xenial_cuda8_cudnn7_py3_slow_test:
           requires:
             - pytorch_linux_xenial_cuda8_cudnn7_py3_build
+      - pytorch_linux_xenial_cuda8_cudnn7_py3_nogpu_test:
+          requires:
+            - pytorch_linux_xenial_cuda8_cudnn7_py3_build
       - pytorch_short_perf_test_gpu:
           requires:
             - pytorch_linux_xenial_cuda8_cudnn7_py3_build

diff --git a/.jenkins/pytorch/enabled-configs.txt b/.jenkins/pytorch/enabled-configs.txt
index d5c6627..d3198d3 100644 (file)
--- a/.jenkins/pytorch/enabled-configs.txt
+++ b/.jenkins/pytorch/enabled-configs.txt
@@ -8,6 +8,7 @@
 pytorch-linux-xenial-cuda8-cudnn7-py3-build
 pytorch-linux-xenial-cuda8-cudnn7-py3-test
 pytorch-linux-xenial-cuda8-cudnn7-py3-multigpu-test
+pytorch-linux-xenial-cuda8-cudnn7-py3-nogpu-test
 pytorch-linux-xenial-cuda9-cudnn7-py2-build
 pytorch-linux-xenial-cuda9-cudnn7-py2-test
 pytorch-linux-xenial-cuda9-cudnn7-py3-build