From 968f59027e74744b2df3d21dc8286594450622ea Mon Sep 17 00:00:00 2001
From: Sharad Vikram <sharad.vikram@gmail.com>
Date: Fri, 29 Jul 2022 21:36:08 -0700
Subject: [PATCH] Expose `module.print` in pybind (#604)

---
 python/src/triton.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/src/triton.cc b/python/src/triton.cc
index 81e1b66fe..940c5bc0a 100644
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -758,6 +758,9 @@ void init_triton_ir(py::module &&m) {
       .def("has_function", &ir::module::has_function)
       .def("get_function", &ir::module::get_function, ret::reference)
       .def("get_or_insert_function", &ir::module::get_or_insert_function, ret::reference)
+      .def("print", [](ir::module *self) {
+          self->print(std::cout);
+      })
       .def("reset_ret_ty", &ir::module::reset_ret_ty)
       .def("set_instr_metadata", [](ir::module *self, const std::string &name, ir::value *value) {
           const auto metadatas = self->get_metadatas();