Skip to content

Commit f7977a6

Browse files
Arm backend: Reject INT16-matmuls on U55 (pytorch#18365)
Make sure that INT16 matmuls are rejected for U55. cc @digantdesai @freddan80 @per @zingo @mansnils @Sebastian-Larsson @robell Signed-off-by: Oscar Andersson <oscar.andersson@arm.com>
1 parent 60d57e5 commit f7977a6

2 files changed

Lines changed: 179 additions & 76 deletions

File tree

backends/arm/operator_support/ethos_u55_support.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -157,10 +157,10 @@ def is_node_supported( # noqa: C901
157157
):
158158
for input_node in node.all_input_nodes:
159159
dtype = _try_determine_dtype(input_node)
160-
if dtype is not None and dtype not in (torch.int8, torch.int16):
160+
if dtype is not None and dtype not in (torch.int8,):
161161
self.reporter.report_reject(
162-
input_node,
163-
f"Input {input_node.name} has unsupported dtype {dtype} (Supports i8, i16).",
162+
node,
163+
f"Input {input_node.name} has unsupported dtype {dtype} (Supports int8).",
164164
)
165165
return False
166166

0 commit comments

Comments
 (0)