From 79141b9662547eeefb3ad533637223de40726e12 Mon Sep 17 00:00:00 2001 From: David Beck Date: Tue, 23 Oct 2018 16:09:36 +0100 Subject: IVGCVSW-2071 : remove GetCompute() from the WorkloadFactory interface Change-Id: I44a9d26d1a5d876d381aee4c6450af62811d0dbb --- src/backends/test/Pooling2dTestImpl.hpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/backends/test/Pooling2dTestImpl.hpp') diff --git a/src/backends/test/Pooling2dTestImpl.hpp b/src/backends/test/Pooling2dTestImpl.hpp index 29263af9bc..90be2897e8 100644 --- a/src/backends/test/Pooling2dTestImpl.hpp +++ b/src/backends/test/Pooling2dTestImpl.hpp @@ -77,10 +77,10 @@ LayerTestResult SimplePooling2dTestImpl(armnn::IWorkloadFactory& workloadF AddOutputToWorkload(queueDescriptor, workloadInfo, outputTensorInfo, outputHandle.get()); // Don't execute if Pooling is not supported, as an exception will be raised. - armnn::Compute compute = workloadFactory.GetCompute(); + armnn::BackendId backend = workloadFactory.GetBackendId(); const size_t reasonIfUnsupportedMaxLen = 255; char reasonIfUnsupported[reasonIfUnsupportedMaxLen+1]; - result.supported = armnn::IsPooling2dSupported(compute, inputTensorInfo, outputTensorInfo, + result.supported = armnn::IsPooling2dSupported(backend, inputTensorInfo, outputTensorInfo, queueDescriptor.m_Parameters, reasonIfUnsupported, reasonIfUnsupportedMaxLen); if (!result.supported) @@ -650,10 +650,10 @@ LayerTestResult ComparePooling2dTestCommon(armnn::IWorkloadFactory& worklo std::unique_ptr inputHandleRef = refWorkloadFactory.CreateTensorHandle(inputTensorInfo); // Don't execute if Pooling is not supported, as an exception will be raised. - armnn::Compute compute = workloadFactory.GetCompute(); + armnn::BackendId backend = workloadFactory.GetBackendId(); const size_t reasonIfUnsupportedMaxLen = 255; char reasonIfUnsupported[reasonIfUnsupportedMaxLen+1]; - comparisonResult.supported = armnn::IsPooling2dSupported(compute, inputTensorInfo, outputTensorInfo, + comparisonResult.supported = armnn::IsPooling2dSupported(backend, inputTensorInfo, outputTensorInfo, data.m_Parameters, reasonIfUnsupported, reasonIfUnsupportedMaxLen); if (!comparisonResult.supported) -- cgit v1.2.1