patch rms norm recompose pass

Differential Revision: D69090807

Author: cccclai

backends/qualcomm/_passes/recompose_rms_norm.py

@@ -34,7 +34,7 @@ def _get_gamma_node(self, output_node):
 
     def call(self, graph_module: torch.fx.GraphModule):
         graph = graph_module.graph
-        partitions = get_source_partitions(graph, [torch.nn.RMSNorm])
+        partitions = get_source_partitions(graph, [torch.nn.RMSNorm, torch.ops.aten.rms_norm.default])
         for _, src_partitions in partitions.items():
             for src_partition in src_partitions:
                 input_len = len(src_partition.input_nodes)