mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

fix GPU profiling

Browse source
This commit is contained in:
George Hotz 2020-12-07 20:03:28 -08:00
commit 3aac9aefce
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ane/2_compile/.gitignore vendored
View file

@ -1,2 +1,3 @@
*.hwx
anecompiler.swap.*
context_switch_log.txt

2
tinygrad/tensor.py
View file

@ -23,6 +23,8 @@ class ProfileOp:
if DEBUG: self.st = time.time()
def __exit__(self, *junk):
if DEBUG:
if cl_queue is not None:
cl_queue.finish()
et = (time.time()-self.st)*1000.
debug_counts[self.name] += 1
debug_times[self.name] += et