jax
e4a381c1
- [pallas:mgpu] Check that swizzle dim is not transposed in copy_smem_to_gmem()
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
304 days ago
[pallas:mgpu] Check that swizzle dim is not transposed in copy_smem_to_gmem() PiperOrigin-RevId: 743910324
References
#27666 - [pallas:mgpu] Check that swizzle dim is not transposed in copy_smem_to_gmem()
Author
cperivol
Committer
Google-ML-Automation
Parents
635805e9
Loading