From 2b652605a08178e539d7ff7ab7e009b4da29bd46 Mon Sep 17 00:00:00 2001
From: Comma Device <device@comma.ai>
Date: Mon, 21 Feb 2022 12:08:52 -0800
Subject: [PATCH] thneed: a few asserts in the optimizer for cases it will fail

---
 selfdrive/modeld/thneed/optimizer.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/selfdrive/modeld/thneed/optimizer.cc b/selfdrive/modeld/thneed/optimizer.cc
index d5e6ce063c..3c7c41873d 100644
--- a/selfdrive/modeld/thneed/optimizer.cc
+++ b/selfdrive/modeld/thneed/optimizer.cc
@@ -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());