1234567891011121314151617181920 |
- # distutils: language = c++
- from cereal.visionipc.visionipc cimport cl_device_id, cl_context, cl_mem
- cdef extern from "common/mat.h":
- cdef struct mat3:
- float v[9]
- cdef extern from "common/clutil.h":
- cdef unsigned long CL_DEVICE_TYPE_DEFAULT
- cl_device_id cl_get_device_id(unsigned long)
- 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)
- float * prepare(cl_mem, int, int, int, int, mat3, cl_mem*)
|