Skip to content

Commit 9841e54

Browse files
Arm backend: Fix error not raised in TosaArg class (#8875)
Signed-off-by: Adrian Lundell <[email protected]>
1 parent ce715c5 commit 9841e54

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

backends/arm/tosa_mapping.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,10 @@ def __init__(self, argument: Any) -> None:
107107
if isinstance(argument, (int, float)):
108108
self.__process_number(argument)
109109
return
110+
if isinstance(argument, torch.dtype):
111+
# Dtype is parsed from fake tensor
112+
return
110113

111-
RuntimeError(
114+
raise RuntimeError(
112115
f"Unhandled node input argument: {argument}, of type {type(argument)}"
113116
)

0 commit comments

Comments
 (0)