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 d8e5e7f commit af07e75Copy full SHA for af07e75
src/pytorch_lightning/accelerators/mps.py
@@ -24,7 +24,7 @@
24
25
# For using the `MPSAccelerator`, user's machine should have `torch>=1.12`, Metal programming framework and
26
# the ARM-based Apple Silicon processors.
27
-_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"
28
29
30
class MPSAccelerator(Accelerator):
0 commit comments