jax
2a828d5d - Merge pull request #23467 from abhinavgoel95:patch-4

Commit

1 year ago

Merge pull request #23467 from abhinavgoel95:patch-4 PiperOrigin-RevId: 685872800

References

#23467 - Revert to nvtx_kern_sum to get correct profiling info and adding post processing logic

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading