Added cython bindings for commonmodel sigmoid

old-commit-hash: 05993f50df
beeps
mitchellgoffpc 2 years ago
parent 0107dfb391
commit 459e4fdb50
  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)
cdef extern from "selfdrive/modeld/models/commonmodel.h":
float sigmoid(float)
cppclass ModelFrame:
int buf_size
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_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):

Loading…
Cancel
Save