Added cython bindings for commonmodel sigmoid

pull/29747/head
mitchellgoffpc 2 years ago
parent 75081b2e54
commit 05993f50df
  1. 2
      selfdrive/modeld/models/commonmodel.pxd
  2. 5
      selfdrive/modeld/models/commonmodel_pyx.pyx

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

@ -8,7 +8,10 @@ from libc.string cimport memcpy
from cereal.visionipc.visionipc cimport cl_mem from cereal.visionipc.visionipc cimport cl_mem
from cereal.visionipc.visionipc_pyx cimport VisionBuf, CLContext as BaseCLContext 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 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): cdef class CLContext(BaseCLContext):
def __cinit__(self): def __cinit__(self):

Loading…
Cancel
Save