Fix: Scaled Matmul at rocm 9.1 #738
shurale-nkn
wants to merge
1 commit into
rocm-jaxlib-v0.9.1
from
fix_scaled_matmul_dot_test_0.9.1
Add ROCm support for scaled matrix multiplication and scaled dot lowe…
6c9aea3a
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub