43 lines · plain
1set(LLVM_OPTIONAL_SOURCES2 AllExtensions.cpp3 InlinerExtension.cpp4 ShardingExtensions.cpp5 )6 7add_mlir_extension_library(MLIRFuncInlinerExtension8 InlinerExtension.cpp9 10 ADDITIONAL_HEADER_DIRS11 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Func/Extensions12 13 LINK_LIBS PUBLIC14 MLIRControlFlowDialect15 MLIRInferTypeOpInterface16 MLIRIR17 MLIRFuncDialect18 )19 20add_mlir_extension_library(MLIRFuncShardingExtensions21 ShardingExtensions.cpp22 23 ADDITIONAL_HEADER_DIRS24 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Func/Extensions25 26 LINK_LIBS PUBLIC27 MLIRFuncDialect28 MLIRIR29 MLIRShardingInterface30 )31 32 33add_mlir_extension_library(MLIRFuncAllExtensions34 AllExtensions.cpp35 36 ADDITIONAL_HEADER_DIRS37 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Func/Extensions38 39 LINK_LIBS PUBLIC40 MLIRFuncInlinerExtension41 MLIRFuncShardingExtensions42 )43