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

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
commit fe26d3fefe
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tinygrad/runtime/ops_nv.py
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