mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/mosaic
History
Sergei Lebedev 928caf83ee [pallas:mosaic_gpu] copy_smem_to_gmem now allows skipping cp.async.commit_group 
This feature is necessary to fix the SMEM->GMEM waiting behavior in
`emit_pipeline`, which used a pessimistic condition prior to this change,
since every copy was its own commit group.

PiperOrigin-RevId: 734553668
2025-03-07 07:43:54 -08:00
..
dialect
[pallas:mosaic_gpu] copy_smem_to_gmem now allows skipping cp.async.commit_group
2025-03-07 07:43:54 -08:00
gpu
fix member access to packed CUDA struct
2025-02-24 08:03:07 +11:00
python
Fix OSS build for the Mosaic GPU dialect
2025-01-02 15:55:03 +00:00
BUILD
[Mosaic] Rename dep name.
2025-03-03 11:01:25 -08:00
serde.cc
[mosaic] Extracted serialization pass traversal logic into a reusable function
2025-01-16 02:58:06 -08:00
serde.h
[mosaic] Extracted serialization pass traversal logic into a reusable function
2025-01-16 02:58:06 -08:00