mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-06 22:22:11 +08:00
Mitchell Goff
5033da5e27
* Added cython bindings for model runners and commonmodel * Removed cython language_level=3 * loop to set CXXFLAGS for both envs old-commit-hash: e2e39d100b32a6706e83c81a1ef3dcdd7ea4001f
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()
|