def test_profile_event(do_enable_profiler=True): def makeParams(): objects = tuple('foo' for _ in range(50)) template = ''.join('{%d}' % i for i in range(len(objects))) return template, objects def get_log(): template, objects = makeParams() logs = [] for _ in range(100000): logs.append(template.format(*objects)) return logs if do_enable_profiler is True: enable_profiler('test_profile_event.json') event = profiler.Event("test_profile_event") event.start() start = time.time() var = mx.nd.ones((1000, 500)) assert len(get_log()) == 100000 var.asnumpy() stop = time.time() event.stop() assert stop > start if do_enable_profiler is True: profiler.set_state('stop')
def test_profile_event(do_enable_profiler=True): def makeParams(): objects = tuple('foo' for _ in range(50)) template = ''.join('{%d}' % i for i in range(len(objects))) return template, objects def doLog(): template, objects = makeParams() for _ in range(100000): logging.info(template.format(*objects)) logging.basicConfig() if do_enable_profiler is True: enable_profiler('test_profile_event.json') event = profiler.Event("test_profile_event") event.start() start = time.time() var = mx.nd.ones((1000, 500)) doLog() var.asnumpy() stop = time.time() event.stop() print('run took: %.3f' % (stop - start)) if do_enable_profiler is True: profiler.set_state('stop')