Add triton's opt

This commit is contained in:
Yan Da
2022-06-04 22:10:00 +08:00
parent 830fe19d58
commit 55cf9a0a97
3 changed files with 40 additions and 0 deletions

19
bin/triton-opt.cpp Normal file
View File

@@ -0,0 +1,19 @@
#include "triton/Dialect/Triton/IR/Dialect.h"
#include "triton/Dialect/TritonGPU/IR/Dialect.h"
#include "mlir/IR/Dialect.h"
#include "mlir/InitAllPasses.h"
#include "mlir/Support/MlirOptMain.h"
int main(int argc, char **argv) {
mlir::registerAllPasses();
mlir::DialectRegistry registry;
registry.insert<mlir::triton::TritonDialect,
mlir::triton::gpu::TritonGPUDialect>();
return mlir::asMainReturnCode(
mlir::MlirOptMain(argc, argv, "Triton (GPU) optimizer driver\n", registry)
);
}