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 = {
(["NO_AVX2"], "medium"),
(["NO_AVX", "NO_AVX2"], "medium"),
(["slow"], "medium"),
+ (["nogpu"], None),
]
configs = []
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
- 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
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