Skip to content

Platform check incorrect on M1 pro device #13946

@adamgayoso

Description

@adamgayoso

Excited to see this get in!

https://github.com/Lightning-AI/lightning/blob/0f6caffa577e2c8d9602dc8fba37c13e128b2120/src/pytorch_lightning/accelerators/mps.py#L27

On my device I see this:

In [5]: import platform
   ...: 

In [6]: platform.platform()
Out[6]: 'macOS-12.3.1-arm64-arm-64bit'

so _MPS_AVAILABLE is incorrectly false

cc @tchaton @rohitgr7 @akihironitta @justusschock

Metadata

Metadata

Assignees

No one assigned

    Labels

    accelerator: mpsApple Silicon GPUbugSomething isn't workingplGeneric label for PyTorch Lightning packagepriority: 0High priority task

    Type

    No type

    Projects

    No projects

    Milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions