github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-26 07:18:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

nv fix size in SET_SEMAPHORE_A (#6213)

Browse Source
This commit is contained in:
nimlgen
2024-08-21 01:47:10 +03:00
committed by GitHub
parent ec52a09393
commit 89c4cffd86
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/runtime/ops_nv.py
View File

@@ -199,7 +199,7 @@ class NVCopyQueue(NVCommandQueue, HWCopyQueue):
if src is not None: self._patch(cmd_idx, offset=1, data=data64(src))
def _signal(self, signal, value=0):
self.q += [nvmethod(4, nv_gpu.NVC6B5_SET_SEMAPHORE_A, 4), *data64(signal.signal_addr), value, 4]
self.q += [nvmethod(4, nv_gpu.NVC6B5_SET_SEMAPHORE_A, 3), *data64(signal.signal_addr), value]
self.q += [nvmethod(4, nv_gpu.NVC6B5_LAUNCH_DMA, 1), 0x14]
def _update_signal(self, cmd_idx, signal=None, value=None):