diff options
Diffstat (limited to 'src/ethosu_driver.c')
| -rw-r--r-- | src/ethosu_driver.c | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/ethosu_driver.c b/src/ethosu_driver.c index dead0b9..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> @@ -425,7 +429,7 @@ int ethosu_init(struct ethosu_driver *drv, } } - drv->fast_memory = (uint64_t)fast_memory; + drv->fast_memory = (uintptr_t)fast_memory; drv->fast_memory_size = fast_memory_size; drv->power_request_counter = 0; |