jax
18326abe - [mosaic_gpu] Don't time the warmup step in cupti profiler.

Commit

1 year ago

[mosaic_gpu] Don't time the warmup step in cupti profiler. Initializing and finalizing cupti has an overhead. PiperOrigin-RevId: 738725435

References

Author

chr1sj0nes

Committer

Google-ML-Automation

Parents