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)}" )