Fix MPS availability check (#13947)

Co-authored-by: Rohit Gupta <[email protected]>
Lightning-AI · Aug 2, 2022 · af07e75 · af07e75
1 parent d8e5e7f
commit af07e75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pytorch_lightning/accelerators/mps.py b/src/pytorch_lightning/accelerators/mps.py
@@ -24,7 +24,7 @@
 
 # For using the `MPSAccelerator`, user's machine should have `torch>=1.12`, Metal programming framework and
 # the ARM-based Apple Silicon processors.
-_MPS_AVAILABLE = _TORCH_GREATER_EQUAL_1_12 and torch.backends.mps.is_available() and platform.platform() == "arm"
+_MPS_AVAILABLE = _TORCH_GREATER_EQUAL_1_12 and torch.backends.mps.is_available() and platform.processor() == "arm"
 
 
 class MPSAccelerator(Accelerator):