Support for MPS (Mac GPU) when using torch backend.

wesselb / lab

A generic interface for linear algebra backends

MIT License

68 stars 5 forks source link

Support for MPS (Mac GPU) when using torch backend. #12

Open magnusross opened 1 year ago

magnusross commented 1 year ago

Hey Wessel!

I am just wondering if it might be possible to add support for the MPS backend in PyTorch as an alternative to Cuda, which would allow for the use of lab on the GPU on Mac.

It is hopefully quite a simple change I think, I have tried an initial implementation #11