update mma encoding & triton-opt

bin/CMakeLists.txt

@@ -4,17 +4,15 @@ get_property(conversion_libs GLOBAL PROPERTY MLIR_CONVERSION_LIBS)
 add_llvm_executable(triton-opt triton-opt.cpp)
 
 # TODO: what's this?
-llvm_update_compile_flags(triton-opt)
+# llvm_update_compile_flags(triton-opt)
 target_link_libraries(triton-opt PRIVATE
+  TritonTransforms
+  TritonGPUTransforms
   ${dialect_libs}
   ${conversion_libs}
   MLIROptLib
-
-  TritonIR
-  TritonTransforms
-  
-  TritonGPUIR
-  TritonGPUTransforms
+  MLIRPass
+  MLIRTransforms
 )
 
 mlir_check_all_link_libraries(triton-opt)

bin/triton-opt.cpp

@@ -1,6 +1,9 @@
 #include "triton/Dialect/Triton/IR/Dialect.h"
 #include "triton/Dialect/TritonGPU/IR/Dialect.h"
 
+#include "triton/Dialect/Triton/Transforms/Passes.h"
+#include "triton/Dialect/TritonGPU/Transforms/Passes.h"
+
 #include "mlir/IR/Dialect.h"
 #include "mlir/InitAllPasses.h"
 #include "mlir/Support/MlirOptMain.h"
@@ -8,7 +11,10 @@
 
 int main(int argc, char **argv) {
   mlir::registerAllPasses();
+  mlir::registerTritonPasses();
+  mlir::registerTritonGPUPasses();
 
+  // TODO: register Triton & TritonGPU passes
   mlir::DialectRegistry registry;
   registry.insert<mlir::triton::TritonDialect,
                   mlir::triton::gpu::TritonGPUDialect,