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

Init TritonGPU to LLVM dialect conversion (#32)

Browse Source 
* add toLLVM pass

* update num-warps setting in mlir
This commit is contained in:
Yan Chunwei
2022-08-04 10:15:45 +08:00
committed by GitHub
parent 3236642e8f
commit b988bae813
9 changed files with 328 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/src/triton.cc
View File

@@ -11,6 +11,7 @@
#include "mlir/Pass/PassManager.h"
#include "mlir/Transforms/Passes.h"
#include "triton/Conversion/TritonGPUToLLVM/TritonGPUToLLVM.h"
#include "triton/Conversion/TritonToTritonGPU/TritonToTritonGPU.h"
#include "triton/Dialect/Triton/IR/Dialect.h"
#include "triton/Dialect/Triton/IR/Types.h"
@@ -1634,8 +1635,12 @@ void init_triton_ir(py::module &&m) {
[](mlir::PassManager &self) {
self.addPass(mlir::createTritonGPUCombineOpsPass());
})
.def("add_triton_gpu_verifier_pass", [](mlir::PassManager &self) {
self.addPass(mlir::createTritonGPUVerifier());
.def("add_triton_gpu_verifier_pass",
[](mlir::PassManager &self) {
self.addPass(mlir::createTritonGPUVerifier());
})
.def("triton_gpu_to_llvm", [](mlir::PassManager &self) {
self.addPass(mlir::triton::createConvertTritonGPUToLLVMPass());
});
}