commit 829e13edbe9487b5a0600688cec6312b867e2f07
author John Mcloughlin <john.mcloughlin@arm.com> Wed Jan 31 11:00:27 2024 +0000
committer John Mcloughlin <john.mcloughlin@arm.com> Wed Jan 31 21:24:23 2024 +0000
tree 2b9176173e73afc1f1cdf62cec334c7ee64e7123
parent e4134fc28ce2a9c751fb22ff84b733d067255849

IVGCVSW-7568 Implement Sub ElementwiseBinary operator GpuFsa

* Added support for Gpu Sub operator
* Added unit tests

Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Change-Id: I1efaa485772a3716e3781566843bd50bd9bab811

src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp

diff --git a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp
index 49ddade..9d4b3b9 100644
--- a/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp
+++ b/src/backends/gpuFsa/test/GpuFsaLayerSupportTests.cpp
@@ -81,4 +81,24 @@
     CHECK(supported);
 }
 
+TEST_CASE("IsLayerSupportedGpuFsaElementWiseBinarySub")
+{
+    TensorInfo input0Info({ 2, 2 }, DataType::Float32);
+    TensorInfo input1Info({ 2, 2 }, DataType::Float32);
+    TensorInfo outputInfo({ 2, 2 }, DataType::Float32);
+
+    ElementwiseBinaryDescriptor desc;
+    desc.m_Operation = BinaryOperation::Sub;
+
+    GpuFsaLayerSupport supportChecker;
+    std::string reasonIfNotSupported;
+    auto supported = supportChecker.IsLayerSupported(LayerType::ElementwiseBinary,
+                                                     {input0Info, input1Info, outputInfo},
+                                                     desc,
+                                                     EmptyOptional(),
+                                                     EmptyOptional(),
+                                                     reasonIfNotSupported);
+    CHECK(supported);
+}
+
 }
\ No newline at end of file