commit 5bda97349eb99151a61ab787a33e9c224ca215be
author Teresa Charlin <teresa.charlinreyes@arm.com> Thu Feb 08 18:46:38 2024 +0000
committer David Monahan <david.monahan@arm.com> Thu Feb 08 20:32:12 2024 +0000
tree 1120735b62fdee950f0e07f2ba0c3d08963ed849
parent 20dda37357ac0d02550f4421de6c8bfe44304f90

IVGCVSW-8276 GpuFsa Op: Add MatMul

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ib95eb0fd71106e684cb7652917b8de9f0ac73f9c

src/backends/gpuFsa/GpuFsaLayerSupport.cpp

diff --git a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
index 1ee80c9..85fb03a 100644
--- a/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
+++ b/src/backends/gpuFsa/GpuFsaLayerSupport.cpp
@@ -10,6 +10,7 @@
 #include <armnn/utility/PolymorphicDowncast.hpp>
 
 #if defined(ARMCOMPUTEGPUFSA_ENABLED)
+#include "layers/GpuFsaBatchMatMul.hpp"
 #include "layers/GpuFsaCast.hpp"
 #include "layers/GpuFsaConvolution2d.hpp"
 #include "layers/GpuFsaDepthwiseConvolution2d.hpp"
@@ -76,6 +77,22 @@
 
     switch (type)
     {
+        case LayerType::BatchMatMul:
+        {
+            if (infos.size() != 3)
+            {
+                throw InvalidArgumentException("Invalid number of BatchMatMul TensorInfos. "
+                                               "TensorInfos should be of format: {input0, input1 output}.");
+            }
+
+            auto desc = PolymorphicDowncast<const BatchMatMulDescriptor*>(&descriptor);
+
+            FORWARD_LAYER_VALIDATE_FUNC(GpuFsaBatchMatMulValidate,
+                                        reasonIfUnsupported,
+                                        infos[0],
+                                        infos[1],
+                                        *desc);
+        }
         case LayerType::Cast:
         {
             if (infos.size() != 2)