Fix ret::reference issue

2022-04-05 16:09:09 +08:00
parent c7ad928e60
commit d7fbddc7d4
1 changed files with 1 additions and 1 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -888,7 +888,7 @@ void init_triton_ir(py::module &&m) {
      }, ret::reference)
      .def("create_block_with_parent", [](mlir::OpBuilder &self, mlir::Region &parent) -> mlir::Block* {
        return self.createBlock(&parent);
-      })
+      }, ret::reference)
      .def("new_block", [](mlir::OpBuilder &self) -> mlir::Block* {
        return new mlir::Block();
      }, ret::reference)