gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Expose module.print in pybind (#604)

Browse Source
This commit is contained in:
Sharad Vikram
2022-07-29 21:36:08 -07:00
committed by GitHub
parent 923d468187
commit 968f59027e
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/src/triton.cc
View File

@@ -758,6 +758,9 @@ void init_triton_ir(py::module &&m) {
.def("has_function", &ir::module::has_function) .def("has_function", &ir::module::has_function)
.def("get_function", &ir::module::get_function, ret::reference) .def("get_function", &ir::module::get_function, ret::reference)
.def("get_or_insert_function", &ir::module::get_or_insert_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("reset_ret_ty", &ir::module::reset_ret_ty)
.def("set_instr_metadata", [](ir::module *self, const std::string &name, ir::value *value) { .def("set_instr_metadata", [](ir::module *self, const std::string &name, ir::value *value) {
const auto metadatas = self->get_metadatas(); const auto metadatas = self->get_metadatas();