xref: /llvm-project/mlir/lib/Dialect/Func/Extensions/CMakeLists.txt (revision adbf21f12b3069b2554efb39f2e92c6cf6f24940) 
1set(LLVM_OPTIONAL_SOURCES
2  AllExtensions.cpp
3  InlinerExtension.cpp
4  MeshShardingExtensions.cpp
5  )
6
7add_mlir_extension_library(MLIRFuncInlinerExtension
8  InlinerExtension.cpp
9
10  ADDITIONAL_HEADER_DIRS
11  ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Func/Extensions
12
13  LINK_LIBS PUBLIC
14  MLIRControlFlowDialect
15  MLIRInferTypeOpInterface
16  MLIRIR
17  MLIRFuncDialect
18  )
19
20add_mlir_extension_library(MLIRFuncMeshShardingExtensions
21  MeshShardingExtensions.cpp
22
23  ADDITIONAL_HEADER_DIRS
24  ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Func/Extensions
25
26  LINK_LIBS PUBLIC
27  MLIRFuncDialect
28  MLIRIR
29  MLIRShardingInterface
30  )
31
32
33add_mlir_extension_library(MLIRFuncAllExtensions
34  AllExtensions.cpp
35
36  ADDITIONAL_HEADER_DIRS
37  ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Func/Extensions
38
39  LINK_LIBS PUBLIC
40  MLIRFuncInlinerExtension
41  MLIRFuncMeshShardingExtensions
42  )
43