Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Mosaic GPU] Use events as the default profiling method
JAX still supports old CUDA versions such as 12.0, where CUPTI leaks memory. PiperOrigin-RevId: 705459909
- Loading branch information