jax
ead0c19c
- [pallas:mgpu] Allowed specifying `m_warps=` in the MMA_* layouts
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
15 hours ago
[pallas:mgpu] Allowed specifying `m_warps=` in the MMA_* layouts PiperOrigin-RevId: 969859436
References
#40160 - [pallas:mgpu] Allowed specifying `m_warps=` in the MMA_* layouts
Author
superbobry
Committer
Google-ML-Automation
Parents
5daac240
Loading
