piiswrong commented on a change in pull request #8972: Profiling enhancements,
python API, vtune and chrome tracing objects, etc.
URL: https://github.com/apache/incubator-mxnet/pull/8972#discussion_r164209441
##########
File path: python/mxnet/profiler.py
##########
@@ -54,5 +60,280 @@ def profiler_set_state(state='stop'):
def dump_profile():
"""Dump profile and stop profiler. Use this to save profile
- in advance in case your program cannot exit normally."""
+ in advance in case your program cannot exit normally.
+ """
check_call(_LIB.MXDumpProfile())
+
+def dump_aggregate_stats(reset=False):
+ """Dump profile aggregate stats to console.
+ """
+ do_reset = 1 if reset is True else 0
+ check_call(_LIB.MXDumpAggregateProfileStats(int(do_reset)))
+
+def profiler_pause():
Review comment:
remove `profiler_`. Its superfluous given mx.profiler.
We might also want to alias the existing apis to remove `profiler_`
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
