diff options
-rw-r--r-- | tests/cancel_inference_test.cpp | 60 |
1 files changed, 1 insertions, 59 deletions
diff --git a/tests/cancel_inference_test.cpp b/tests/cancel_inference_test.cpp index 7f3fe74..7075426 100644 --- a/tests/cancel_inference_test.cpp +++ b/tests/cancel_inference_test.cpp @@ -1,5 +1,5 @@ /* - * SPDX-FileCopyrightText: Copyright 2022-2023 Arm Limited and/or its affiliates <open-source-office@arm.com> + * SPDX-FileCopyrightText: Copyright 2022-2024 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 @@ -75,63 +75,6 @@ void testCancelInference(const Device &device) { } catch (std::exception &e) { throw TestFailureException("Inference run test: ", e.what()); } } -void testRejectInference(const Device &device) { - try { - auto network = std::make_shared<Network>(device, networkModelData, sizeof(networkModelData)); - - std::vector<std::shared_ptr<Buffer>> inputBuffers; - std::vector<std::shared_ptr<Buffer>> outputBuffers; - - auto inputBuffer = std::make_shared<Buffer>(device, sizeof(inputData)); - std::memcpy(inputBuffer->data(), inputData, sizeof(inputData)); - - inputBuffers.push_back(inputBuffer); - outputBuffers.push_back(std::make_shared<Buffer>(device, sizeof(expectedOutputData))); - std::vector<uint8_t> enabledCounters(Inference::getMaxPmuEventCounters()); - - std::shared_ptr<Inference> inferences[5]; - - for (int i = 0; i < 5; i++) { - inferences[i] = std::make_shared<Inference>(network, - inputBuffers.begin(), - inputBuffers.end(), - outputBuffers.begin(), - outputBuffers.end(), - enabledCounters, - false); - - InferenceStatus status = inferences[i]->status(); - TEST_ASSERT(status == InferenceStatus::RUNNING); - } - - auto inference = std::make_shared<Inference>(network, - inputBuffers.begin(), - inputBuffers.end(), - outputBuffers.begin(), - outputBuffers.end(), - enabledCounters, - false); - - bool timedout = inference->wait(defaultTimeout); - TEST_ASSERT(!timedout); - - InferenceStatus status = inference->status(); - TEST_ASSERT(status == InferenceStatus::REJECTED); - - for (int i = 0; i < 5; i++) { - - bool success = inferences[i]->cancel(); - TEST_ASSERT(success); - - InferenceStatus status = inferences[i]->status(); - TEST_ASSERT(status == InferenceStatus::ABORTED); - - bool timedout = inference->wait(defaultTimeout); - TEST_ASSERT(!timedout); - } - } catch (std::exception &e) { throw TestFailureException("Inference run test: ", e.what()); } -} - } // namespace int main() { @@ -139,7 +82,6 @@ int main() { try { testCancelInference(device); - testRejectInference(device); } catch (TestFailureException &e) { std::cerr << "Test failure: " << e.what() << std::endl; return 1; |