mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-04 13:02:09 +08:00
Harald Schäfer
f6885dcbec
* Revert "dmonitoringmodeld: use cl transform (#34235)" This reverts commit684b0b9d4d. * Revert "load model before calling convert_fp16_to_fp32" This reverts commit31606a7d15. * Revert "bump tinygrad" This reverts commit44f58ff758. * Revert "Tinygrad runner (#34171)" This reverts commit7b5a4fbb03. * Allow init buffer * typo
33 lines
718 B
C++
33 lines
718 B
C++
#include "selfdrive/modeld/thneed/thneed.h"
|
|
|
|
#include <cassert>
|
|
|
|
#include "common/clutil.h"
|
|
#include "common/timing.h"
|
|
|
|
Thneed::Thneed(bool do_clinit, cl_context _context) {
|
|
context = _context;
|
|
if (do_clinit) clinit();
|
|
char *thneed_debug_env = getenv("THNEED_DEBUG");
|
|
debug = (thneed_debug_env != NULL) ? atoi(thneed_debug_env) : 0;
|
|
}
|
|
|
|
void Thneed::execute(float **finputs, float *foutput, bool slow) {
|
|
uint64_t tb, te;
|
|
if (debug >= 1) tb = nanos_since_boot();
|
|
|
|
// ****** copy inputs
|
|
copy_inputs(finputs);
|
|
|
|
// ****** run commands
|
|
clexec();
|
|
|
|
// ****** copy outputs
|
|
copy_output(foutput);
|
|
|
|
if (debug >= 1) {
|
|
te = nanos_since_boot();
|
|
printf("model exec in %lu us\n", (te-tb)/1000);
|
|
}
|
|
}
|