mawei/dragonpilot
1
0
Fork 0
mirror of https://github.com/dragonpilot/dragonpilot.git synced 2026-06-21 05:52:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(modeld): Fix for unpredictable behavior in commonmodel_pyx.pyx (#32482)

Browse Source 
Avoid undefined behavior

Co-authored-by: 19igari <koki.igari@turing-motors.com>
This commit is contained in:
19igari
2024-05-22 10:41:04 +09:00
committed by GitHub
parent d234085496
commit 25cc8a96ef
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
selfdrive/modeld/models/commonmodel_pyx.pyx
+5 -1
View File
@@ -37,7 +37,11 @@ cdef class ModelFrame:
def prepare(self, VisionBuf buf, float[:] projection, CLMem output):
cdef mat3 cprojection
memcpy(cprojection.v, &projection[0], 9*sizeof(float))
cdef float * data = self.frame.prepare(buf.buf.buf_cl, buf.width, buf.height, buf.stride, buf.uv_offset, cprojection, output.mem)
cdef float * data
if output is None:
data = self.frame.prepare(buf.buf.buf_cl, buf.width, buf.height, buf.stride, buf.uv_offset, cprojection, NULL)
else:
data = self.frame.prepare(buf.buf.buf_cl, buf.width, buf.height, buf.stride, buf.uv_offset, cprojection, output.mem)
if not data:
return None
return np.asarray(<cnp.float32_t[:self.frame.buf_size]> data)