torchmetrics icon indicating copy to clipboard operation
torchmetrics copied to clipboard

Published 20 hours ago • verified publisher icon Lightning-AI

Investigate MPS support

Open justusschock opened this issue 2 years ago • 3 comments

🚀 Feature

We need to investigate support for the new 'mps' backend. I assume this will not be working, as many operations are not yet supported. However, we should give it a try and currently raise an error if necessary.

For the record: https://github.com/PyTorchLightning/pytorch-lightning/pull/13123

justusschock avatar May 24 '22 16:05 justusschock

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

stale[bot] avatar Jul 31 '22 13:07 stale[bot]

I am facing an issue on MPS.

saikatkumardey avatar Aug 11 '22 11:08 saikatkumardey

So far we don't officially support MPS for metrics. Most of the stuff should work, however. For the failing ops, we can only wait for official support from the pytorch side. Did you try tu run with PYTORCH_ENABLE_MPS_FALLBACK=1?

justusschock avatar Aug 11 '22 13:08 justusschock

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

stale[bot] avatar Oct 29 '22 10:10 stale[bot]