mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic
History
Adam Paszke 1ea5b26524 [Mosaic][Pallas] Extend support for slicing and concatenation 
... and squeezing. Those are useful when implementing batched matrix multiplication kernels.

PiperOrigin-RevId: 563381169
2023-09-07 03:45:01 -07:00
..
dialect/tpu
[Mosaic][Pallas] Extend support for slicing and concatenation
2023-09-07 03:45:01 -07:00
python
[Mosaic][Pallas] Extend support for slicing and concatenation
2023-09-07 03:45:01 -07:00
BUILD
[MOSAIC] apply_vector_layout C++ rewrite (1) VectorLayout functions
2023-08-29 22:57:13 -07:00