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]