diff options
| author | Glenn Johnson <glenn-johnson@uiowa.edu> | 2020-01-16 12:02:53 -0600 |
|---|---|---|
| committer | Adam J. Stewart <ajstewart426@gmail.com> | 2020-01-16 12:02:53 -0600 |
| commit | 5aeab7dbe5215b58bf468e449b4fa6d4a82994ed (patch) | |
| tree | 3601a4ced00f5c88eff8195198bdb6494873ca2d /var | |
| parent | 8ad0be96aa83815c1ebb9d1b77f701466bbfa776 (diff) | |
| download | spack-5aeab7dbe5215b58bf468e449b4fa6d4a82994ed.tar.gz spack-5aeab7dbe5215b58bf468e449b4fa6d4a82994ed.tar.bz2 spack-5aeab7dbe5215b58bf468e449b4fa6d4a82994ed.tar.xz spack-5aeab7dbe5215b58bf468e449b4fa6d4a82994ed.zip | |
Use CudaPackage mixin for py-torch (#14540)
This PR adds CudaPackage in order to pick up the cuda/compiler conflicts
defined in CudaPackage.
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 3 |
1 files changed, 1 insertions, 2 deletions
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') |