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

2026-06-25 07:52:03 +08:00 · 2022-02-21 12:08:52 -08:00
parent 724bab8ef3
commit 2b652605a0
1 changed files with 5 additions and 0 deletions
@@ -173,6 +173,11 @@ int Thneed::optimize() {

          if (lastout == in) {
            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));

            cl_mem tmp = make_image_like(context, *(cl_mem *)lastout.data());