diff --git a/backends/arm/operators/op_log.py b/backends/arm/operators/op_log.py index 34911075065..9942cbf4702 100644 --- a/backends/arm/operators/op_log.py +++ b/backends/arm/operators/op_log.py @@ -33,7 +33,14 @@ def define_node( inputs: List[TosaArg], output: TosaArg, ) -> None: - assert len(node.all_input_nodes) == 1 - assert inputs[0].dtype == output.dtype == ts.DType.FP32 + if len(node.all_input_nodes) != 1: + raise ValueError( + f"Expected 1 input for {self.target}, got {len(node.all_input_nodes)}" + ) + if inputs[0].dtype != ts.DType.FP32 or output.dtype != ts.DType.FP32: + raise ValueError( + f"Input and output for {self.target} need to be FP32, got input_dtype: " + f"{inputs[0].dtype} and output_dtype: {output.dtype}" + ) tosa_graph.addOperator(ts.TosaOp.Op().LOG, [inputs[0].name], [output.name])