jax
908ff49e - [mosaic_gpu] Warmup the kernel when doing CUPTI based profiling

Commit

336 days ago

[mosaic_gpu] Warmup the kernel when doing CUPTI based profiling Closes #26144. PiperOrigin-RevId: 730159757

References

Author

superbobry

Committer

Google-ML-Automation

Parents