summaryrefslogtreecommitdiff
path: root/var
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2019-11-10 16:46:04 -0600
committerGitHub <noreply@github.com>2019-11-10 16:46:04 -0600
commit1ca91d728a9f4324d15793ad32a95054c5883ea0 (patch)
tree015d9e6fe42a8b822d5dc4b8290dcba5b6d02f4d /var
parentff330631ed0c428e5cd01f747c33100dc6ab295e (diff)
downloadspack-1ca91d728a9f4324d15793ad32a95054c5883ea0.tar.gz
spack-1ca91d728a9f4324d15793ad32a95054c5883ea0.tar.bz2
spack-1ca91d728a9f4324d15793ad32a95054c5883ea0.tar.xz
spack-1ca91d728a9f4324d15793ad32a95054c5883ea0.zip
Update default PyTorch variants (#13667)
Diffstat (limited to 'var')
-rw-r--r--var/spack/repos/builtin/packages/py-torch/package.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py
index 8e4c1de0ce..b5d8074deb 100644
--- a/var/spack/repos/builtin/packages/py-torch/package.py
+++ b/var/spack/repos/builtin/packages/py-torch/package.py
@@ -62,16 +62,16 @@ class PyTorch(PythonPackage):
version('0.3.1', tag='v0.3.1', submodules=True)
variant('cuda', default=True, description='Enables CUDA build')
- variant('cudnn', default=False, description='Enables the cuDNN 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')
variant('test', default=False, description='Enables the test build')
variant('miopen', default=False, description='Enables the MIOpen build')
- variant('mkldnn', default=False, description='Enables use of MKLDNN')
+ variant('mkldnn', default=True, description='Enables use of MKLDNN')
variant('nnpack', default=False, description='Enables NNPACK build')
variant('qnnpack', default=False, description='Enables QNNPACK build (quantized 8-bit operators)')
variant('distributed', default=False, description='Enables distributed (c10d, gloo, mpi, etc.) build')
- variant('nccl', default=False, description='Use Spack-installed NCCL')
+ variant('nccl', default=True, description='Use Spack-installed NCCL')
variant('caffe2', default=False, description='Enables Caffe2 operators build')
variant('gloo', default=False, description='Enables features related to distributed support')
variant('opencv', default=False, description='Enables use of OpenCV for additional operators')