github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

Define ProfileOp class once (#131)

Browse Source 
* define `ProfileOp` class once

* clean `ProfileOp` class

* removed `else: pass`
This commit is contained in:
Mufeed VH
2020-11-25 09:09:13 +05:30
committed by GitHub
parent 03994e0011
commit 0bbf66627c
1 changed files with 9 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
tinygrad/tensor.py
View File

@@ -19,25 +19,19 @@ if DEBUG:
for name, _ in sorted(debug_times.items(), key=lambda x: -x[1]):
print("%20s : %3d %10.2f ms" % (name, debug_counts[name], debug_times[name]))
atexit.register(print_debug_exit)
class ProfileOp:
def __init__(self, name, x, backward=False):
self.name = ("back_" if backward else "")+name
self.x = x
def __enter__(self):
self.st = time.time()
def __exit__(self, *junk):
class ProfileOp:
def __init__(self, name, x, backward=False):
self.name = ("back_" if backward else "")+name
self.x = x
def __enter__(self):
if DEBUG: self.st = time.time()
def __exit__(self, *junk):
if DEBUG:
et = (time.time()-self.st)*1000.
debug_counts[self.name] += 1
debug_times[self.name] += et
print("%20s : %7.2f ms %s" % (self.name, et, [y.shape for y in self.x]))
else:
class ProfileOp:
def __init__(self, name, x, backward=False):
pass
def __enter__(self):
pass
def __exit__(self, *junk):
pass
cl_ctx, cl_queue = None, None
def require_init_gpu():