summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2022-06-02 17:51:18 -0700
committerGitHub <noreply@github.com>2022-06-02 20:51:18 -0400
commit043cc688ef8ab174e507990464ae525ee38e9fae (patch)
tree13afc791f5f48547dec06694612ad2b5a5b5f240
parente52527029a79d5a60ad1da3eb57b5400fd1c73db (diff)
downloadspack-043cc688ef8ab174e507990464ae525ee38e9fae.tar.gz
spack-043cc688ef8ab174e507990464ae525ee38e9fae.tar.bz2
spack-043cc688ef8ab174e507990464ae525ee38e9fae.tar.xz
spack-043cc688ef8ab174e507990464ae525ee38e9fae.zip
py-torch: fix Apple M1 support (#30833)
-rw-r--r--var/spack/repos/builtin/packages/py-torch/package.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py
index 8a89e31468..3a9b57cc00 100644
--- a/var/spack/repos/builtin/packages/py-torch/package.py
+++ b/var/spack/repos/builtin/packages/py-torch/package.py
@@ -84,6 +84,9 @@ class PyTorch(PythonPackage, CudaPackage):
conflicts('+breakpad', when='target=ppc64:')
conflicts('+breakpad', when='target=ppc64le:')
+ # https://github.com/pytorch/pytorch/issues/77811
+ conflicts('+qnnpack', when='platform=darwin target=aarch64:')
+
conflicts('cuda_arch=none', when='+cuda',
msg='Must specify CUDA compute capabilities of your GPU, see '
'https://developer.nvidia.com/cuda-gpus')
@@ -173,7 +176,7 @@ class PyTorch(PythonPackage, CudaPackage):
# https://github.com/pytorch/pytorch/issues/60328
patch('https://github.com/pytorch/pytorch/pull/59220.patch?full_index=1',
sha256='6d5717267f901e8ee493dfacd08734d9bcc48ad29a76ca9ef702368e96bee675',
- when='@1.2:')
+ when='@1.2:1.11')
# Fixes build on older systems with glibc <2.12
patch('https://github.com/pytorch/pytorch/pull/55063.patch?full_index=1',