commit db1a2834b4e9d74ed538943634212eccbd4a789b
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Tue Nov 12 16:15:11 2019 +0000
committer Matteo Martincigh <matteo.martincigh@arm.com> Fri Nov 15 11:57:50 2019 +0000
tree 10162ed219468bc043ed66e40ea657fb867b0ee1
parent 60538ada2b90704abcf6473144639103d80287a5

Add FP16 support to DebugWorkload

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Ia879f2d84a1b977474ee0dafa976f2aab32bd3ae

src/backends/reference/RefWorkloadFactory.cpp

diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp
index c2cb51a..7fd9343 100644
--- a/src/backends/reference/RefWorkloadFactory.cpp
+++ b/src/backends/reference/RefWorkloadFactory.cpp
@@ -172,10 +172,15 @@
 std::unique_ptr<IWorkload> RefWorkloadFactory::CreateDebug(const DebugQueueDescriptor& descriptor,
                                                            const WorkloadInfo& info) const
 {
+    if (IsFloat16(info))
+    {
+        return std::make_unique<RefDebugFloat16Workload>(descriptor, info);
+    }
     if (IsQSymm16(info))
     {
         return std::make_unique<RefDebugQSymm16Workload>(descriptor, info);
     }
+
     return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymm8Workload>(descriptor, info);
 }