ROCm / triton

Development repository for the Triton language and compiler
MIT License
96 stars 29 forks source link

Softmax kernel #634

Closed rahulbatra85 closed 2 months ago

brunomazzottiamd commented 2 months ago

I think https://github.com/ROCm/triton/pull/633#issuecomment-2332406892 and https://github.com/ROCm/triton/pull/633#issuecomment-2332425921 is also applicable to this PR.

rahulbatra85 commented 2 months ago

I think #633 (comment) and #633 (comment) is also applicable to this PR.

Fixed

rahulbatra85 commented 2 months ago

@rahulbatra85, can you please update the README file as you did in #633?

Besides this, I have nothing to add. IMHO you did a great job!

Fixed