mirror of
https://github.com/dragonpilot/dragonpilot.git
synced 2026-06-26 08:22:05 +08:00
Milan Medić
69fb3c2ed5
* Adding pre-commit hook for cython static analysis * Adding changes to cython files to pass static analysis * Revert "Adding changes to cython files to pass static analysis" This reverts commit 9a0eb733199abd9eef1eac3d024ef2760348d67c. * Adding ignore rule for indentation of 4 spaces (E111) * Fixes for cython-lint static analysis * Revert "Fixes for cython-lint static analysis" This reverts commit 972741735b2bdc73460d65a4d7ea167dfc0f4644. * Adding two new rules into ignore list (2 new lines after difinition of class) * Adding fixes for cython static analysis
38 lines
987 B
Cython
38 lines
987 B
Cython
# distutils: language = c++
|
|
# cython: c_string_encoding=ascii
|
|
|
|
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()
|