tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 23:46:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

nv: sync after gpfifo setup (#11209)

Browse Source
This commit is contained in:
nimlgen
2025-07-13 14:40:11 +03:00
committed by GitHub
parent d90d837013
commit 55c54d9745
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/runtime/ops_nv.py
+5 -5
View File
@@ -580,13 +580,13 @@ class NVDevice(HCQCompiled[NVSignal]):
self.shared_mem_window, self.local_mem_window = 0x729400000000, 0x729300000000
NVComputeQueue().setup(compute_class=self.iface.compute_class, local_mem_window=self.local_mem_window, shared_mem_window=self.shared_mem_window) \
.signal(self.timeline_signal, self.timeline_value).submit(self)
.signal(self.timeline_signal, self.next_timeline()).submit(self)
cast(NVCopyQueue, NVCopyQueue().wait(self.timeline_signal, self.timeline_value)) \
.setup(copy_class=self.iface.dma_class) \
.signal(self.timeline_signal, self.timeline_value + 1).submit(self)
NVCopyQueue().wait(self.timeline_signal, self.timeline_value - 1) \
.setup(copy_class=self.iface.dma_class) \
.signal(self.timeline_signal, self.next_timeline()).submit(self)
self.timeline_value += 2
self.synchronize()
def _ensure_has_local_memory(self, required):
if self.slm_per_thread >= required or ((maxlm:=getenv("NV_MAX_LOCAL_MEMORY_PER_THREAD")) > 0 and required >= maxlm): return