IVGCVSW-4611 PRELU float16 VTS test skipped in CpuRef
authorTeresa Charlin <teresa.charlinreyes@arm.com>
Mon, 25 May 2020 15:16:44 +0000 (16:16 +0100)
committerTeresa Charlin <teresa.charlinreyes@arm.com>
Mon, 25 May 2020 15:16:44 +0000 (16:16 +0100)
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Id908e0bcbefd3284da51cdbe8ec0c2181b9b553e

src/backends/reference/RefLayerSupport.cpp
src/backends/reference/test/RefLayerSupportTests.cpp

index 034cd12..3bcb7e0 100644 (file)
@@ -366,9 +366,10 @@ bool RefLayerSupport::IsConcatSupported(const std::vector<const TensorInfo*> inp
 bool RefLayerSupport::IsConstantSupported(const TensorInfo& output,
                                           Optional<std::string&> reasonIfUnsupported) const
 {
-    std::array<DataType,7> supportedTypes =
+    std::array<DataType,8> supportedTypes =
     {
         DataType::BFloat16,
+        DataType::Float16,
         DataType::Float32,
         DataType::QAsymmS8,
         DataType::QAsymmU8,
index 2a27a9d..1648583 100644 (file)
@@ -241,7 +241,7 @@ BOOST_AUTO_TEST_CASE(IsConstantSupportedRef)
 
     bool result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory,
             armnn::DataType::Float16>(reasonIfUnsupported);
-    BOOST_CHECK(!result);
+    BOOST_CHECK(result);
 
     result = IsConstantLayerSupportedTests<armnn::RefWorkloadFactory,
             armnn::DataType::Float32>(reasonIfUnsupported);