[FRONTEND] Allow tl.where to select pointers (#595)

2022-07-22 00:54:27 +08:00
parent af85f5fa46
commit f28caddbf8
3 changed files with 58 additions and 18 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -878,6 +878,7 @@ void init_triton_ir(py::module &&m) {
      .def("create_int_cast", &ir::builder::create_int_cast, ret::reference)
      .def("create_downcast", &ir::builder::create_downcast, ret::reference)
      .def("create_int_to_ptr", &ir::builder::create_int_to_ptr, ret::reference)
+      .def("create_ptr_to_int", &ir::builder::create_ptr_to_int, ret::reference)
      // phi
      .def("create_phi", &ir::builder::create_phi, ret::reference)
      // Binary instructions