aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/WorkloadTestUtils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/test/WorkloadTestUtils.hpp')
-rw-r--r--src/backends/backendsCommon/test/WorkloadTestUtils.hpp114
1 files changed, 5 insertions, 109 deletions
diff --git a/src/backends/backendsCommon/test/WorkloadTestUtils.hpp b/src/backends/backendsCommon/test/WorkloadTestUtils.hpp
index 3173561a94..cb605af2d3 100644
--- a/src/backends/backendsCommon/test/WorkloadTestUtils.hpp
+++ b/src/backends/backendsCommon/test/WorkloadTestUtils.hpp
@@ -1,113 +1,9 @@
//
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
-#pragma once
-#include <armnn/Tensor.hpp>
-
-#include <armnn/backends/IBackendInternal.hpp>
-#include <armnn/backends/IMemoryManager.hpp>
-#include <backendsCommon/Workload.hpp>
-#include <backendsCommon/WorkloadInfo.hpp>
-
-namespace armnn
-{
-class ITensorHandle;
-} // namespace armnn
-
-namespace
-{
-
-template <typename QueueDescriptor>
-void AddInputToWorkload(QueueDescriptor& descriptor,
- armnn::WorkloadInfo& info,
- const armnn::TensorInfo& tensorInfo,
- armnn::ITensorHandle* tensorHandle)
-{
- descriptor.m_Inputs.push_back(tensorHandle);
- info.m_InputTensorInfos.push_back(tensorInfo);
-}
-
-template <typename QueueDescriptor>
-void AddOutputToWorkload(QueueDescriptor& descriptor,
- armnn::WorkloadInfo& info,
- const armnn::TensorInfo& tensorInfo,
- armnn::ITensorHandle* tensorHandle)
-{
- descriptor.m_Outputs.push_back(tensorHandle);
- info.m_OutputTensorInfos.push_back(tensorInfo);
-}
-
-template <typename QueueDescriptor>
-void SetWorkloadInput(QueueDescriptor& descriptor,
- armnn::WorkloadInfo& info,
- unsigned int index,
- const armnn::TensorInfo& tensorInfo,
- armnn::ITensorHandle* tensorHandle)
-{
- descriptor.m_Inputs[index] = tensorHandle;
- info.m_InputTensorInfos[index] = tensorInfo;
-}
-
-template <typename QueueDescriptor>
-void SetWorkloadOutput(QueueDescriptor& descriptor,
- armnn::WorkloadInfo& info,
- unsigned int index,
- const armnn::TensorInfo& tensorInfo,
- armnn::ITensorHandle* tensorHandle)
-{
- descriptor.m_Outputs[index] = tensorHandle;
- info.m_OutputTensorInfos[index] = tensorInfo;
-}
-
-inline void ExecuteWorkload(armnn::IWorkload& workload,
- const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- bool memoryManagementRequested = true)
-{
- const bool manageMemory = memoryManager && memoryManagementRequested;
-
- // Acquire working memory (if needed)
- if (manageMemory)
- {
- memoryManager->Acquire();
- }
-
- // Perform PostAllocationConfiguration
- workload.PostAllocationConfigure();
-
- // Execute the workload
- workload.Execute();
-
- // Release working memory (if needed)
- if (manageMemory)
- {
- memoryManager->Release();
- }
-}
-
-inline armnn::Optional<armnn::DataType> GetBiasTypeFromWeightsType(armnn::Optional<armnn::DataType> weightsType)
-{
- if (!weightsType)
- {
- return weightsType;
- }
-
- switch(weightsType.value())
- {
- case armnn::DataType::BFloat16:
- case armnn::DataType::Float16:
- case armnn::DataType::Float32:
- return weightsType;
- case armnn::DataType::QAsymmS8:
- case armnn::DataType::QAsymmU8:
- case armnn::DataType::QSymmS8:
- case armnn::DataType::QSymmS16:
- return armnn::DataType::Signed32;
- default:
- ARMNN_ASSERT_MSG(false, "GetBiasTypeFromWeightsType(): Unsupported data type.");
- }
- return armnn::EmptyOptional();
-}
-
-} // anonymous namespace
+// This file is deprecated and will be removed soon.
+// Please use the new header in armnnTestUtils instead.
+// This will use the new armnnTestUtils header.
+#include "../../../armnnTestUtils/WorkloadTestUtils.hpp"