We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ff64a92 commit ad1a70aCopy full SHA for ad1a70a
src/lightning/fabric/accelerators/mps.py
@@ -72,7 +72,7 @@ def auto_device_count() -> int:
72
def is_available() -> bool:
73
"""MPS is only available on a machine with the ARM-based Apple Silicon processors."""
74
mps_disabled = os.getenv("DISABLE_MPS", "0") == "1"
75
- return not mps_disabled and torch.backends.mps.is_available() and platform.processor() in ("arm", "arm64")
+ return not mps_disabled and torch.backends.mps.is_available()
76
77
@classmethod
78
@override
0 commit comments