From b05828a019c5818b743ad755ca56875030f8a333 Mon Sep 17 00:00:00 2001 From: Sebastian Larsson Date: Thu, 11 Jun 2026 14:18:41 +0200 Subject: [PATCH] Arm backend: Fix U55 convolution dtype reject messages Report the actual unsupported input, weight, or bias dtype instead of the convolution node dtype when rejecting U55 convolution partitions. Change-Id: I2e9e3f0363f668af477351ebcfc2c8ef8f1df1aa Signed-off-by: Sebastian Larsson --- backends/arm/operator_support/ethos_u55_support.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backends/arm/operator_support/ethos_u55_support.py b/backends/arm/operator_support/ethos_u55_support.py index 5cd9918b567..574b5aa0bdc 100644 --- a/backends/arm/operator_support/ethos_u55_support.py +++ b/backends/arm/operator_support/ethos_u55_support.py @@ -125,13 +125,13 @@ def is_node_supported( # noqa: C901 ifm_dtype = _try_determine_dtype(ifm) if ifm_dtype is not None and ifm_dtype not in (torch.int8, torch.int16): self.reporter.report_reject( - node, f"Unsupported input dtype {dtype} (Supports i8, i16)." + node, f"Unsupported input dtype {ifm_dtype} (Supports i8, i16)." ) return False weight_dtype = _try_determine_dtype(weight) if weight_dtype is not None and weight_dtype not in (torch.int8,): self.reporter.report_reject( - node, f"Unsupported weight dtype {dtype} (Supports i8)." + node, f"Unsupported weight dtype {weight_dtype} (Supports i8)." ) return False if len(node.all_input_nodes) > 2: @@ -139,7 +139,7 @@ def is_node_supported( # noqa: C901 bias_dtype = _try_determine_dtype(bias) if bias_dtype is not None and bias_dtype not in (torch.int32,): self.reporter.report_reject( - node, f"Unsupported bias dtype {dtype} (Supports i32)." + node, f"Unsupported bias dtype {bias_dtype} (Supports i32)." ) return False