[BACKEND][CODEGEN] Fix reduce uint (#547)

2022-06-13 16:43:57 -07:00
parent 58c8889235
commit 93209c07e0
6 changed files with 61 additions and 25 deletions
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -571,6 +571,8 @@ void init_triton_ir(py::module &&m) {
      .value("FADD", ir::reduce_inst::FADD)
      .value("MIN", ir::reduce_inst::MIN)
      .value("MAX", ir::reduce_inst::MAX)
+      .value("UMIN", ir::reduce_inst::UMIN)
+      .value("UMAX", ir::reduce_inst::UMAX)
      .value("FMIN", ir::reduce_inst::FMIN)
      .value("FMAX", ir::reduce_inst::FMAX)
      .value("XOR", ir::reduce_inst::XOR);