aboutsummaryrefslogtreecommitdiff
path: root/applications/message_handler/test/cancel_reject_inference_test.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'applications/message_handler/test/cancel_reject_inference_test.cpp')
-rw-r--r--applications/message_handler/test/cancel_reject_inference_test.cpp255
1 files changed, 0 insertions, 255 deletions
diff --git a/applications/message_handler/test/cancel_reject_inference_test.cpp b/applications/message_handler/test/cancel_reject_inference_test.cpp
deleted file mode 100644
index 9f4f9b4..0000000
--- a/applications/message_handler/test/cancel_reject_inference_test.cpp
+++ /dev/null
@@ -1,255 +0,0 @@
-/*
- * Copyright (c) 2022 Arm Limited.
- *
- * SPDX-License-Identifier: Apache-2.0
- *
- * Licensed under the Apache License, Version 2.0 (the License); you may
- * not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an AS IS BASIS, WITHOUT
- * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/****************************************************************************
- * Includes
- ****************************************************************************/
-
-#include "FreeRTOS.h"
-#include "queue.h"
-#include "semphr.h"
-#include "task.h"
-
-#include <inttypes.h>
-#include <stdio.h>
-
-#include "ethosu_core_interface.h"
-#include "indexed_networks.hpp"
-#include "message_client.hpp"
-#include "message_handler.hpp"
-#include "message_queue.hpp"
-#include "networks.hpp"
-#include "test_assertions.hpp"
-#include "test_helpers.hpp"
-
-#include <mailbox.hpp>
-#include <mhu_dummy.hpp>
-
-/* Disable semihosting */
-__asm(".global __use_no_semihosting\n\t");
-
-using namespace EthosU;
-using namespace MessageHandler;
-
-/****************************************************************************
- * Defines
- ****************************************************************************/
-
-// TensorArena static initialisation
-constexpr size_t arenaSize = TENSOR_ARENA_SIZE;
-
-__attribute__((section(".bss.tensor_arena"), aligned(16))) uint8_t tensorArena[arenaSize];
-
-// Message queue from remote host
-__attribute__((section("ethosu_core_in_queue"))) MessageQueue::Queue<1000> inputMessageQueue;
-
-// Message queue to remote host
-__attribute__((section("ethosu_core_out_queue"))) MessageQueue::Queue<1000> outputMessageQueue;
-
-namespace {
-Mailbox::MHUDummy mailbox;
-} // namespace
-
-/****************************************************************************
- * Application
- ****************************************************************************/
-namespace {
-
-struct TaskParams {
- TaskParams() :
- messageNotify(xSemaphoreCreateBinary()),
- inferenceInputQueue(std::make_shared<Queue<ethosu_core_inference_req>>()),
- inferenceOutputQueue(xQueueCreate(5, sizeof(ethosu_core_inference_rsp))),
- networks(std::make_shared<WithIndexedNetworks>()) {}
-
- SemaphoreHandle_t messageNotify;
- // Used to pass inference requests to the inference runner task
- std::shared_ptr<Queue<ethosu_core_inference_req>> inferenceInputQueue;
- // Queue for message responses to the remote host
- QueueHandle_t inferenceOutputQueue;
- // Networks provider
- std::shared_ptr<Networks> networks;
-};
-
-void messageTask(void *pvParameters) {
- printf("Starting message task\n");
- TaskParams *params = reinterpret_cast<TaskParams *>(pvParameters);
-
- IncomingMessageHandler process(*inputMessageQueue.toQueue(),
- *outputMessageQueue.toQueue(),
- mailbox,
- params->inferenceInputQueue,
- params->inferenceOutputQueue,
- params->messageNotify,
- params->networks);
- process.run();
-}
-
-void testCancelInference(MessageClient client) {
- const uint64_t fake_inference_user_arg = 42;
- const uint32_t network_index = 0;
- ethosu_core_inference_req inference_req =
- inferenceIndexedRequest(fake_inference_user_arg, network_index, nullptr, 0, nullptr, 0);
-
- const uint64_t fake_cancel_inference_user_arg = 55;
- ethosu_core_cancel_inference_req cancel_req = {fake_cancel_inference_user_arg, fake_inference_user_arg};
-
- ethosu_core_inference_rsp inference_rsp;
- ethosu_core_cancel_inference_rsp cancel_rsp;
-
- TEST_ASSERT(client.sendInputMessage(ETHOSU_CORE_MSG_INFERENCE_REQ, inference_req));
- TEST_ASSERT(client.sendInputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_REQ, cancel_req));
-
- TEST_ASSERT(client.waitAndReadOutputMessage(ETHOSU_CORE_MSG_INFERENCE_RSP, inference_rsp));
- TEST_ASSERT(client.waitAndReadOutputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_RSP, cancel_rsp));
-
- TEST_ASSERT(inference_req.user_arg == inference_rsp.user_arg);
- TEST_ASSERT(inference_rsp.status == ETHOSU_CORE_STATUS_ABORTED);
-
- TEST_ASSERT(cancel_req.user_arg == cancel_rsp.user_arg);
- TEST_ASSERT(cancel_rsp.status == ETHOSU_CORE_STATUS_OK);
-}
-
-void testCancelNonExistentInference(MessageClient client) {
- const uint64_t fake_inference_user_arg = 42;
- const uint64_t fake_cancel_inference_user_arg = 55;
- ethosu_core_cancel_inference_req cancel_req = {fake_cancel_inference_user_arg, fake_inference_user_arg};
- ethosu_core_cancel_inference_rsp cancel_rsp;
-
- TEST_ASSERT(client.sendInputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_REQ, cancel_req));
- TEST_ASSERT(client.waitAndReadOutputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_RSP, cancel_rsp));
-
- TEST_ASSERT(cancel_req.user_arg == cancel_rsp.user_arg);
- TEST_ASSERT(cancel_rsp.status == ETHOSU_CORE_STATUS_ERROR);
-}
-
-void testCannotCancelRunningInference(MessageClient client,
- std::shared_ptr<Queue<ethosu_core_inference_req>> inferenceInputQueue) {
- const uint64_t fake_inference_user_arg = 42;
- const uint32_t network_index = 0;
- ethosu_core_inference_req inference_req =
- inferenceIndexedRequest(fake_inference_user_arg, network_index, nullptr, 0, nullptr, 0);
-
- const uint64_t fake_cancel_inference_user_arg = 55;
- ethosu_core_cancel_inference_req cancel_req = {fake_cancel_inference_user_arg, fake_inference_user_arg};
- ethosu_core_cancel_inference_rsp cancel_rsp;
-
- TEST_ASSERT(client.sendInputMessage(ETHOSU_CORE_MSG_INFERENCE_REQ, inference_req));
-
- // fake start of the inference by removing the inference from the queue
- ethosu_core_inference_req start_req;
- inferenceInputQueue->pop(start_req);
- TEST_ASSERT(inference_req.user_arg == start_req.user_arg);
-
- TEST_ASSERT(client.sendInputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_REQ, cancel_req));
- TEST_ASSERT(client.waitAndReadOutputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_RSP, cancel_rsp));
-
- TEST_ASSERT(cancel_req.user_arg == cancel_rsp.user_arg);
- TEST_ASSERT(cancel_rsp.status == ETHOSU_CORE_STATUS_ERROR);
-}
-
-void testRejectInference(MessageClient client) {
- int runs = 6;
- const uint64_t fake_inference_user_arg = 42;
- const uint32_t network_index = 0;
- const uint64_t fake_cancel_inference_user_arg = 55;
- ethosu_core_inference_req req;
- ethosu_core_inference_rsp rsp;
-
- for (int i = 0; i < runs; i++) {
-
- req = inferenceIndexedRequest(fake_inference_user_arg + i, network_index, nullptr, 0, nullptr, 0);
- TEST_ASSERT(client.sendInputMessage(ETHOSU_CORE_MSG_INFERENCE_REQ, req));
- vTaskDelay(150);
- }
-
- TEST_ASSERT(client.waitAndReadOutputMessage(ETHOSU_CORE_MSG_INFERENCE_RSP, rsp));
- TEST_ASSERT(uint64_t(fake_inference_user_arg + runs - 1) == rsp.user_arg);
- TEST_ASSERT(rsp.status == ETHOSU_CORE_STATUS_REJECTED);
-
- // let's cleanup the queue
- ethosu_core_cancel_inference_req cancel_req = {0, 0};
- ethosu_core_cancel_inference_rsp cancel_rsp;
- ethosu_core_inference_rsp inference_rsp;
-
- for (int i = 0; i < runs - 1; i++) {
- cancel_req.user_arg = fake_cancel_inference_user_arg + i;
- cancel_req.inference_handle = fake_inference_user_arg + i;
- TEST_ASSERT(client.sendInputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_REQ, cancel_req));
-
- TEST_ASSERT(client.waitAndReadOutputMessage(ETHOSU_CORE_MSG_INFERENCE_RSP, inference_rsp));
- TEST_ASSERT(inference_rsp.user_arg = cancel_req.inference_handle);
-
- TEST_ASSERT(client.waitAndReadOutputMessage(ETHOSU_CORE_MSG_CANCEL_INFERENCE_RSP, cancel_rsp));
- TEST_ASSERT(cancel_req.user_arg == cancel_rsp.user_arg);
- TEST_ASSERT(cancel_rsp.status == ETHOSU_CORE_STATUS_OK);
- }
-}
-
-void clientTask(void *pvParameters) {
- printf("Starting client task\n");
- TaskParams *params = reinterpret_cast<TaskParams *>(pvParameters);
-
- MessageClient client(*inputMessageQueue.toQueue(), *outputMessageQueue.toQueue(), mailbox);
-
- vTaskDelay(50);
-
- testCancelInference(client);
- testCancelNonExistentInference(client);
- testCannotCancelRunningInference(client, params->inferenceInputQueue);
- testRejectInference(client);
-
- exit(0);
-}
-
-/*
- * Keep task parameters as global data as FreeRTOS resets the stack when the
- * scheduler is started.
- */
-TaskParams taskParams;
-
-} // namespace
-
-// FreeRTOS application. NOTE: Additional tasks may require increased heap size.
-int main() {
- BaseType_t ret;
-
- if (!mailbox.verifyHardware()) {
- printf("Failed to verify mailbox hardware\n");
- return 1;
- }
-
- // Task for handling incoming /outgoing messages from the remote host
- ret = xTaskCreate(messageTask, "messageTask", 1024, &taskParams, 2, nullptr);
- if (ret != pdPASS) {
- printf("Failed to create 'messageTask'\n");
- return ret;
- }
-
- // Task for handling incoming /outgoing messages from the remote host
- ret = xTaskCreate(clientTask, "clientTask", 1024, &taskParams, 2, nullptr);
- if (ret != pdPASS) {
- printf("Failed to create 'messageTask'\n");
- return ret;
- }
-
- // Start Scheduler
- vTaskStartScheduler();
-
- return 1;
-}
generated by cgit v1.2.1 (git 2.23.0) at 2024-05-19 15:51:59 +0000