From 55a8ffda24fff5515803df10fb4863d46a1effdf Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Wed, 7 Apr 2021 20:10:49 +0100 Subject: IVGCVSW-5823 Refactor Async Network API * Moved IAsyncNetwork into IRuntime. * All LoadedNetworks can be executed Asynchronously. Signed-off-by: Mike Kelly Change-Id: Ibbc901ab9110dc2f881425b75489bccf9ad54169 --- .../backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'src/backends') diff --git a/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp b/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp index 2ccd2b13af..66ccdbf1d9 100644 --- a/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp +++ b/src/backends/backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp @@ -40,15 +40,15 @@ void AsyncEndToEndTestImpl(INetworkPtr network, // Creates AsyncNetwork NetworkId networkId = 0; std::string errorMessage; - const INetworkProperties networkProperties; - auto asyncNetwork = runtime->CreateAsyncNetwork(networkId, std::move(optNet), errorMessage, networkProperties); + const INetworkProperties networkProperties(false, false, true); + runtime->LoadNetwork(networkId, std::move(optNet), errorMessage, networkProperties); InputTensors inputTensors; inputTensors.reserve(inputTensorData.size()); for (auto&& it : inputTensorData) { inputTensors.push_back({it.first, - ConstTensor(asyncNetwork->GetInputTensorInfo(it.first), it.second.data())}); + ConstTensor(runtime->GetInputTensorInfo(networkId, it.first), it.second.data())}); } OutputTensors outputTensors; @@ -59,16 +59,16 @@ void AsyncEndToEndTestImpl(INetworkPtr network, std::vector out(it.second.size()); outputStorage.emplace(it.first, out); outputTensors.push_back({it.first, - Tensor(asyncNetwork->GetOutputTensorInfo(it.first), + Tensor(runtime->GetOutputTensorInfo(networkId, it.first), outputStorage.at(it.first).data())}); } // Create WorkingMemHandle for this async network - std::unique_ptr workingMemHandle = asyncNetwork->CreateWorkingMemHandle(); + std::unique_ptr workingMemHandle = runtime->CreateWorkingMemHandle(networkId); IWorkingMemHandle& workingMemHandleRef = *workingMemHandle.get(); // Run the async network - asyncNetwork->Execute(inputTensors, outputTensors, workingMemHandleRef); + runtime->Execute(workingMemHandleRef, inputTensors, outputTensors); // Checks the results. for (auto&& it : expectedOutputData) -- cgit v1.2.1