1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "FloatingPointConverter.hpp"
#include <Half.hpp>
#include <malloc.h>
#include <iostream>
#include <algorithm>
#include <boost/test/unit_test.hpp>
BOOST_AUTO_TEST_SUITE(TestFPConversion)
BOOST_AUTO_TEST_CASE(TestConvertFp32ToFp16)
{
using namespace half_float::literal;
float floatArray[] = { 1.0f, 2.0f, 0.5f, 3.1f, 2.4f,
5.666f, 6.444f, 7.1f, 432.121f, 12.22f };
size_t numFloats = sizeof(floatArray) / sizeof(floatArray[0]);
std::vector<armnn::Half> convertedBuffer(numFloats, 0.0_h);
armnnUtils::FloatingPointConverter::ConvertFloat32To16(floatArray, numFloats, convertedBuffer.data());
for (size_t i = 0; i < numFloats; i++)
{
armnn::Half expected(floatArray[i]);
armnn::Half actual = convertedBuffer[i];
BOOST_CHECK_EQUAL(expected, actual);
float convertedHalf = actual;
BOOST_CHECK_CLOSE(floatArray[i], convertedHalf, 0.07);
}
}
BOOST_AUTO_TEST_CASE(TestConvertFp16ToFp32)
{
using namespace half_float::literal;
armnn::Half halfArray[] = { 1.0_h, 2.0_h, 0.5_h, 3.1_h, 2.4_h,
5.666_h, 6.444_h, 7.1_h, 432.121_h, 12.22_h };
size_t numFloats = sizeof(halfArray) / sizeof(halfArray[0]);
std::vector<float> convertedBuffer(numFloats, 0.0f);
armnnUtils::FloatingPointConverter::ConvertFloat16To32(halfArray, numFloats, convertedBuffer.data());
for (size_t i = 0; i < numFloats; i++)
{
float expected(halfArray[i]);
float actual = convertedBuffer[i];
BOOST_CHECK_EQUAL(expected, actual);
}
}
BOOST_AUTO_TEST_SUITE_END()
|
