Skip to content

Commit 3ab43c7

Browse files
committed
Add acquire_and_run function
1 parent b1e7c0a commit 3ab43c7

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/pulp_nnx_neureka.c

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,14 @@ int neureka_nnx_dispatch(neureka_dev_t *dev, neureka_task_t *task) {
5656
return 0;
5757
}
5858

59+
int neureka_nnx_acquire_and_run(neureka_dev_t *dev, neureka_task_t *task) {
60+
if (hwpe_task_queue_acquire_task(&dev->hwpe_dev, &task->id)) {
61+
return 1;
62+
}
63+
hwpe_task_queue_release_and_run(&dev->hwpe_dev);
64+
return 0;
65+
}
66+
5967
int neureka_nnx_resolve_check(neureka_dev_t *dev, neureka_task_t *task) {
6068
#if __PLATFORM__ == ARCHI_PLATFORM_GVSOC
6169
// GVSOC model has a broken running_id so resolve_check

0 commit comments

Comments
 (0)