Merge pull request #68 from pulp-platform/lukamac/pr/gap9-ne16-perf-f…

…ixes Fix task waiting
pulp-platform · Nov 13, 2023 · be3d6dd · be3d6dd
2 parents 26b0e82 + 8a86e8e
commit be3d6dd
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/dory/Hardware_targets/PULP/GAP9_NE16/Utils_files/execute.h b/dory/Hardware_targets/PULP/GAP9_NE16/Utils_files/execute.h
@@ -51,9 +51,15 @@ static inline void execute_stride2x2_blocking(nnx_task_t *task, Layer tile,
 }
 
 static inline void execute_wait(nnx_task_t *task) {
-  while (!ne16_empty()) {
-    ne16_event_wait();
-  }
+#if __PLATFORM__ == ARCHI_PLATFORM_GVSOC && defined GAP_SDK
+  // Temporary hack because the gvsoc model of ne16 in gap_sdk
+  // has a broken running_id.
+  while(!ne16_empty())
+    ;
+#else
+  while(!nnx_resolve_check(task))
+    ;
+#endif
 }
 
 #endif // __EXECUTE_H__