zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-torch: fix Apple M1 support (#30833)

Browse Source
This commit is contained in:
Adam J. Stewart 2022-06-02 17:51:18 -07:00 committed by GitHub
parent e52527029a
commit 043cc688ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
var/spack/repos/builtin/packages/py-torch/package.py
View File

@ -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',