Skip to content

Some torch functions not accepted on MPS #1177

@Andrewwango

Description

@Andrewwango

Environment

Apple M1

Describe the bug

Some torch functions e.g. torch.logspace are unsupported on Apple silicon, leading NotImplementedError to errors. Since torch.logspace in particular is used in DPIR, the quickstart therefore does not run end-to-end on Apple M1.

Bug uncovered during ICMS tutorial.

For now, we should include a warning in deepinv.utils.get_device when M1 is selected @samuro95 warning the user of potential problems.

Steps to reproduce

Run the quickstart DPIR example on Apple M1.

Metadata

Metadata

Assignees

No one assigned

    Labels

    priority: lowNice to have, non-urgent issue or PR.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions