aboutsummaryrefslogtreecommitdiff
path: root/src/ethosu_driver.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/ethosu_driver.c')
-rw-r--r--src/ethosu_driver.c8
1 files changed, 6 insertions, 2 deletions
diff --git a/src/ethosu_driver.c b/src/ethosu_driver.c
index 5128455..919fc11 100644
--- a/src/ethosu_driver.c
+++ b/src/ethosu_driver.c
@@ -23,10 +23,14 @@
#include "ethosu_device.h"
#include "ethosu_log.h"
-#ifdef ETHOSU55
+#if defined(ETHOSU55)
#include "ethosu_config_u55.h"
-#else
+#elif defined(ETHOSU65)
#include "ethosu_config_u65.h"
+#elif defined(ETHOSU85)
+#include "ethosu_config_u85.h"
+#else
+#error Missing device type macro
#endif
#include <assert.h>