blob: 24109605cc80158683be5d9239540d84a624d756 [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
Sadik Armagana097d2a2021-11-24 15:47:28 +00008#include <UnitTests.hpp>
9#include <backendsCommon/test/LayerTests.hpp>
Colm Donelanade8c1d2021-03-24 14:51:31 +000010#include <neon/NeonWorkloadFactory.hpp>
Colm Donelanade8c1d2021-03-24 14:51:31 +000011
Sadik Armagan1625efc2021-06-10 18:24:34 +010012#include <doctest/doctest.h>
Colm Donelanade8c1d2021-03-24 14:51:31 +000013
Sadik Armagan1625efc2021-06-10 18:24:34 +010014TEST_SUITE("Compute_ArmComputeNeon")
15{
Colm Donelanade8c1d2021-03-24 14:51:31 +000016using namespace armnn;
17
18using FactoryType = NeonWorkloadFactory;
19
20// ============================================================================
21// This is a specific subset of NeonLayerTests that can fail because of a known problem
22// in the Android NDK. https://github.com/android/ndk/issues/1135
23// We extract them here so then in the case of a debug Android build they can be excluded.
24// The tests will pass in a release build. The problem has been corrected in NDK r21.
25
26// Softmax
27ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta1, SimpleSoftmaxTest, 1.0f)
28ARMNN_AUTO_TEST_CASE_WITH_THF(SimpleSoftmaxBeta2, SimpleSoftmaxTest, 2.0f)
29
30// LogSoftmax
31ARMNN_AUTO_TEST_CASE_WITH_THF(LogSoftmaxFloat32_1, LogSoftmaxTest1<DataType::Float32>)
32
33ARMNN_AUTO_TEST_CASE_WITH_THF(L2Normalization1dNhwc, L2Normalization1dTest, DataLayout::NHWC)
34
35ARMNN_AUTO_TEST_CASE_WITH_THF(LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNorm,
36 LstmLayerFloat32NoCifgWithPeepholeWithProjectionWithLayerNormTest)
37
Colm Donelanade8c1d2021-03-24 14:51:31 +000038// ReduceSum
39ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumFloat32, ReduceSumSimpleTest<DataType::Float32>)
40
41ARMNN_AUTO_TEST_CASE_WITH_THF(ReduceSumSingleAxisFloat32_3, ReduceSumSingleAxisTest3<DataType::Float32>)
42
Colm Donelan92c50722021-03-29 14:59:04 +010043#if defined(ARMNNREF_ENABLED)
44
45ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta1WithReference, CompareSoftmaxTest, 1.0f)
46ARMNN_COMPARE_REF_AUTO_TEST_CASE_WITH_THF(CompareSoftmaxBeta2WithReference, CompareSoftmaxTest, 2.0f)
47
48#endif
Colm Donelanade8c1d2021-03-24 14:51:31 +000049
Sadik Armagan1625efc2021-06-10 18:24:34 +010050}