mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-22 22:53:54 +08:00
e2e39d100b
* Added cython bindings for model runners and commonmodel * Removed cython language_level=3 * loop to set CXXFLAGS for both envs
24 lines
685 B
Cython
24 lines
685 B
Cython
# 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":
|
|
cppclass ModelFrame:
|
|
int buf_size
|
|
ModelFrame(cl_device_id, cl_context)
|
|
float * prepare(cl_mem, int, int, int, int, mat3, cl_mem*)
|
|
|
|
cdef extern from "selfdrive/modeld/runners/runmodel.h":
|
|
cdef int USE_CPU_RUNTIME
|
|
cdef int USE_GPU_RUNTIME
|
|
cdef int USE_DSP_RUNTIME
|