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

check on cuEventSynchronize (#3933)

Browse Source
This commit is contained in:
nimlgen
2024-03-26 16:14:38 +03:00
committed by GitHub
parent 778d17fbd3
commit 739f47eb0f
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_cuda.py
View File

@@ -43,7 +43,7 @@ def cu_time_execution(cb, enable=False) -> Optional[float]:
cuda.cuEventRecord(evs[0], None)
cb()
cuda.cuEventRecord(evs[1], None)
cuda.cuEventSynchronize(evs[1])
check(cuda.cuEventSynchronize(evs[1]))
cuda.cuEventElapsedTime(ctypes.byref(ret := ctypes.c_float()), evs[0], evs[1])
for ev in evs: cuda.cuEventDestroy_v2(ev)
return ret.value * 1e-3