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