Fix ne16 relu flag in the layer template

pulp-platform · Feb 1, 2024 · 5573413 · 5573413
1 parent fd5bdf9
commit 5573413
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...argets/PULP/GAP9_NE16/Templates/layer_templates/layer_L2_c_conv_ne16_multicore_template.c b/...argets/PULP/GAP9_NE16/Templates/layer_templates/layer_L2_c_conv_ne16_multicore_template.c
@@ -366,7 +366,7 @@ void ${func_name}(void *args) {
             &ne16_tasks[i],
             (ne16_quant_t) {
                 .shift_amount = ${out_shift},
-                .function = quantFunctionRelu, // todo: un-hardcode it
+                .function = ${"quantFunctionIdentity" if node.min < 0 else "quantFunctionRelu"},
                 .flag_rounding = ne16TaskFlagFalse
             }, (ne16_norm_t) {
                 .mode  = ${"normMode32Bit" if act_dim_bit == 32 else "normMode8Bit" if act_dim_bit == 8 else ""},