[aotinductor] add no weight change version of fuse_parallel_linear (p…

…ytorch#115791)

Summary: We need a new version of fuse_parallel_linear w/o creating new weights for real-time update.

Reviewed By: khabinov

Differential Revision: D52128296

Pull Request resolved: pytorch#115791
Approved by: https://github.com/khabinov

torch/fx/passes/pass_manager.py

@@ -228,6 +228,16 @@ def remove_pass(self, _passes: List[Callable]):
         self.passes = passes_left
         self._validated = False
 
+    def replace_pass(self, _target, _replacement):
+        passes_left = []
+        for ps in self.passes:
+            if ps.__name__ == _target.__name__:
+                passes_left.append(_replacement)
+            else:
+                passes_left.append(ps)
+        self.passes = passes_left
+        self._validated = False
+
     def validate(self):
         """
         Validates that current pass schedule defined by `self.passes` is valid