Skip to content

Commit af07e75

Browse files
Fix MPS availability check (#13947)
Co-authored-by: Rohit Gupta <[email protected]>
1 parent d8e5e7f commit af07e75

File tree

1 file changed

+1
-1
lines changed
  • src/pytorch_lightning/accelerators

1 file changed

+1
-1
lines changed

src/pytorch_lightning/accelerators/mps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424

2525
# For using the `MPSAccelerator`, user's machine should have `torch>=1.12`, Metal programming framework and
2626
# the ARM-based Apple Silicon processors.
27-
_MPS_AVAILABLE = _TORCH_GREATER_EQUAL_1_12 and torch.backends.mps.is_available() and platform.platform() == "arm"
27+
_MPS_AVAILABLE = _TORCH_GREATER_EQUAL_1_12 and torch.backends.mps.is_available() and platform.processor() == "arm"
2828

2929

3030
class MPSAccelerator(Accelerator):

0 commit comments

Comments
 (0)