wip: add neureka support to test app

pulp-platform · Jan 14, 2024 · bf21d0c · bf21d0c
1 parent 499fed2
commit bf21d0c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/test/app/Makefile b/test/app/Makefile
@@ -40,6 +40,8 @@ INC_DIRS += $(ACC_DIR)/hal $(ACC_DIR)/gvsoc $(ACC_DIR)/bsp
 INC_DIRS += gen/inc
 
 INC_FLAGS += $(addprefix -I,$(INC_DIRS))
+APP_CFLAGS += $(INC_FLAGS)
+
 
 # Source files
 
@@ -58,7 +60,10 @@ APP_SRCS += $(wildcard gen/src/*.c)
 
 # Flags
 
-APP_CFLAGS += $(INC_FLAGS) -O2 -w -Wall -Werror -flto
+ACCELERATOR_UPPERCASE := $(shell echo $(ACCELERATOR) | tr [:lower:] [:upper:])
+APP_CFLAGS += -DNNX_ACCELERATOR=$(ACCELERATOR) -DNNX_$(ACCELERATOR_UPPERCASE)
+
+APP_CFLAGS += -O2 -w -Wall -Werror -flto
 APP_LDFLAGS += -flto
 
 include $(RULES_DIR)/pmsis_rules.mk
diff --git a/test/app/src/nnx_layer.c b/test/app/src/nnx_layer.c
@@ -19,12 +19,21 @@
  */
 
 #include "nnx_layer.h"
+#include <pmsis.h>
+
+#ifdef NNX_NE16
 #include "ne16.h"
 #include "ne16_gvsoc.h"
 #include "ne16_pulp_bsp.h"
 #include "ne16_task.h"
 #include "pulp_nnx_ne16.h"
-#include <pmsis.h>
+#elif defined NNX_NEUREKA
+#include "neureka.h"
+#include "neureka_gvsoc.h"
+#include "neureka_pulp_bsp.h"
+#include "neureka_task.h"
+#include "pulp_nnx_neureka.h"
+#endif
 
 // Generated headers
 #include "bias.h"