From 6c9d079cfb11a70927706f960fba39d77e2fa96b Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sun, 12 Mar 2023 00:03:27 -0700 Subject: [PATCH] py-torch: NNPACK requires AVX2 (#35994) --- var/spack/repos/builtin/packages/py-torch/package.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/var/spack/repos/builtin/packages/py-torch/package.py b/var/spack/repos/builtin/packages/py-torch/package.py index 6e022ad70fa..42379ac6f4b 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")