From 386ff1a721cdca3689b009ba31f2d3ac8bea2fae Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Mon, 29 Mar 2021 15:04:50 +0100 Subject: IVGCVSW-5790 Merge async prototype * Added thread safe execution mechanism for armnn * Removed duplicate function bool Compare(T a, T b, float tolerance) * Added StridedSliceAsyncEndToEndTest * Fixed memory leak Signed-off-by: Mike Kelly Change-Id: I2d367fc77ee7c01b8953138543e76af5e691211f --- include/armnn/backends/IWorkload.hpp | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'include/armnn/backends/IWorkload.hpp') diff --git a/include/armnn/backends/IWorkload.hpp b/include/armnn/backends/IWorkload.hpp index 0bd8d2db75..a4827ebcdf 100644 --- a/include/armnn/backends/IWorkload.hpp +++ b/include/armnn/backends/IWorkload.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2020 Arm Ltd. All rights reserved. +// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -9,6 +9,15 @@ namespace armnn { +namespace experimental +{ + +struct WorkingMemDescriptor; + +} // end experimental namespace + +using namespace armnn::experimental; + /// Workload interface to enqueue a layer computation. class IWorkload { public: @@ -18,9 +27,11 @@ public: virtual void Execute() const = 0; + virtual void ExecuteAsync(WorkingMemDescriptor& desc) = 0; + virtual profiling::ProfilingGuid GetGuid() const = 0; - virtual void RegisterDebugCallback(const DebugCallbackFunction & /*func*/) {} + virtual void RegisterDebugCallback(const DebugCallbackFunction& /*func*/) {} }; } //namespace armnn -- cgit v1.2.1