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

more pass template

Browse Source
This commit is contained in:
Philippe Tillet
2023-01-06 14:26:06 -08:00
parent b16aeb6541
commit 18c7a72973
9 changed files with 92 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
python/src/triton.cc
View File

@@ -1345,14 +1345,18 @@ void init_triton_ir(py::module &&m) {
mlir::createTritonGPUCombineOpsPass(computeCapability));
})
.def("add_tritongpu_optimize_load_convert_pass",
[](mlir::PassManager &self) {
self.addPass(mlir::createTritonGPUOptimizeLoadConvertPass());
})
[](mlir::PassManager &self) {
self.addPass(mlir::createTritonGPUOptimizeLoadConvertPass());
})
.def("add_tritongpu_sink_conversions_from_shared_pass",
[](mlir::PassManager &self) {
self.addPass(
mlir::createTritonGPUSinkConversionsFromSharedPass());
})
[](mlir::PassManager &self) {
self.addPass(mlir::createTritonGPUSinkConversionsFromSharedPass());
})
.def("add_tritongpu_decompose_conversions_to_dot_operand_pass",
[](mlir::PassManager &self) {
self.addPass(
mlir::createTritonGPUDecomposeConversionsToDotOperandPass());
})
.def("add_triton_gpu_to_llvm",
[](mlir::PassManager &self) {
self.addPass(mlir::triton::createConvertTritonGPUToLLVMPass());