From 0427b43ae785f2620afaf9c1a1ef5dc8cbb8aa2c Mon Sep 17 00:00:00 2001
From: Adrian Lundell <adrian.lundell@arm.com>
Date: Fri, 28 Feb 2025 12:34:52 +0100
Subject: [PATCH] Arm backend: Fix error not raised in TosaArg class

Signed-off-by: Adrian Lundell <adrian.lundell@arm.com>
Change-Id: Ia92e2bd28967f7c0e0ee351c51ff4b6bcc80bf0c
---
 backends/arm/tosa_mapping.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/backends/arm/tosa_mapping.py b/backends/arm/tosa_mapping.py
index 9a8b6b2c35d..292d6209bb3 100644
--- a/backends/arm/tosa_mapping.py
+++ b/backends/arm/tosa_mapping.py
@@ -107,7 +107,10 @@ def __init__(self, argument: Any) -> None:
         if isinstance(argument, (int, float)):
             self.__process_number(argument)
             return
+        if isinstance(argument, torch.dtype):
+            # Dtype is parsed from fake tensor
+            return
 
-        RuntimeError(
+        raise RuntimeError(
             f"Unhandled node input argument: {argument}, of type {type(argument)}"
         )