Skip to content

Commit ad1a70a

Browse files
committed
Fix: Simplify MPS availability check by removing platform dependency
1 parent ff64a92 commit ad1a70a

File tree

1 file changed

+1
-1
lines changed
  • src/lightning/fabric/accelerators

1 file changed

+1
-1
lines changed

src/lightning/fabric/accelerators/mps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def auto_device_count() -> int:
7272
def is_available() -> bool:
7373
"""MPS is only available on a machine with the ARM-based Apple Silicon processors."""
7474
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")
75+
return not mps_disabled and torch.backends.mps.is_available()
7676

7777
@classmethod
7878
@override

0 commit comments

Comments
 (0)