functorch icon indicating copy to clipboard operation
functorch copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

MPS Support

Open Mahnoo06 opened this issue 3 years ago • 1 comments

Hello, I am using functorch for my project. It's a good substitute for Jax and makes life easy. However it does not have GPU support for M1 chips which is a time-constraint for me because I have an M1-based system. Now that PyTorch support M1-GPU I think functorch should also make this feature available.

Mahnoo06 avatar Aug 11 '22 15:08 Mahnoo06

Hey @Mahnoo06, MPS support is a "prototype" feature in PyTorch so we haven't actively tried to support it yet. That being said I know M1 machines are popular, so we'll try to enable this.

zou3519 avatar Aug 11 '22 19:08 zou3519