[Triton-MLIR][Backend] Fix reduce conversion and unit tests for int dtypes (#826)

This commit is contained in:
Qingyi Liu
2022-11-01 17:42:59 +08:00
committed by GitHub
parent 031c2ae77b
commit cdc0ec5077
5 changed files with 208 additions and 148 deletions

View File

@@ -87,8 +87,16 @@ void init_triton_ir(py::module &&m) {
.value("FADD", mlir::triton::RedOp::FADD)
.value("MIN", mlir::triton::RedOp::MIN)
.value("MAX", mlir::triton::RedOp::MAX)
.value("UMIN", mlir::triton::RedOp::UMIN)
.value("UMAX", mlir::triton::RedOp::UMAX)
.value("ARGMIN", mlir::triton::RedOp::ARGMIN)
.value("ARGMAX", mlir::triton::RedOp::ARGMAX)
.value("ARGUMIN", mlir::triton::RedOp::ARGUMIN)
.value("ARGUMAX", mlir::triton::RedOp::ARGUMAX)
.value("FMIN", mlir::triton::RedOp::FMIN)
.value("FMAX", mlir::triton::RedOp::FMAX)
.value("ARGFMIN", mlir::triton::RedOp::ARGFMIN)
.value("ARGFMAX", mlir::triton::RedOp::ARGFMAX)
.value("XOR", mlir::triton::RedOp::XOR);
py::enum_<mlir::triton::RMWOp>(m, "ATOMIC_OP")