From bc61cab381a32f8c6714eb2300a4d63f590f2fe6 Mon Sep 17 00:00:00 2001
From: Max Bergfelt <max.bergfelt@arm.com>
Date: Tue, 22 Jul 2025 13:26:34 +0200
Subject: [PATCH] MLBEDSW-10999: Added TOSA reader validation for MUL shift
 rank

Added validation of tensor ranks to the TOSA reader to catch and throw an error on MUL shift rank 0 tensors (scalars) before they are converted to rank 1.

Change-Id: Ia53531b6dbbbb9a9745333958bc7dcdee86c0d8f
Signed-off-by: Max Bergfelt <max.bergfelt@arm.com>
---
 ethosu/regor/tosa/tosa_reader.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/ethosu/regor/tosa/tosa_reader.cpp b/ethosu/regor/tosa/tosa_reader.cpp
index 458955f7..1cb7db61 100644
--- a/ethosu/regor/tosa/tosa_reader.cpp
+++ b/ethosu/regor/tosa/tosa_reader.cpp
@@ -753,6 +753,13 @@ void TosaReader::LoadGraphs(const tosaFb::TosaGraph *model, std::list<GraphBuild
                             "Failed to set OUTPAD attribute on TRANSPOSE_CONV2D");
                     }
                     break;
+                    case tosaFb::Op::MUL:
+                    {
+                        const auto &mulShiftTensor = input_tensors[2];
+                        tosa_assert(shapes.at(mulShiftTensor).count == 1,
+                            ("MUL shift tensor " + mulShiftTensor + " needs to have rank 1.").c_str());
+                    }
+                    break;
                     default:
                         break;
                 }
-- 
GitLab