diff options
Diffstat (limited to 'src/backends/neon/test/NeonCreateWorkloadTests.cpp')
| -rw-r--r-- | src/backends/neon/test/NeonCreateWorkloadTests.cpp | 24 |
1 files changed, 20 insertions, 4 deletions
diff --git a/src/backends/neon/test/NeonCreateWorkloadTests.cpp b/src/backends/neon/test/NeonCreateWorkloadTests.cpp index 86fc457130..d4065898a8 100644 --- a/src/backends/neon/test/NeonCreateWorkloadTests.cpp +++ b/src/backends/neon/test/NeonCreateWorkloadTests.cpp @@ -125,7 +125,7 @@ BOOST_AUTO_TEST_CASE(CreateAdditionFloatWorkload) #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC BOOST_AUTO_TEST_CASE(CreateSubtractionFloat16Workload) { - NeonCreateElementwiseWorkloadTest<NeonSubtractionFloatWorkload, + NeonCreateElementwiseWorkloadTest<NeonSubtractionWorkload, SubtractionQueueDescriptor, SubtractionLayer, DataType::Float16>(); @@ -134,16 +134,24 @@ BOOST_AUTO_TEST_CASE(CreateSubtractionFloat16Workload) BOOST_AUTO_TEST_CASE(CreateSubtractionFloatWorkload) { - NeonCreateElementwiseWorkloadTest<NeonSubtractionFloatWorkload, + NeonCreateElementwiseWorkloadTest<NeonSubtractionWorkload, SubtractionQueueDescriptor, SubtractionLayer, DataType::Float32>(); } +BOOST_AUTO_TEST_CASE(CreateSubtractionUint8Workload) +{ + NeonCreateElementwiseWorkloadTest<NeonSubtractionWorkload, + SubtractionQueueDescriptor, + SubtractionLayer, + DataType::QuantisedAsymm8>(); +} + #ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC BOOST_AUTO_TEST_CASE(CreateMultiplicationFloat16Workload) { - NeonCreateElementwiseWorkloadTest<NeonMultiplicationFloatWorkload, + NeonCreateElementwiseWorkloadTest<NeonMultiplicationWorkload, MultiplicationQueueDescriptor, MultiplicationLayer, DataType::Float16>(); @@ -152,12 +160,20 @@ BOOST_AUTO_TEST_CASE(CreateMultiplicationFloat16Workload) BOOST_AUTO_TEST_CASE(CreateMultiplicationFloatWorkload) { - NeonCreateElementwiseWorkloadTest<NeonMultiplicationFloatWorkload, + NeonCreateElementwiseWorkloadTest<NeonMultiplicationWorkload, MultiplicationQueueDescriptor, MultiplicationLayer, DataType::Float32>(); } +BOOST_AUTO_TEST_CASE(CreateMultiplicationUint8Workload) +{ + NeonCreateElementwiseWorkloadTest<NeonMultiplicationWorkload, + MultiplicationQueueDescriptor, + MultiplicationLayer, + DataType::QuantisedAsymm8>(); +} + template <typename BatchNormalizationWorkloadType, typename armnn::DataType DataType> static void NeonCreateBatchNormalizationWorkloadTest(DataLayout dataLayout) { |