diff options
Diffstat (limited to 'src/backends')
| -rw-r--r-- | src/backends/cl/test/ClCreateWorkloadTests.cpp | 5 | ||||
| -rw-r--r-- | src/backends/neon/test/NeonCreateWorkloadTests.cpp | 5 |
2 files changed, 4 insertions, 6 deletions
diff --git a/src/backends/cl/test/ClCreateWorkloadTests.cpp b/src/backends/cl/test/ClCreateWorkloadTests.cpp index fc5ccfe487..4bd3d3a33d 100644 --- a/src/backends/cl/test/ClCreateWorkloadTests.cpp +++ b/src/backends/cl/test/ClCreateWorkloadTests.cpp @@ -322,7 +322,7 @@ BOOST_AUTO_TEST_CASE(CreateConvolution2dFastMathEnabledWorkload) ClWorkloadFactoryHelper::GetFactory(ClWorkloadFactoryHelper::GetMemoryManager(), modelOptions); auto workload = - CreateConvolution2dWorkloadTest<ClConvolution2dWorkload, armnn::DataType::Float32>(factory, + CreateConvolution2dWorkloadFastMathTest<ClConvolution2dWorkload, armnn::DataType::Float32>(factory, graph, DataLayout::NCHW, modelOptions); @@ -331,8 +331,7 @@ BOOST_AUTO_TEST_CASE(CreateConvolution2dFastMathEnabledWorkload) auto conv2dWorkload = PolymorphicDowncast<ClConvolution2dWorkload*>(workload.get()); IgnoreUnused(conv2dWorkload); ARMNN_ASSERT(conv2dWorkload != nullptr); - // fast_math enabled but configuration does not match with WINOGRAD - ARMNN_ASSERT(conv2dWorkload->GetConvolutionMethod() == arm_compute::ConvolutionMethod::GEMM); + ARMNN_ASSERT(conv2dWorkload->GetConvolutionMethod() == arm_compute::ConvolutionMethod::WINOGRAD); } template <typename DepthwiseConvolutionWorkloadType, typename armnn::DataType DataType> diff --git a/src/backends/neon/test/NeonCreateWorkloadTests.cpp b/src/backends/neon/test/NeonCreateWorkloadTests.cpp index 99ff9ae8b8..c994bfe55a 100644 --- a/src/backends/neon/test/NeonCreateWorkloadTests.cpp +++ b/src/backends/neon/test/NeonCreateWorkloadTests.cpp @@ -292,7 +292,7 @@ BOOST_AUTO_TEST_CASE(CreateConvolution2dFastMathEnabledWorkload) NeonWorkloadFactoryHelper::GetFactory(NeonWorkloadFactoryHelper::GetMemoryManager(), modelOptions); auto workload = - CreateConvolution2dWorkloadTest<NeonConvolution2dWorkload, armnn::DataType::Float32>(factory, + CreateConvolution2dWorkloadFastMathTest<NeonConvolution2dWorkload, armnn::DataType::Float32>(factory, graph, DataLayout::NCHW, modelOptions); @@ -301,8 +301,7 @@ BOOST_AUTO_TEST_CASE(CreateConvolution2dFastMathEnabledWorkload) auto conv2dWorkload = PolymorphicDowncast<NeonConvolution2dWorkload*>(workload.get()); IgnoreUnused(conv2dWorkload); ARMNN_ASSERT(conv2dWorkload != nullptr); - // fast_math enabled but configuration does not match with WINOGRAD - ARMNN_ASSERT(conv2dWorkload->GetConvolutionMethod() == arm_compute::ConvolutionMethod::GEMM); + ARMNN_ASSERT(conv2dWorkload->GetConvolutionMethod() == arm_compute::ConvolutionMethod::WINOGRAD); } template <typename armnn::DataType DataType> |