From 89c4cffd86e7e08142aaba121643d46b7f38c3fe Mon Sep 17 00:00:00 2001 From: nimlgen <138685161+nimlgen@users.noreply.github.com> Date: Wed, 21 Aug 2024 01:47:10 +0300 Subject: [PATCH] nv fix size in SET_SEMAPHORE_A (#6213) --- tinygrad/runtime/ops_nv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/runtime/ops_nv.py b/tinygrad/runtime/ops_nv.py index ba58e26ab0..38c9072faa 100644 --- a/tinygrad/runtime/ops_nv.py +++ b/tinygrad/runtime/ops_nv.py @@ -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):