New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Does this rep surport tensorcore? #5

Open

Rane2021 opened this issue Apr 24, 2024 · 2 comments

Rane2021 commented Apr 24, 2024

very nice work! i want to know does this support tensor core or AMD GPU?

Iron-Bound commented Apr 25, 2024

No as it requires Cuda.
For AMD GPU's try triton https://github.com/ROCm/triton/tree/triton-mlir/python/perf-kernels

Repeerc commented Jul 2, 2024

I have implemented the v2 minimal version on AMD GPU:
https://github.com/Repeerc/flash-attention-v2-RDNA3-minimal
and changing rocWMMA to CUDA wmma should work on Nvidia's tensor core

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment