jax
Fix: Scaled Matmul at rocm
#736
Closed

Fix: Scaled Matmul at rocm #736

shurale-nkn wants to merge 1 commit into amd-main from fix_dot_scale
shurale-nkn
shurale-nkn _scaled_matmul lowering for rocm
3346ee4f
shurale-nkn shurale-nkn closed this 225 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone