diff --git a/ane/2_compile/.gitignore b/ane/2_compile/.gitignore index c26071cfa5..496fb1079e 100644 --- a/ane/2_compile/.gitignore +++ b/ane/2_compile/.gitignore @@ -1,2 +1,3 @@ *.hwx anecompiler.swap.* +context_switch_log.txt diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py index 7e8007c194..973e775823 100644 --- a/tinygrad/tensor.py +++ b/tinygrad/tensor.py @@ -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