lapis-opt: register extensions needed for inlining (#53)

createInlinerPass() can be used in lapis pipelines
where it didn't work before.

Signed-off-by: Brian Kelley <bmkelle@sandia.gov>

Author: brian-kelley

mlir/tools/lapis-opt/CMakeLists.txt

@@ -2,16 +2,25 @@ get_property(dialect_libs GLOBAL PROPERTY MLIR_DIALECT_LIBS)
 
 add_llvm_executable(lapis-opt lapis-opt.cpp)
 llvm_update_compile_flags(lapis-opt)
+
 target_link_libraries(lapis-opt
   PRIVATE
   ${dialect_libs}
   ${translation_libs}
   ${test_libs}
-  MLIRIR
+
+  MLIRAffineAnalysis
+  MLIRAnalysis
+  MLIRCastInterfaces
+  MLIRDialect
+  MLIROptLib
   MLIRParser
-  MLIRSPIRVDialect
+  MLIRPass
+  MLIRTransforms
+  MLIRTransformUtils
   MLIRSupport
-  MLIROptLib
+  MLIRIR
+  MLIRFuncAllExtensions
   )
 
 mlir_check_link_libraries(lapis-opt)

mlir/tools/lapis-opt/lapis-opt.cpp

@@ -17,12 +17,14 @@
 #include "lapis/Dialect/PartTensor/Pipelines/Passes.h"
 #include "lapis/Dialect/PartTensor/Transforms/Passes.h"
 #endif
+#include "mlir/InitAllDialects.h"
 #include "mlir/Dialect/Arith/IR/Arith.h"
 #include "mlir/Dialect/Arith/IR/ValueBoundsOpInterfaceImpl.h"
 #include "mlir/Dialect/Arith/Transforms/BufferizableOpInterfaceImpl.h"
 #include "mlir/Dialect/Bufferization/IR/Bufferization.h"
 #include "mlir/Dialect/Bufferization/Transforms/FuncBufferizableOpInterfaceImpl.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
+#include "mlir/Dialect/Func/Extensions/AllExtensions.h"
 #include "mlir/Dialect/Linalg/IR/Linalg.h"
 #include "mlir/Dialect/Linalg/Transforms/AllInterfaces.h"
 #include "mlir/Dialect/Linalg/Transforms/RuntimeOpVerification.h"
@@ -86,6 +88,9 @@ int main(int argc, char **argv) {
   tensor::registerValueBoundsOpInterfaceExternalModels(registry);
   vector::registerBufferizableOpInterfaceExternalModels(registry);
 
+  LLVM::registerInlinerInterface(registry);
+  func::registerAllExtensions(registry);
+
   // Register LAPIS pipelines and passes
 #ifdef ENABLE_PART_TENSOR
   mlir::registerPartTensorPasses();