thneed: a few asserts in the optimizer for cases it will fail

old-commit-hash: 2b652605a0
taco
Comma Device 3 years ago
parent d4f372cfd7
commit 123fd2f6ee
  1. 5
      selfdrive/modeld/thneed/optimizer.cc

@ -173,6 +173,11 @@ int Thneed::optimize() {
if (lastout == in) { if (lastout == in) {
short neuron = *(int*)kq[i]->args[kq[i]->get_arg_num("neuron")].data(); short neuron = *(int*)kq[i]->args[kq[i]->get_arg_num("neuron")].data();
assert(neuron <= 5);
// ELU isn't supported in fc_Wtx
assert(!(kq[i-1]->name == "fc_Wtx" && neuron == 5));
kq[i-1]->args[kq[i-1]->get_arg_num("neuron")] = string((char *)&neuron, sizeof(neuron)); kq[i-1]->args[kq[i-1]->get_arg_num("neuron")] = string((char *)&neuron, sizeof(neuron));
cl_mem tmp = make_image_like(context, *(cl_mem *)lastout.data()); cl_mem tmp = make_image_like(context, *(cl_mem *)lastout.data());

Loading…
Cancel
Save