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/GpuFsaLayerSupport.cpp

diff --git a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
index e98275c..2e5c7d5 100644
--- a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
+++ b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
@@ -13,6 +13,7 @@
 #include "layers/GpuFsaConvolution2d.hpp"
 #include "layers/GpuFsaDepthwiseConvolution2d.hpp"
 #include "layers/GpuFsaElementwiseBinaryAdd.hpp"
+#include "layers/GpuFsaElementwiseBinarySub.hpp"
 #endif
 
 #include <vector>
@@ -144,6 +145,13 @@
                                             infos[0],
                                             infos[1]);
             }
+            else if (desc->m_Operation == BinaryOperation::Sub)
+            {
+                FORWARD_LAYER_VALIDATE_FUNC(GpuFsaElementwiseBinarySubValidate,
+                                            reasonIfUnsupported,
+                                            infos[0],
+                                            infos[1]);
+            }
             else
             {
                 throw InvalidArgumentException("Invalid ElementwiseBinary BinaryOperation operation.");