aboutsummaryrefslogtreecommitdiff
path: root/include/ethosu_driver.h
diff options
context:
space:
mode:
authorBhavik Patel <bhavik.patel@arm.com>2020-06-18 15:25:15 +0200
committerBhavik Patel <bhavik.patel@arm.com>2020-07-03 09:06:03 +0200
commitdae5be07b76e5361593d8c2fa4717970c2a5fc19 (patch)
treebf051793cce3899f249522a8449c40432f11ef4b /include/ethosu_driver.h
parent8e32b0b72be3b109a921bcb33778eb515d27ef70 (diff)
downloadethos-u-core-driver-dae5be07b76e5361593d8c2fa4717970c2a5fc19.tar.gz
MLBEDSW-2378 Set NPU base address in ethosu_init
Change-Id: I1145834000ff81d6e497a8fa77bf997478a80372
Diffstat (limited to 'include/ethosu_driver.h')
-rw-r--r--include/ethosu_driver.h11
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.