HazyResearch/ThunderKittens

When will ThunderKittens support AMD GPUs, specifically the W7900?

Opened this issue · 1 comments

Firstly, thank you for the amazing work on this project. I am very interested in using ThunderKittens with my AMD GPU, specifically the W7900. Could you please let me know if there are any plans to support AMD GPUs in the near future? If so, is there an estimated timeline for when this support might be available?

Hello team,

I wanted to also check if there is specific timeline for AMD ROCm implementation? I have 2xAMD MI60 (GCN5 architecture). I would be great to see a general implementation of ThunderKittens AMD GPUs so that everyone could benefit from the library.

@benjaminfspector

Please let us know.
Thanks!