nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

nv sync before free for binded commands (#4759)

Browse Source 
* nv sync before free for binded commands

* shorter comment
This commit is contained in:
nimlgen
2024-05-28 20:49:29 +03:00
committed by GitHub
parent e614b7c696
commit fe26d3fefe
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_nv.py
+3 -1
View File
@@ -86,7 +86,9 @@ class NVCompiler(Compiler):
class HWQueue:
def __init__(self): self.q, self.binded_device, self.next_cmd_index = [], None, 0
def __del__(self):
if self.binded_device is not None: self.binded_device._gpu_free(self.hw_page)
if self.binded_device is not None:
self.binded_device.synchronize() # Synchronize to ensure the buffer is no longer in use.
self.binded_device._gpu_free(self.hw_page)
def ptr(self) -> int: return self.next_cmd_index