You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
37 lines
1005 B
37 lines
1005 B
# distutils: language = c++
|
|
# cython: c_string_encoding=ascii, language_level=3
|
|
|
|
from libcpp.string cimport string
|
|
|
|
from .runmodel cimport USE_CPU_RUNTIME, USE_GPU_RUNTIME, USE_DSP_RUNTIME
|
|
from selfdrive.modeld.models.commonmodel_pyx cimport CLMem
|
|
|
|
class Runtime:
|
|
CPU = USE_CPU_RUNTIME
|
|
GPU = USE_GPU_RUNTIME
|
|
DSP = USE_DSP_RUNTIME
|
|
|
|
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()
|
|
|