From 5aeab7dbe5215b58bf468e449b4fa6d4a82994ed Mon Sep 17 00:00:00 2001 From: Glenn Johnson Date: Thu, 16 Jan 2020 12:02:53 -0600 Subject: Use CudaPackage mixin for py-torch (#14540) This PR adds CudaPackage in order to pick up the cuda/compiler conflicts defined in CudaPackage. --- var/spack/repos/builtin/packages/py-torch/package.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'var') diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index b34d0b08f7..73f191d306 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -7,7 +7,7 @@ from spack import * # TODO: try switching to CMakePackage for more control over build -class PyTorch(PythonPackage): +class PyTorch(PythonPackage, CudaPackage): """Tensors and Dynamic neural networks in Python with strong GPU acceleration.""" @@ -61,7 +61,6 @@ class PyTorch(PythonPackage): version('0.4.0', tag='v0.4.0', submodules=True) version('0.3.1', tag='v0.3.1', submodules=True) - variant('cuda', default=True, description='Enables CUDA build') variant('cudnn', default=True, description='Enables the cuDNN build') variant('magma', default=False, description='Enables the MAGMA build') variant('fbgemm', default=False, description='Enables the FBGEMM build') -- cgit v1.2.3-70-g09d2