diff options
| author | Adam J. Stewart <ajstewart426@gmail.com> | 2023-03-12 00:03:27 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-12 08:03:27 +0100 |
| commit | 6c9d079cfb11a70927706f960fba39d77e2fa96b (patch) | |
| tree | 2b7e0ada85cbb580a07b8cf3e48db179e57ed83e /var | |
| parent | a741350e69cef19f7639117b34ac889bf337d6e7 (diff) | |
| download | spack-6c9d079cfb11a70927706f960fba39d77e2fa96b.tar.gz spack-6c9d079cfb11a70927706f960fba39d77e2fa96b.tar.bz2 spack-6c9d079cfb11a70927706f960fba39d77e2fa96b.tar.xz spack-6c9d079cfb11a70927706f960fba39d77e2fa96b.zip | |
py-torch: NNPACK requires AVX2 (#35994)
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-torch/package.py | 3 |
1 files changed, 2 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 6e022ad70f..42379ac6f4 100644 --- a/var/spack/repos/builtin/packages/py-torch/package.py +++ b/var/spack/repos/builtin/packages/py-torch/package.py @@ -74,7 +74,8 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage): variant("nccl", default=True, description="Use NCCL", when="+cuda platform=cray") variant("nccl", default=True, description="Use NCCL", when="+rocm platform=linux") variant("nccl", default=True, description="Use NCCL", when="+rocm platform=cray") - variant("nnpack", default=True, description="Use NNPACK") + # Requires AVX2: https://discuss.pytorch.org/t/107518 + variant("nnpack", default=True, description="Use NNPACK", when="target=x86_64_v3:") variant("numa", default=True, description="Use NUMA", when="platform=linux") variant("numa", default=True, description="Use NUMA", when="platform=cray") variant("numpy", default=True, description="Use NumPy") |