github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-30 17:28:24 -05:00
Code Issues Packages Projects Releases Wiki Activity

nv faster signal on dma queue (#4789)

Browse Source
This commit is contained in:
nimlgen
2024-06-02 21:47:24 +03:00
committed by GitHub
parent 707099487a
commit e8b5f2040d
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tinygrad/runtime/ops_nv.py
View File

@@ -186,6 +186,12 @@ class HWCopyQueue(HWQueue):
self.next_cmd_index += 1
return self
def signal(self, signal, value=0):
self.q += [nvmethod(4, nv_gpu.NVC6B5_SET_SEMAPHORE_A, 4), *nvdata64(ctypes.addressof(from_mv(signal))), value, 4]
self.q += [nvmethod(4, nv_gpu.NVC6B5_LAUNCH_DMA, 1), 0x14]
self.next_cmd_index += 1
return self
def submit(self, dev:NVDevice):
if len(self.q) == 0: return
dev.dma_put_value = self._submit(dev, dev.dma_gpu_ring, dev.dma_put_value, dev.dma_gpfifo_entries,