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
Chris Jones 38d52a19ef [mosaic_gpu] Force flush all cupti activity, then unsubscribe. 
With default flushing, it is possible for events to be missed. We should only unsubscribe after we are finished with cupti.

PiperOrigin-RevId: 737939327
2025-03-18 03:35:03 -07:00
..
dialect
[Mosaic] Allow part of x2 int casts.
2025-03-13 18:57:36 -07:00
gpu
[mosaic_gpu] Force flush all cupti activity, then unsubscribe.
2025-03-18 03:35:03 -07: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