src/backends/cl/test/ClLayerSupportTests.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include <layers/ConvertFp16ToFp32Layer.hpp>
   7 #include <layers/ConvertFp32ToFp16Layer.hpp>
   8 #include <test/TensorHelpers.hpp>
   9
  10 #include <backendsCommon/CpuTensorHandle.hpp>
  11 #include <cl/ClWorkloadFactory.hpp>
  12 #include <cl/test/ClContextControlFixture.hpp>
  13 #include <backendsCommon/test/IsLayerSupportedTestImpl.hpp>
  14 #include <backendsCommon/test/LayerTests.hpp>
  15
  16 #include <boost/test/unit_test.hpp>
  17
  18 #include <string>
  19
  20 BOOST_AUTO_TEST_SUITE(ClLayerSupport)
  21
  22 BOOST_FIXTURE_TEST_CASE(IsLayerSupportedFloat16Cl, ClContextControlFixture)
  23 {
  24     armnn::ClWorkloadFactory factory;
  25     IsLayerSupportedTests<armnn::ClWorkloadFactory, armnn::DataType::Float16>(&factory);
  26 }
  27
  28 BOOST_FIXTURE_TEST_CASE(IsLayerSupportedFloat32Cl, ClContextControlFixture)
  29 {
  30     armnn::ClWorkloadFactory factory;
  31     IsLayerSupportedTests<armnn::ClWorkloadFactory, armnn::DataType::Float32>(&factory);
  32 }
  33
  34 BOOST_FIXTURE_TEST_CASE(IsLayerSupportedUint8Cl, ClContextControlFixture)
  35 {
  36     armnn::ClWorkloadFactory factory;
  37     IsLayerSupportedTests<armnn::ClWorkloadFactory, armnn::DataType::QuantisedAsymm8>(&factory);
  38 }
  39
  40 BOOST_FIXTURE_TEST_CASE(IsConvertFp16ToFp32SupportedCl, ClContextControlFixture)
  41 {
  42     std::string reasonIfUnsupported;
  43
  44     bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp16ToFp32Layer,
  45       armnn::DataType::Float16, armnn::DataType::Float32>(reasonIfUnsupported);
  46
  47     BOOST_CHECK(result);
  48 }
  49
  50 BOOST_FIXTURE_TEST_CASE(IsConvertFp16ToFp32SupportedFp32InputCl, ClContextControlFixture)
  51 {
  52     std::string reasonIfUnsupported;
  53
  54     bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp16ToFp32Layer,
  55       armnn::DataType::Float32, armnn::DataType::Float32>(reasonIfUnsupported);
  56
  57     BOOST_CHECK(!result);
  58     BOOST_CHECK_EQUAL(reasonIfUnsupported, "Input should be Float16");
  59 }
  60
  61 BOOST_FIXTURE_TEST_CASE(IsConvertFp16ToFp32SupportedFp16OutputCl, ClContextControlFixture)
  62 {
  63     std::string reasonIfUnsupported;
  64
  65     bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp16ToFp32Layer,
  66       armnn::DataType::Float16, armnn::DataType::Float16>(reasonIfUnsupported);
  67
  68     BOOST_CHECK(!result);
  69     BOOST_CHECK_EQUAL(reasonIfUnsupported, "Output should be Float32");
  70 }
  71
  72 BOOST_FIXTURE_TEST_CASE(IsConvertFp32ToFp16SupportedCl, ClContextControlFixture)
  73 {
  74     std::string reasonIfUnsupported;
  75
  76     bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
  77       armnn::DataType::Float32, armnn::DataType::Float16>(reasonIfUnsupported);
  78
  79     BOOST_CHECK(result);
  80 }
  81
  82 BOOST_FIXTURE_TEST_CASE(IsConvertFp32ToFp16SupportedFp16InputCl, ClContextControlFixture)
  83 {
  84     std::string reasonIfUnsupported;
  85
  86     bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
  87       armnn::DataType::Float16, armnn::DataType::Float16>(reasonIfUnsupported);
  88
  89     BOOST_CHECK(!result);
  90     BOOST_CHECK_EQUAL(reasonIfUnsupported, "Input should be Float32");
  91 }
  92
  93 BOOST_FIXTURE_TEST_CASE(IsConvertFp32ToFp16SupportedFp32OutputCl, ClContextControlFixture)
  94 {
  95     std::string reasonIfUnsupported;
  96
  97     bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
  98       armnn::DataType::Float32, armnn::DataType::Float32>(reasonIfUnsupported);
  99
 100     BOOST_CHECK(!result);
 101     BOOST_CHECK_EQUAL(reasonIfUnsupported, "Output should be Float16");
 102 }
 103
 104 BOOST_AUTO_TEST_SUITE_END()