From 1ca91d728a9f4324d15793ad32a95054c5883ea0 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sun, 10 Nov 2019 16:46:04 -0600 Subject: [PATCH] Update default PyTorch variants (#13667) --- var/spack/repos/builtin/packages/py-torch/package.py | 6 +++--- 1 file 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')