jax
10ac6b7e - [Mosaic GPU] Add support for tiled swizzle=16 (i.e. no swizzle) loads and stores

Commit

1 year ago

[Mosaic GPU] Add support for tiled swizzle=16 (i.e. no swizzle) loads and stores The tiling still makes it possible to do it without bank conflicts. PiperOrigin-RevId: 721701635

References

#219 - CI: 02/03/25 upstream sync

Author

apaszke

Committer

Google-ML-Automation

Parents

efacec4c

jax 10ac6b7e - [Mosaic GPU] Add support for tiled swizzle=16 (i.e. no swizzle) loads and stores

jax
10ac6b7e - [Mosaic GPU] Add support for tiled swizzle=16 (i.e. no swizzle) loads and stores