1 files changed, 7 insertions, 7 deletions
diff --git a/targets/corstone-300/target.cpp b/targets/corstone-300/target.cpp
index 5d93854..2344995 100644
--- a/targets/corstone-300/target.cpp
+++ b/targets/corstone-300/target.cpp
@@ -135,7 +135,7 @@ void HardFault_Handler() {
 
 #ifdef ETHOSU
 void ethosuIrqHandler() {
-    ethosu_irq_handler();
+    ethosu_irq_handler(ethosu0_driver);
 }
 #endif
 
@@ -158,12 +158,12 @@ void targetSetup() {
 
 #ifdef ETHOSU
     // Initialize Ethos-U NPU driver
-    if (ethosu_init_v4(ethosu0_driver,
-                       reinterpret_cast<void *>(ETHOSU_BASE_ADDRESS),
-                       ethosu_scratch,
-                       ETHOSU_FAST_MEMORY_SIZE,
-                       1,
-                       1)) {
+    if (ethosu_init(ethosu0_driver,
+                    reinterpret_cast<void *>(ETHOSU_BASE_ADDRESS),
+                    ethosu_scratch,
+                    ETHOSU_FAST_MEMORY_SIZE,
+                    1,
+                    1)) {
         printf("Failed to initialize NPU.\n");
         return;
     }