From d7fbddc7d463c192073e54f9d1e269fc9f70f4df Mon Sep 17 00:00:00 2001
From: Yan Da <dyanab@connect.ust.hk>
Date: Tue, 5 Apr 2022 16:09:09 +0800
Subject: [PATCH] Fix ret::reference issue

---
 python/src/triton.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/src/triton.cc b/python/src/triton.cc
index c0bc61693..9dae11ba5 100644
--- 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)