diff options
author | Adam J. Stewart <ajstewart426@gmail.com> | 2020-04-26 11:29:52 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-04-26 11:29:52 -0500 |
commit | 63a0647282da89a31244ecba920329d48f13990b (patch) | |
tree | 682d3b14a5d7e208b36e7a16420886091f26b0b3 | |
parent | 75042c381c9fd1412f0fce99aab9ebd43fd528c8 (diff) | |
download | spack-63a0647282da89a31244ecba920329d48f13990b.tar.gz spack-63a0647282da89a31244ecba920329d48f13990b.tar.bz2 spack-63a0647282da89a31244ecba920329d48f13990b.tar.xz spack-63a0647282da89a31244ecba920329d48f13990b.zip |
PyTorch: disable XNNPACK build (#16304)
-rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index f786ad14bc..7329ba9174 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -74,6 +74,7 @@ class PyTorch(PythonPackage, CudaPackage): 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('xnnpack', default=False, description='Enables XNNPACK build') variant('distributed', default=False, description='Enables distributed (c10d, gloo, mpi, etc.) build') variant('nccl', default=True, description='Use Spack-installed NCCL') variant('caffe2', default=False, description='Enables Caffe2 operators build') @@ -94,6 +95,7 @@ class PyTorch(PythonPackage, CudaPackage): conflicts('+miopen', when='@:0.4') conflicts('+mkldnn', when='@:0.3') conflicts('+qnnpack', when='@:0.4') + conflicts('+xnnpack', when='@:1.4') conflicts('+nccl', when='~cuda') conflicts('+opencv', when='@:0.4') conflicts('+ffmpeg', when='@:0.4') @@ -149,6 +151,8 @@ class PyTorch(PythonPackage, CudaPackage): # TODO: add dependency: https://github.com/Maratyszcza/NNPACK # depends_on('nnpack', when='+nnpack') depends_on('qnnpack', when='+qnnpack') + # TODO: add dependency: https://github.com/google/XNNPACK + # depends_on('xnnpack', when='+xnnpack') depends_on('mpi', when='+distributed') depends_on('nccl', when='+nccl') depends_on('gloo', when='+gloo') @@ -239,6 +243,7 @@ class PyTorch(PythonPackage, CudaPackage): enable_or_disable('nnpack') enable_or_disable('qnnpack') + enable_or_disable('xnnpack') enable_or_disable('distributed') enable_or_disable('nccl') |