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

1
include/triton/ir/Dialect.h
View File

@@ -6,6 +6,7 @@
#include "mlir/Interfaces/ControlFlowInterfaces.h" #include "mlir/Interfaces/ControlFlowInterfaces.h"
#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/Dialect/StandardOps/IR/Ops.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h" #include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Dialect/SCF/SCF.h"
// #include "mlir/Dialect/ControlFlow/IR/ControlFlow.h" // #include "mlir/Dialect/ControlFlow/IR/ControlFlow.h"
#include "triton/ir/Dialect.h.inc" #include "triton/ir/Dialect.h.inc"

1
lib/ir/CMakeLists.txt
View File

@@ -17,4 +17,5 @@ add_mlir_dialect_library(TritonIR
MLIRStandard MLIRStandard
MLIRTensor MLIRTensor
MLIRSCF
) )

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"); throw std::runtime_error("invalid function type");
}) })
// // Structured control flow // Structured control flow
// .def("create_scf_for", [](mlir::OpBuilder &self) { .def("create_for", [](mlir::OpBuilder &self, MlirValue &lb, MlirValue &ub,
// return self.create<mlir::scf::ForOp>(/*fill this*/); MlirValue &step) {
// }) auto loc = self.getUnknownLoc();
// .def("create_scf_yield") return wrap(
// .def("create_scf_if") self.create<mlir::scf::ForOp>(loc, unwrap(lb), unwrap(ub), unwrap(step))
// .def("create_scf_while") );
})
// .def("create_yield")
// .def("create_if")
// .def("create_while")
// miscellious // miscellious
.def("create_make_range", [](mlir::OpBuilder &self, int start, int end) -> MlirValue { .def("create_make_range", [](mlir::OpBuilder &self, int start, int end) -> MlirValue {