From 94cc3ec7a780eac38f0347c15c313316d2151697 Mon Sep 17 00:00:00 2001 From: winskuo-quic Date: Tue, 11 Feb 2025 15:25:25 +0800 Subject: [PATCH] raise error when new pattern found --- backends/qualcomm/_passes/convert_bmm_to_matmul.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backends/qualcomm/_passes/convert_bmm_to_matmul.py b/backends/qualcomm/_passes/convert_bmm_to_matmul.py index f0773f69833..480a985a90a 100644 --- a/backends/qualcomm/_passes/convert_bmm_to_matmul.py +++ b/backends/qualcomm/_passes/convert_bmm_to_matmul.py @@ -52,7 +52,9 @@ def call(self, graph_module: torch.fx.GraphModule): for src_partition in src_partitions: op_cnt = Counter([n.target for n in src_partition.nodes]) if op_cnt not in self.patterns: - continue + raise AssertionError( + "Found a new pattern needed be converted to linear op" + ) inputs = src_partition.input_nodes bmm_node = [n for n in src_partition.nodes if n.target == self.bmm][0]