blob: 1fef43957d24d11e658f555e320b53251dabd904 [file] [log] [blame]
Colm Donelanade8c1d2021-03-24 14:51:31 +00001//
2// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#include "NeonWorkloadFactoryHelper.hpp"
7
Colm Donelanade8c1d2021-03-24 14:51:31 +00008#include <neon/NeonWorkloadFactory.hpp>
Colm Donelan92c50722021-03-29 14:59:04 +01009#include <test/UnitTests.hpp>
Colm Donelanade8c1d2021-03-24 14:51:31 +000010
11BOOST_AUTO_TEST_SUITE(Compute_ArmComputeNeon)
12
13using namespace armnn;
14
15using FactoryType = NeonWorkloadFactory;
16
17// ============================================================================
18// This is a specific subset of NeonLayerTests that can fail because of a known problem
19// in the Android NDK. https://github.com/android/ndk/issues/1135
20// We extract them here so then in the case of a debug Android build they can be excluded.
21// The tests will pass in a release build. The problem has been corrected in NDK r21.
22
23// Softmax
24ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
25ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
26
27// LogSoftmax
28ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_1, LogSoftmaxTest1<DataType::Float32>)
29
30ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dNhwc, L2Normalization1dTest, DataLayout::NHWC)
31
32ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm,
33 LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest)
34
Colm Donelanade8c1d2021-03-24 14:51:31 +000035// ReduceSum
36ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumFloat32, ReduceSumSimpleTest<DataType::Float32>)
37
38ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_3, ReduceSumSingleAxisTest3<DataType::Float32>)
39
Colm Donelan92c50722021-03-29 14:59:04 +010040#if defined(ARMNNREF_ENABLED)
41
42ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta1WithReference, CompareSoftmaxTest, 1.0f)
43ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta2WithReference, CompareSoftmaxTest, 2.0f)
44
45#endif
Colm Donelanade8c1d2021-03-24 14:51:31 +000046
47BOOST_AUTO_TEST_SUITE_END()