github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-15 17:15:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

hcq exec no embedded signal (#5142)

Browse Source
This commit is contained in:
nimlgen
2024-07-04 13:29:21 +03:00
committed by GitHub
parent 0c3a35e5c2
commit 84b3e3bb6f
3 changed files with 29 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tinygrad/runtime/ops_amd.py
View File

@@ -104,7 +104,7 @@ class HWPM4Queue(HWQueue):
self._invalidate_cache()
return self._mark_command_end()
def exec(self, prg, kernargs, global_size:Tuple[int,int,int]=(1,1,1), local_size:Tuple[int,int,int]=(1,1,1), signal=None, signal_value=0):
def exec(self, prg, kernargs, global_size:Tuple[int,int,int]=(1,1,1), local_size:Tuple[int,int,int]=(1,1,1)):
self._invalidate_cache()
user_data = [*data64_le(kernargs)]
@@ -130,7 +130,6 @@ class HWPM4Queue(HWQueue):
self.q += [amd_gpu.PACKET3(amd_gpu.PACKET3_DISPATCH_DIRECT, 3), *global_size, CS_W32_EN | FORCE_START_AT_000 | COMPUTE_SHADER_EN]
self.q += [amd_gpu.PACKET3(amd_gpu.PACKET3_EVENT_WRITE, 0), amd_gpu.EVENT_TYPE(7) | amd_gpu.EVENT_INDEX(4)]
if signal is not None: self.signal(signal, signal_value)
return self._mark_command_end()
def update_exec(self, cmd_idx, global_size, local_size):