mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-02-19 01:53:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added cython bindings for commonmodel sigmoid

Browse Source 
old-commit-hash: 05993f50df
This commit is contained in:
mitchellgoffpc
2023-08-31 19:40:53 -07:00
parent 0107dfb391
commit 459e4fdb50
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
selfdrive/modeld/models/commonmodel.pxd
View File

@@ -12,6 +12,8 @@ cdef extern from "common/clutil.h":
cl_context cl_create_context(cl_device_id)
cdef extern from "selfdrive/modeld/models/commonmodel.h":
float sigmoid(float)
cppclass ModelFrame:
int buf_size
ModelFrame(cl_device_id, cl_context)

5
selfdrive/modeld/models/commonmodel_pyx.pyx
View File

@@ -8,7 +8,10 @@ from libc.string cimport memcpy
from cereal.visionipc.visionipc cimport cl_mem
from cereal.visionipc.visionipc_pyx cimport VisionBuf, CLContext as BaseCLContext
from .commonmodel cimport CL_DEVICE_TYPE_DEFAULT, cl_get_device_id, cl_create_context
from .commonmodel cimport mat3, ModelFrame as cppModelFrame
from .commonmodel cimport mat3, sigmoid as cppSigmoid, ModelFrame as cppModelFrame
def sigmoid(x):
return cppSigmoid(x)
cdef class CLContext(BaseCLContext):
def __cinit__(self):