pull/31506/head
Adeeb Shihadeh 1 year ago
parent 04e5eb32c2
commit e6ab7e45fb
  1. 6
      selfdrive/modeld/thneed/serialize.cc

@ -11,7 +11,7 @@ using namespace json11;
extern map<cl_program, string> g_program_source; extern map<cl_program, string> g_program_source;
void Thneed::load(const char *filename) { void Thneed::load(const char *filename) {
LOGD("Thneed::load: loading from %s\n", filename); LOGI("Thneed::load: loading from %s\n", filename);
string buf = util::read_file(filename); string buf = util::read_file(filename);
int jsz = *(int *)buf.data(); int jsz = *(int *)buf.data();
@ -96,7 +96,7 @@ void Thneed::load(const char *filename) {
cl_mem aa = real_mem[*(cl_mem*)(mobj["buffer_id"].string_value().data())]; cl_mem aa = real_mem[*(cl_mem*)(mobj["buffer_id"].string_value().data())];
input_clmem.push_back(aa); input_clmem.push_back(aa);
input_sizes.push_back(sz); input_sizes.push_back(sz);
LOGD("Thneed::load: adding input %s with size %d\n", mobj["name"].string_value().data(), sz); LOGI("Thneed::load: adding input %s with size %d\n", mobj["name"].string_value().data(), sz);
cl_int cl_err; cl_int cl_err;
void *ret = clEnqueueMapBuffer(command_queue, aa, CL_TRUE, CL_MAP_WRITE, 0, sz, 0, NULL, NULL, &cl_err); void *ret = clEnqueueMapBuffer(command_queue, aa, CL_TRUE, CL_MAP_WRITE, 0, sz, 0, NULL, NULL, &cl_err);
@ -108,7 +108,7 @@ void Thneed::load(const char *filename) {
for (auto &obj : jdat["outputs"].array_items()) { for (auto &obj : jdat["outputs"].array_items()) {
auto mobj = obj.object_items(); auto mobj = obj.object_items();
int sz = mobj["size"].int_value(); int sz = mobj["size"].int_value();
LOGD("Thneed::save: adding output with size %d\n", sz); LOGI("Thneed::save: adding output with size %d\n", sz);
// TODO: support multiple outputs // TODO: support multiple outputs
output = real_mem[*(cl_mem*)(mobj["buffer_id"].string_value().data())]; output = real_mem[*(cl_mem*)(mobj["buffer_id"].string_value().data())];
assert(output != NULL); assert(output != NULL);

Loading…
Cancel
Save