Add API to set secure and privilege level of NPU
Change-Id: I8ca309ea4e5885865d5c9cf848500f45f83e08a2
diff --git a/include/ethosu_driver.h b/include/ethosu_driver.h
index 433ad06..345f82f 100644
--- a/include/ethosu_driver.h
+++ b/include/ethosu_driver.h
@@ -89,9 +89,15 @@
/**
* Initialize the Ethos-U driver.
*/
-int ethosu_init_v2(const void *base_address, const void *fast_memory, const size_t fast_memory_size);
+int ethosu_init_v3(const void *base_address,
+ const void *fast_memory,
+ const size_t fast_memory_size,
+ uint32_t secure_enable,
+ uint32_t privilege_enable);
-#define ethosu_init(base_address) ethosu_init_v2(base_address, NULL, 0)
+#define ethosu_init(base_address) ethosu_init_v3(base_address, NULL, 0, 1, 1)
+#define ethosu_init_v2(base_address, fast_memory, fast_memory_size) \
+ ethosu_init_v3(base_address, fast_memory, fast_memory_size, 1, 1)
/**
* Get Ethos-U version.