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

Introducing SCF

Browse Source
This commit is contained in:
Yan Da
2022-03-26 17:02:32 +08:00
parent c53f3486e4
commit 0d139ec460
3 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
python/src/triton.cc
View File

@@ -834,13 +834,17 @@ void init_triton_ir(py::module &&m) {
}
throw std::runtime_error("invalid function type");
})
// // Structured control flow
// .def("create_scf_for", [](mlir::OpBuilder &self) {
// return self.create<mlir::scf::ForOp>(/*fill this*/);
// })
// .def("create_scf_yield")
// .def("create_scf_if")
// .def("create_scf_while")
// Structured control flow
.def("create_for", [](mlir::OpBuilder &self, MlirValue &lb, MlirValue &ub,
MlirValue &step) {
auto loc = self.getUnknownLoc();
return wrap(
self.create<mlir::scf::ForOp>(loc, unwrap(lb), unwrap(ub), unwrap(step))
);
})
// .def("create_yield")
// .def("create_if")
// .def("create_while")
// miscellious
.def("create_make_range", [](mlir::OpBuilder &self, int start, int end) -> MlirValue {