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

get readable error from wait_check (#6965)

Browse Source 
Co-authored-by: Mesozoic Egg <mesozoic.egg@proton.me>
This commit is contained in:
mesozoic-egg
2024-10-09 22:28:58 +08:00
committed by GitHub
parent 20d3c2d113
commit 0e8bcda07e
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_metal.py
View File

@@ -48,7 +48,7 @@ def to_struct(*t: int, _type: type = ctypes.c_ulong):
def wait_check(cbuf: Any):
msg(cbuf, "waitUntilCompleted")
if (error := cast(int, msg(cbuf, "error", restype=ctypes.c_ulong))) != 0: raise RuntimeError(error)
error_check(msg(cbuf, "error", restype=objc_instance))
def elapsed_time(cbuf: objc_id):
return cast(float, msg(cbuf, "GPUEndTime", restype=ctypes.c_double)) - cast(float, msg(cbuf, "GPUStartTime", restype=ctypes.c_double))