NVIDIA / modulus-launch

Repo of optimized training recipes for accelerating PyTorch workflows of AI driven surrogates for physical systems
Apache License 2.0
56 stars 27 forks source link

🚀[FEA]: Add timing to `LaunchLogger.log_minibatch` #119

Open akshaysubr opened 1 year ago

akshaysubr commented 1 year ago

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Medium

Please provide a clear description of problem you would like to solve.

It would be very convenient to add some cuda event based timing code to log_minibatch in the LaunchLogger. This will allow users to very easily get performance numbers without any extra work.

Describe any alternatives you have considered

No response