zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 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( variant(
"mps", "mps",
default=is_darwin and macos_version() >= Version("12.3"), 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", when="@1.12: platform=darwin",
) )
variant("nccl", default=True, description="Use NCCL", when="+cuda platform=linux") variant("nccl", default=True, description="Use NCCL", when="+cuda platform=linux")