[apps] Fix dropout kernel

pulp-platform · Jun 19, 2024 · 6f13393 · 6f13393
1 parent ae65e91
commit 6f13393
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/apps/dropout/kernel/dropout.h b/apps/dropout/kernel/dropout.h
@@ -27,8 +27,6 @@
 // Use asm, by default
 #undef INTRINSICS
 
-#include "runtime.h"
-
 #ifndef SPIKE
 #include "printf.h"
 #endif

diff --git a/apps/dropout/main.c b/apps/dropout/main.c
@@ -23,6 +23,17 @@
 
 #include "kernel/dropout.h"
 
+#include <stdint.h>
+#include <string.h>
+
+#include "runtime.h"
+
+#ifndef SPIKE
+#include "printf.h"
+#else
+#include <stdio.h>
+#endif
+
 extern const unsigned int N;
 extern const float SCALE;
 extern const float I[] __attribute__((aligned(4 * NR_LANES)));