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