hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update default PyTorch variants (#13667)

Browse source
This commit is contained in:
Adam J. Stewart 2019-11-10 16:46:04 -06:00 committed by GitHub
parent ff330631ed
commit 1ca91d728a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
var/spack/repos/builtin/packages/py-torch/package.py
View file

@ -62,16 +62,16 @@ class PyTorch(PythonPackage):
version('0.3.1', tag='v0.3.1', submodules=True) version('0.3.1', tag='v0.3.1', submodules=True)
variant('cuda', default=True, description='Enables CUDA build') 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('magma', default=False, description='Enables the MAGMA build')
variant('fbgemm', default=False, description='Enables the FBGEMM build') variant('fbgemm', default=False, description='Enables the FBGEMM build')
variant('test', default=False, description='Enables the test build') variant('test', default=False, description='Enables the test build')
variant('miopen', default=False, description='Enables the MIOpen 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('nnpack', default=False, description='Enables NNPACK build')
variant('qnnpack', default=False, description='Enables QNNPACK build (quantized 8-bit operators)') 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('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('caffe2', default=False, description='Enables Caffe2 operators build')
variant('gloo', default=False, description='Enables features related to distributed support') variant('gloo', default=False, description='Enables features related to distributed support')
variant('opencv', default=False, description='Enables use of OpenCV for additional operators') variant('opencv', default=False, description='Enables use of OpenCV for additional operators')