Init TritonGPU to LLVM dialect conversion (#32)

* 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

View File

@@ -22,6 +22,7 @@ int main(int argc, char **argv) {
mlir::registerTritonGPUPasses();
mlir::test::registerTestAlignmentPass();
mlir::triton::registerConvertTritonToTritonGPUPass();
mlir::triton::registerConvertTritonGPUToLLVMPass();
// TODO: register Triton & TritonGPU passes
mlir::DialectRegistry registry;