jax
246853b1
- Add ROCm support for scaled matrix multiplication and scaled dot lowering.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
25 days ago
Add ROCm support for scaled matrix multiplication and scaled dot lowering.
References
fix_scaled_matmul_dot_test
#35995 - [ROCm] Add lowering for ScaledMatmul, ScaledDot
Author
shurale-nkn
Committer
shurale-nkn
Parents
3ad2cd96
Loading