diff options
Diffstat (limited to 'include/ethosu_driver.h')
-rw-r--r-- | include/ethosu_driver.h | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/include/ethosu_driver.h b/include/ethosu_driver.h index 7ab420a..f192357 100644 --- a/include/ethosu_driver.h +++ b/include/ethosu_driver.h @@ -22,6 +22,9 @@ * Includes ******************************************************************************/ +#include "ethosu_device.h" + +#include <stdbool.h> #include <stdint.h> #ifdef __cplusplus @@ -36,6 +39,12 @@ extern "C" { * Types ******************************************************************************/ +struct ethosu_driver +{ + struct ethosu_device dev; + bool abort_inference; +}; + struct ethosu_version_id { // Ethos-U id @@ -73,7 +82,7 @@ struct ethosu_version /** * Initialize the Ethos-U driver. */ -int ethosu_init(void); +int ethosu_init(const void *base_address); /** * Get Ethos-U version. |