mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix GPU profiling

Browse Source
This commit is contained in:
George Hotz
2020-12-07 20:03:28 -08:00
parent b2eca6d45f
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