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
gulsumgudukbay
approved these changes
on 2026-04-20
Login to write a write a comment.
Login via GitHub