register conversion in triton-opt

This commit is contained in:
Yan Da
2022-06-07 19:33:51 +08:00
parent 0e11435448
commit 560e29229b
4 changed files with 13 additions and 1 deletions

View File

@@ -4,6 +4,8 @@
#include "triton/Dialect/Triton/Transforms/Passes.h"
#include "triton/Dialect/TritonGPU/Transforms/Passes.h"
#include "triton/Conversion/Passes.h"
#include "mlir/IR/Dialect.h"
#include "mlir/InitAllPasses.h"
#include "mlir/Support/MlirOptMain.h"
@@ -13,6 +15,7 @@ int main(int argc, char **argv) {
mlir::registerAllPasses();
mlir::registerTritonPasses();
mlir::registerTritonGPUPasses();
mlir::triton::registerConvertTritonToTritonGPUPass();
// TODO: register Triton & TritonGPU passes
mlir::DialectRegistry registry;