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

fix: mockgpu cuda timing (#10343)

Browse Source
This commit is contained in:
wozeparrot
2025-05-15 14:14:14 -07:00
committed by GitHub
parent a825608dc2
commit f59ecf2116
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/mockgpu/cuda/cuda.py
View File

@@ -100,7 +100,7 @@ def cuEventSynchronize(hEvent) -> int:
def cuEventElapsedTime(pMilliseconds, hStart, hEnd) -> int:
if hStart.value not in cuda_state.events or hEnd.value not in cuda_state.events:
return orig_cuda.CUDA_ERROR_INVALID_VALUE
elapsed = (cuda_state.events[hEnd.value] - cuda_state.events[hStart.value]) * 1000
elapsed = (cuda_state.events[hEnd.value] - cuda_state.events[hStart.value]) * 1e-6
pMilliseconds._obj.value = elapsed
return orig_cuda.CUDA_SUCCESS