diff options
Diffstat (limited to 'kernel/ethosu_inference.c')
-rw-r--r-- | kernel/ethosu_inference.c | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/kernel/ethosu_inference.c b/kernel/ethosu_inference.c index 58501f7..4f56126 100644 --- a/kernel/ethosu_inference.c +++ b/kernel/ethosu_inference.c @@ -25,7 +25,7 @@ #include "ethosu_inference.h" #include "ethosu_buffer.h" -#include "ethosu_core_interface.h" +#include "ethosu_core_rpmsg.h" #include "ethosu_device.h" #include "ethosu_network.h" #include "ethosu_cancel_inference.h" @@ -413,19 +413,19 @@ int ethosu_inference_put(struct ethosu_inference *inf) } void ethosu_inference_rsp(struct ethosu_device *edev, - struct ethosu_core_inference_rsp *rsp) + int msg_id, + struct ethosu_core_msg_inference_rsp *rsp) { - int id = (int)rsp->user_arg; struct ethosu_mailbox_msg *msg; struct ethosu_inference *inf; int ret; int i; - msg = ethosu_mailbox_find(&edev->mailbox, id); + msg = ethosu_mailbox_find(&edev->mailbox, msg_id); if (IS_ERR(msg)) { dev_warn(edev->dev, "Id for inference msg not found. Id=%d\n", - id); + msg_id); return; } |