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')
