overload add()

Author: PragmaTwice

mlir/lib/Bindings/Python/Pass.cpp

@@ -160,7 +160,7 @@ void mlir::python::populatePassManagerSubmodule(nb::module_ &m) {
           "Add textual pipeline elements to the pass manager. Throws a "
           "ValueError if the pipeline can't be parsed.")
       .def(
-          "add_python_pass",
+          "add",
           [](PyPassManager &passManager, const nb::callable &run,
              std::optional<std::string> &name, const std::string &argument,
              const std::string &description, const std::string &opName) {

mlir/test/python/python_pass.py

@@ -78,10 +78,10 @@ def __call__(self, m):
 
         # CHECK: hello from pass 1!!!
         # CHECK-LABEL: Dump After custom_pass_1
-        pm.add_python_pass(custom_pass_1)
+        pm.add(custom_pass_1)
         # CHECK-LABEL: Dump After CustomPass2
         # CHECK: arith.muli
-        pm.add_python_pass(custom_pass_2, "CustomPass2")
+        pm.add(custom_pass_2, "CustomPass2")
         # CHECK-LABEL: Dump After ArithToLLVMConversionPass
         # CHECK: llvm.mul
         pm.add("convert-arith-to-llvm")