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.c7
1 files changed, 6 insertions, 1 deletions
diff --git a/src/ethosu_driver.c b/src/ethosu_driver.c
index ee9d08c..11c51f6 100644
--- a/src/ethosu_driver.c
+++ b/src/ethosu_driver.c
@@ -21,10 +21,15 @@
******************************************************************************/
#include "ethosu_driver.h"
-#include "ethosu_config.h"
#include "ethosu_device.h"
#include "ethosu_log.h"
+#ifdef ETHOSU55
+#include "ethosu_config_u55.h"
+#else
+#include "ethosu_config_u65.h"
+#endif
+
#include <assert.h>
#include <cmsis_compiler.h>
#include <inttypes.h>