mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
27 lines
641 B
Python
27 lines
641 B
Python
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
from datetime import datetime
|
||
|
|
import cProfile, pstats, io
|
||
|
|
|
||
|
|
def timestamp():
|
||
|
|
return datetime.now().timestamp()
|
||
|
|
|
||
|
|
class Profiler(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.pr = cProfile.Profile()
|
||
|
|
pass
|
||
|
|
|
||
|
|
def __enter__(self):
|
||
|
|
self.pr.enable()
|
||
|
|
|
||
|
|
def __exit__(self ,type, value, traceback):
|
||
|
|
self.pr.disable()
|
||
|
|
s = io.StringIO()
|
||
|
|
sortby = 'cumtime'
|
||
|
|
ps = pstats.Stats(self.pr, stream=s).sort_stats(sortby)
|
||
|
|
ps.print_stats(.2)
|
||
|
|
print(s.getvalue())
|