commit 8b1fe55bf4788ac85a0d8f9059ca0305904ddc5d
author Davide Grohmann <davide.grohmann@arm.com> Thu Apr 07 16:58:32 2022 +0200
committer Davide Grohmann <davide.grohmann@arm.com> Mon May 09 09:38:51 2022 +0200
tree 248f62319537bd4daecb62db44171b533d7a4ffe
parent 7e8f508765632c42cc44fd8ad704c9d90943ab32

Reset firmware when inference cancellation fails

If an inference fails to be cancelled and it is still in a pending
state, reset the whole firmware.  Indeed the assumption is that
something is at fault on the firmware side and it is not progressing.

Change-Id: I2f568b2167d86cda3cb96a5e83551b018f5fc55d

kernel/ethosu_cancel_inference.c

diff --git a/kernel/ethosu_cancel_inference.c b/kernel/ethosu_cancel_inference.c
index 09778ee..befdd2f 100644
--- a/kernel/ethosu_cancel_inference.c
+++ b/kernel/ethosu_cancel_inference.c
@@ -159,6 +159,15 @@
 		goto put_kref;
 	}
 
+	/* if cancellation failed and the inference did not complete then reset
+	 * the firmware */
+	if (cancellation->uapi->status == ETHOSU_UAPI_STATUS_ERROR &&
+	    !cancellation->inf->done) {
+		ret = ethosu_firmware_reset(cancellation->edev);
+		if (ret)
+			goto put_kref;
+	}
+
 put_kref:
 	kref_put(&cancellation->kref, &ethosu_cancel_inference_destroy);