mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-03-02 06:14:00 +08:00
7b5a4fbb03
* squash * bump tg * bump tg * debump tinygrad * bump tinygrad * bump tg * Skip init iteration * fixes * cleanups * skip first test sample * typos * linter unhappy * update cpu usage * OPENCL just zeros for now * imports * Try printing * Runs again, but slower * unused import * Allow more buffer with tg and all on gpu * bump tinygrad * seems ok * stricter timings for driving looser for dm * try llvm * check nvidia * More timeout for now * make test pass * Revert "try llvm" This reverts commit ef136e478320101fea262bae3579e558da991902. * small fixes * whitespace * revert test timeout * No model runners * Always CPU always fast * No onnx runtime GPU * more cores * cleanup * Is this faster * Is this faster * at least runs * FP32 is faster than 16 * fix deps * whitespace * comment --------- Co-authored-by: Adeeb Shihadeh <adeebshihadeh@gmail.com>
20 lines
582 B
Cython
20 lines
582 B
Cython
# distutils: language = c++
|
|
|
|
from msgq.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)
|
|
cl_mem * prepare(cl_mem, int, int, int, int, mat3)
|
|
unsigned char * buffer_from_cl(cl_mem*);
|