github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 06:58:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

remu err handling (#5208)

Browse Source 
* add error handling

* use pre release

* minor

* works
This commit is contained in:
qazal
2024-06-28 13:15:18 +03:00
committed by GitHub
parent 7f7fa26e03
commit 3e56c8422c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
extra/mockgpu/amd/amdgpu.py
View File

@@ -21,7 +21,7 @@ WAIT_REG_MEM_FUNCTION_GEQ = 5 # >=
try:
remu = ctypes.CDLL("/usr/local/lib/libremu.so")
remu.run_asm.restype = ctypes.c_uint32
remu.run_asm.restype = ctypes.c_int32
remu.run_asm.argtypes = [ctypes.c_void_p, ctypes.c_uint32, ctypes.c_uint32, ctypes.c_uint32, ctypes.c_uint32, ctypes.c_uint32, ctypes.c_uint32, ctypes.c_uint32, ctypes.c_void_p]
except Exception: pass
@@ -152,7 +152,8 @@ class PM4Executor(AMDQueue):
if st <= prg_addr <= st+sz: prg_sz = sz - (prg_addr - st)
assert prg_sz > 0, "Invalid prg ptr (not found in mapped ranges)"
remu.run_asm(prg_addr, prg_sz, *gl, *lc, args_addr)
err = remu.run_asm(prg_addr, prg_sz, *gl, *lc, args_addr)
if err != 0: raise RuntimeError("remu does not support the new instruction introduced in this kernel")
def _exec_event_write(self, n):
assert n == 0