[Analysis] Added Axis Info Analysis (#8)
This commit is contained in:
@@ -10,10 +10,14 @@ add_llvm_executable(triton-opt triton-opt.cpp)
|
||||
# TODO: what's this?
|
||||
# llvm_update_compile_flags(triton-opt)
|
||||
target_link_libraries(triton-opt PRIVATE
|
||||
TritonAnalysis
|
||||
TritonTransforms
|
||||
TritonGPUTransforms
|
||||
${dialect_libs}
|
||||
${conversion_libs}
|
||||
# tests
|
||||
TritonTestAnalysis
|
||||
# MLIR core
|
||||
MLIROptLib
|
||||
MLIRPass
|
||||
MLIRTransforms
|
||||
|
@@ -10,11 +10,17 @@
|
||||
#include "mlir/InitAllPasses.h"
|
||||
#include "mlir/Support/MlirOptMain.h"
|
||||
|
||||
namespace mlir{
|
||||
namespace test{
|
||||
void registerTestAlignmentPass();
|
||||
}
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
mlir::registerAllPasses();
|
||||
mlir::registerTritonPasses();
|
||||
mlir::registerTritonGPUPasses();
|
||||
mlir::test::registerTestAlignmentPass();
|
||||
mlir::triton::registerConvertTritonToTritonGPUPass();
|
||||
|
||||
// TODO: register Triton & TritonGPU passes
|
||||
|
Reference in New Issue
Block a user