search

SHI-Labs / NATTEN

Neighborhood Attention Extension. Bringing attention to a neighborhood near you!
https://shi-labs.com/natten/
Other
339 stars 25 forks source link

MPS support #102

Open ronensession opened 6 months ago

ronensession commented 6 months ago

Hi there, Im currently using a library that uses Natten's features, currently working locally on my Mac using the new M3 chip. When will there be support for mps on Natten? or is there a fix around that I could do to make it take advantage of the gpu that is on my chip?

alihassanijr commented 6 months ago

Hello,

Unfortunately we don't have any Metal kernels yet. In theory we could port our naive kernels to Metal, but I'm afraid it's not something that we can do in the short term. It is however something that we have had on our list, and we'll now prioritize it.

pkiers commented 4 months ago

Is there any update on this?

filtercodes commented 3 months ago

How is this going? I think there is a lot of people interested in this to happen, and would be willing to help on certain tasks. Is there a branch dedicated for MPS kernel update?