1 files changed, 7 insertions, 2 deletions
diff --git a/applications/message_handler_openamp/inference_runner.hpp b/applications/message_handler_openamp/inference_runner.hpp
index c9461a0..f87aa7a 100644
--- a/applications/message_handler_openamp/inference_runner.hpp
+++ b/applications/message_handler_openamp/inference_runner.hpp
@@ -1,6 +1,5 @@
 /*
  * SPDX-FileCopyrightText: Copyright 2022-2023 Arm Limited and/or its affiliates <open-source-office@arm.com>
- *
  * SPDX-License-Identifier: Apache-2.0
  *
  * Licensed under the Apache License, Version 2.0 (the License); you may
@@ -27,6 +26,12 @@
 #include <inference_process.hpp>
 
 /*****************************************************************************
+ * JobContext
+ *****************************************************************************/
+
+struct JobContext;
+
+/*****************************************************************************
  * InferenceRunner
  *****************************************************************************/
 
@@ -45,7 +50,7 @@ private:
                                 const uint64_t msgId,
                                 const EthosU::ethosu_core_msg_inference_req &request);
     InferenceProcess::InferenceJob makeInferenceJob(const EthosU::ethosu_core_msg_inference_req &request,
-                                                    EthosU::ethosu_core_msg_inference_rsp &response);
+                                                    JobContext &context);
 
     MessageHandler::InferenceQueue &inferenceQueue;
     MessageHandler::ResponseQueue &responseQueue;