【new ir】add ir pybind api

paddle/fluid/pybind/ir.cc

@@ -64,6 +64,8 @@ void BindProgram(py::module *m) {
 void BindBlock(py::module *m) {
   py::class_<Block> block(*m, "Block");
   block.def("front", &Block::front, return_value_policy::reference)
+      .def("get_program",
+           [](Block &self) { return self.GetParentOp()->GetParentProgram(); })
       .def("get_ops",
            [](Block &self) -> py::list {
              py::list op_list;
@@ -81,9 +83,11 @@ void BindBlock(py::module *m) {
 void BindOperation(py::module *m) {
   py::class_<Operation> op(*m, "Operation");
   op.def("name", &Operation::name)
-      .def("get_parent", &Operation::GetParent, return_value_policy::reference)
+      .def("get_block", &Operation::GetParent, return_value_policy::reference)
       .def("num_results", &Operation::num_results)
+      .def("num_operands", &Operation::num_operands)
       .def("result", &Operation::result)
+      .def("operand", &Operation::op_operand)
       .def("operands",
            [](Operation &self) -> py::list {
              py::list op_list;

test/ir/new_ir/test_ir_pybind.py

@@ -41,6 +41,12 @@ def test_program(self):
         newir_program = get_ir_program()
         newir_program.print()
 
+        block = newir_program.block()
+        program = block.get_program()
+        program.print()
+
+        self.assertEqual(newir_program, program)
+
     def test_block(self):
         newir_program = get_ir_program()
         block = newir_program.block()
@@ -57,7 +63,7 @@ def test_operation(self):
         matmul_op = newir_program.block().get_ops()[1]
         add_op = newir_program.block().get_ops()[2]
         tanh_op = newir_program.block().get_ops()[3]
-        parent_block = tanh_op.get_parent()
+        parent_block = tanh_op.get_block()
         parent_ops_num = len(parent_block.get_ops())
         self.assertTrue(parent_ops_num, 4)
         self.assertTrue(tanh_op.num_results(), 1)