diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-02-28 12:45:21 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-03-03 15:59:39 +0000 |
commit | c3bf6efb48a4540c8addcc02813c9381e4fceb1f (patch) | |
tree | 55fd9c3fa639ee32aa847dda2a525a7badee9e8d /src/armnn/test/UtilsTests.cpp | |
parent | 490b7becb8029ead26423b0d62e631a929e55d6c (diff) | |
download | armnn-c3bf6efb48a4540c8addcc02813c9381e4fceb1f.tar.gz |
IVGCVSW-4508 Add BFloat16 data type
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ie2fcf06d0bae9e2ef958e60ab9e8b121fdc7b590
Diffstat (limited to 'src/armnn/test/UtilsTests.cpp')
-rw-r--r-- | src/armnn/test/UtilsTests.cpp | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/src/armnn/test/UtilsTests.cpp b/src/armnn/test/UtilsTests.cpp index 4c371d6ed9..faf4480029 100644 --- a/src/armnn/test/UtilsTests.cpp +++ b/src/armnn/test/UtilsTests.cpp @@ -85,6 +85,25 @@ BOOST_AUTO_TEST_CASE(HalfType) BOOST_CHECK((GetDataTypeName(armnn::DataType::Float16) == std::string("Float16"))); } +BOOST_AUTO_TEST_CASE(BFloatType) +{ + armnn::BFloat16 a = 16256; + + // Test BFloat16 type + BOOST_CHECK_EQUAL(sizeof(a), 2); + + // Test utility function returns correct type. + using ResolvedType = armnn::ResolveType<armnn::DataType::BFloat16>; + constexpr bool isBFloat16Type = std::is_same<armnn::BFloat16, ResolvedType>::value; + BOOST_CHECK(isBFloat16Type); + + //Test utility functions return correct size + BOOST_CHECK(GetDataTypeSize(armnn::DataType::BFloat16) == 2); + + //Test utility functions return correct name + BOOST_CHECK((GetDataTypeName(armnn::DataType::BFloat16) == std::string("BFloat16"))); +} + BOOST_AUTO_TEST_CASE(GraphTopologicalSortSimpleTest) { std::map<int, std::vector<int>> graph; |