hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

py-torch: NNPACK requires AVX2 (#35994)

Browse source
This commit is contained in:
Adam J. Stewart 2023-03-12 00:03:27 -07:00 committed by GitHub
parent a741350e69
commit 6c9d079cfb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
var/spack/repos/builtin/packages/py-torch/package.py
View file

@ -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="+cuda platform=cray")
variant("nccl", default=True, description="Use NCCL", when="+rocm platform=linux") variant("nccl", default=True, description="Use NCCL", when="+rocm platform=linux")
variant("nccl", default=True, description="Use NCCL", when="+rocm platform=cray") 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=linux")
variant("numa", default=True, description="Use NUMA", when="platform=cray") variant("numa", default=True, description="Use NUMA", when="platform=cray")
variant("numpy", default=True, description="Use NumPy") variant("numpy", default=True, description="Use NumPy")