xref: /aosp_15_r20/external/armnn/delegate/test/CastTest.cpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1*89c4ff92SAndroid Build Coastguard Worker //
2*89c4ff92SAndroid Build Coastguard Worker // Copyright © 2021, 2023 Arm Ltd and Contributors. All rights reserved.
3*89c4ff92SAndroid Build Coastguard Worker // SPDX-License-Identifier: MIT
4*89c4ff92SAndroid Build Coastguard Worker //
5*89c4ff92SAndroid Build Coastguard Worker 
6*89c4ff92SAndroid Build Coastguard Worker #include "CastTestHelper.hpp"
7*89c4ff92SAndroid Build Coastguard Worker 
8*89c4ff92SAndroid Build Coastguard Worker #include <armnn_delegate.hpp>
9*89c4ff92SAndroid Build Coastguard Worker 
10*89c4ff92SAndroid Build Coastguard Worker #include <flatbuffers/flatbuffers.h>
11*89c4ff92SAndroid Build Coastguard Worker #include <schema_generated.h>
12*89c4ff92SAndroid Build Coastguard Worker 
13*89c4ff92SAndroid Build Coastguard Worker #include <doctest/doctest.h>
14*89c4ff92SAndroid Build Coastguard Worker 
15*89c4ff92SAndroid Build Coastguard Worker namespace armnnDelegate
16*89c4ff92SAndroid Build Coastguard Worker {
17*89c4ff92SAndroid Build Coastguard Worker 
CastUint8ToFp32Test(std::vector<armnn::BackendId> & backends)18*89c4ff92SAndroid Build Coastguard Worker void CastUint8ToFp32Test(std::vector<armnn::BackendId>& backends)
19*89c4ff92SAndroid Build Coastguard Worker {
20*89c4ff92SAndroid Build Coastguard Worker     std::vector<int32_t> inputShape  {1, 3, 2, 3};
21*89c4ff92SAndroid Build Coastguard Worker 
22*89c4ff92SAndroid Build Coastguard Worker     std::vector<uint8_t> inputValues { 1, 3, 1, 3, 1, 3, 1, 3, 1,
23*89c4ff92SAndroid Build Coastguard Worker                                         3, 1, 3, 1, 2, 1, 3, 1, 3 };
24*89c4ff92SAndroid Build Coastguard Worker 
25*89c4ff92SAndroid Build Coastguard Worker     std::vector<float> expectedOutputValues { 1.0f, 3.0f, 1.0f, 3.0f, 1.0f, 3.0f, 1.0f, 3.0f, 1.0f,
26*89c4ff92SAndroid Build Coastguard Worker                                               3.0f, 1.0f, 3.0f, 1.0f, 2.0f, 1.0f, 3.0f, 1.0f, 3.0f };
27*89c4ff92SAndroid Build Coastguard Worker 
28*89c4ff92SAndroid Build Coastguard Worker     CastTest<uint8_t, float>(::tflite::TensorType_UINT8,
29*89c4ff92SAndroid Build Coastguard Worker                              ::tflite::TensorType_FLOAT32,
30*89c4ff92SAndroid Build Coastguard Worker                              backends,
31*89c4ff92SAndroid Build Coastguard Worker                              inputShape,
32*89c4ff92SAndroid Build Coastguard Worker                              inputValues,
33*89c4ff92SAndroid Build Coastguard Worker                              expectedOutputValues);
34*89c4ff92SAndroid Build Coastguard Worker }
35*89c4ff92SAndroid Build Coastguard Worker 
CastInt32ToFp32Test(std::vector<armnn::BackendId> & backends)36*89c4ff92SAndroid Build Coastguard Worker void CastInt32ToFp32Test(std::vector<armnn::BackendId>& backends)
37*89c4ff92SAndroid Build Coastguard Worker {
38*89c4ff92SAndroid Build Coastguard Worker     std::vector<int32_t> inputShape  {1, 3, 2, 3};
39*89c4ff92SAndroid Build Coastguard Worker 
40*89c4ff92SAndroid Build Coastguard Worker     std::vector<int32_t> inputValues { -1, -3, -1, -3, -1, -3, -1, -3, 1,
41*89c4ff92SAndroid Build Coastguard Worker                                        3, 1, 3, 1, 2, 1, 3, 1, 3 };
42*89c4ff92SAndroid Build Coastguard Worker 
43*89c4ff92SAndroid Build Coastguard Worker     std::vector<float> expectedOutputValues { -1.0f, -3.0f, -1.0f, -3.0f, -1.0f, -3.0f, -1.0f, -3.0f, 1.0f,
44*89c4ff92SAndroid Build Coastguard Worker                                               3.0f, 1.0f, 3.0f, 1.0f, 2.0f, 1.0f, 3.0f, 1.0f, 3.0f };
45*89c4ff92SAndroid Build Coastguard Worker 
46*89c4ff92SAndroid Build Coastguard Worker     CastTest<int32_t, float>(::tflite::TensorType_INT32,
47*89c4ff92SAndroid Build Coastguard Worker                              ::tflite::TensorType_FLOAT32,
48*89c4ff92SAndroid Build Coastguard Worker                              backends,
49*89c4ff92SAndroid Build Coastguard Worker                              inputShape,
50*89c4ff92SAndroid Build Coastguard Worker                              inputValues,
51*89c4ff92SAndroid Build Coastguard Worker                              expectedOutputValues);
52*89c4ff92SAndroid Build Coastguard Worker }
53*89c4ff92SAndroid Build Coastguard Worker 
54*89c4ff92SAndroid Build Coastguard Worker // CAST Test Suite
55*89c4ff92SAndroid Build Coastguard Worker TEST_SUITE("CAST_CpuRefTests")
56*89c4ff92SAndroid Build Coastguard Worker {
57*89c4ff92SAndroid Build Coastguard Worker 
58*89c4ff92SAndroid Build Coastguard Worker TEST_CASE ("CAST_UINT8_TO_FP32_CpuRef_Test")
59*89c4ff92SAndroid Build Coastguard Worker {
60*89c4ff92SAndroid Build Coastguard Worker     std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
61*89c4ff92SAndroid Build Coastguard Worker     CastUint8ToFp32Test(backends);
62*89c4ff92SAndroid Build Coastguard Worker }
63*89c4ff92SAndroid Build Coastguard Worker 
64*89c4ff92SAndroid Build Coastguard Worker TEST_CASE ("CAST_INT32_TO_FP32_CpuRef_Test")
65*89c4ff92SAndroid Build Coastguard Worker {
66*89c4ff92SAndroid Build Coastguard Worker     std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
67*89c4ff92SAndroid Build Coastguard Worker     CastInt32ToFp32Test(backends);
68*89c4ff92SAndroid Build Coastguard Worker }
69*89c4ff92SAndroid Build Coastguard Worker 
70*89c4ff92SAndroid Build Coastguard Worker }
71*89c4ff92SAndroid Build Coastguard Worker 
72*89c4ff92SAndroid Build Coastguard Worker TEST_SUITE("CAST_CpuAccTests")
73*89c4ff92SAndroid Build Coastguard Worker {
74*89c4ff92SAndroid Build Coastguard Worker 
75*89c4ff92SAndroid Build Coastguard Worker TEST_CASE ("CAST_INT32_TO_FP32_CpuAcc_Test")
76*89c4ff92SAndroid Build Coastguard Worker {
77*89c4ff92SAndroid Build Coastguard Worker     std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
78*89c4ff92SAndroid Build Coastguard Worker     CastInt32ToFp32Test(backends);
79*89c4ff92SAndroid Build Coastguard Worker }
80*89c4ff92SAndroid Build Coastguard Worker 
81*89c4ff92SAndroid Build Coastguard Worker }
82*89c4ff92SAndroid Build Coastguard Worker 
83*89c4ff92SAndroid Build Coastguard Worker TEST_SUITE("CAST_GpuAccTests")
84*89c4ff92SAndroid Build Coastguard Worker {
85*89c4ff92SAndroid Build Coastguard Worker 
86*89c4ff92SAndroid Build Coastguard Worker TEST_CASE ("CAST_INT32_TO_FP32_GpuAcc_Test")
87*89c4ff92SAndroid Build Coastguard Worker {
88*89c4ff92SAndroid Build Coastguard Worker     std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
89*89c4ff92SAndroid Build Coastguard Worker     CastInt32ToFp32Test(backends);
90*89c4ff92SAndroid Build Coastguard Worker }
91*89c4ff92SAndroid Build Coastguard Worker 
92*89c4ff92SAndroid Build Coastguard Worker }
93*89c4ff92SAndroid Build Coastguard Worker // End of CAST Test Suite
94*89c4ff92SAndroid Build Coastguard Worker 
95*89c4ff92SAndroid Build Coastguard Worker } // namespace armnnDelegate