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

py-torch: add note about MPS variant (#34018)

Browse source
This commit is contained in:
Adam J. Stewart 2022-11-19 19:10:06 -06:00 committed by GitHub
parent 632b36ab5d
commit 93ce943301
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -66,7 +66,7 @@ class PyTorch(PythonPackage, CudaPackage, ROCmPackage):
variant(
"mps",
default=is_darwin and macos_version() >= Version("12.3"),
description="Use MPS for macOS build",
description="Use MPS for macOS build (requires full Xcode suite)",
when="@1.12: platform=darwin",
)
variant("nccl", default=True, description="Use NCCL", when="+cuda platform=linux")