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

create shr

Browse Source
This commit is contained in:
Yan Da
2022-03-26 16:41:49 +08:00
parent ba16116f96
commit c53f3486e4
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
python/src/triton.cc
View File

@@ -950,12 +950,18 @@ void init_triton_ir(py::module &&m) {
mlir::Value(self.create<mlir::arith::ShLIOp>(loc, unwrap(lhs), unwrap(rhs)))
);
})
// .def("create_lshr", &ir::builder::create_lshr, ret::reference,
// py::arg("lhs"), py::arg("rhs"),
// py::arg("has_nuw")=false, py::arg("has_nsw")=false)
// .def("create_ashr", &ir::builder::create_ashr, ret::reference,
// py::arg("lhs"), py::arg("rhs"),
// py::arg("has_nuw")=false, py::arg("has_nsw")=false)
.def("create_lshr", [](mlir::OpBuilder &self, MlirValue &lhs, MlirValue &rhs) -> MlirValue {
auto loc = self.getUnknownLoc();
return wrap(
mlir::Value(self.create<mlir::arith::ShRUIOp>(loc, unwrap(lhs), unwrap(rhs)))
);
})
.def("create_ashr", [](mlir::OpBuilder &self, MlirValue &lhs, MlirValue &rhs) -> MlirValue {
auto loc = self.getUnknownLoc();
return wrap(
mlir::Value(self.create<mlir::arith::ShRSIOp>(loc, unwrap(lhs), unwrap(rhs)))
);
})
// GEP
.def("create_gep", [](mlir::OpBuilder &self, MlirValue &ptr, MlirValue &offset) -> MlirValue {
auto loc = self.getUnknownLoc();