debug_bmm_to_matmul

Author: Min Guo

backends/qualcomm/_passes/convert_bmm_to_matmul.py

@@ -50,9 +50,13 @@ def call(self, graph_module: torch.fx.GraphModule):
         )
         for _, src_partitions in partitions.items():
             for src_partition in src_partitions:
+                print("partition node")
+                print(src_partition.nodes)
                 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]

backends/qualcomm/_passes/convert_to_linear.py

@@ -193,6 +193,8 @@ def _convert(self, graph_module: torch.fx.GraphModule):
         partitions = get_source_partitions(graph_module.graph, [torch.nn.Linear])
         for _, src_partitions in partitions.items():
             for src_partition in src_partitions:
+                print("partition node to linear")
+                print(src_partition.nodes)
                 op_cnt = Counter(
                     [
                         n.target