mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-18 21:14:01 +08:00
e2e39d100b
* Added cython bindings for model runners and commonmodel * Removed cython language_level=3 * loop to set CXXFLAGS for both envs
33 lines
858 B
Cython
33 lines
858 B
Cython
# distutils: language = c++
|
|
# cython: c_string_encoding=ascii
|
|
|
|
from libcpp.string cimport string
|
|
from libc.string cimport memcpy
|
|
|
|
from selfdrive.modeld.models.commonmodel_pyx cimport CLMem
|
|
|
|
cdef class RunModel:
|
|
def __dealloc__(self):
|
|
del self.model
|
|
|
|
def addInput(self, string name, float[:] buffer):
|
|
if buffer is not None:
|
|
self.model.addInput(name, &buffer[0], len(buffer))
|
|
else:
|
|
self.model.addInput(name, NULL, 0)
|
|
|
|
def setInputBuffer(self, string name, float[:] buffer):
|
|
if buffer is not None:
|
|
self.model.setInputBuffer(name, &buffer[0], len(buffer))
|
|
else:
|
|
self.model.setInputBuffer(name, NULL, 0)
|
|
|
|
def getCLBuffer(self, string name):
|
|
cdef void * cl_buf = self.model.getCLBuffer(name)
|
|
if not cl_buf:
|
|
return None
|
|
return CLMem.create(cl_buf)
|
|
|
|
def execute(self):
|
|
self.model.execute()
|