26 lines · plain
1set(LLVM_OPTIONAL_SOURCES2 AllExtensions.cpp3 ShardingExtensions.cpp4 )5 6add_mlir_extension_library(MLIRTensorShardingExtensions7 ShardingExtensions.cpp8 9 ADDITIONAL_HEADER_DIRS10 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor/Extensions11 12 LINK_LIBS PUBLIC13 MLIRTensorDialect14 MLIRIR15 MLIRShardingInterface16 )17 18add_mlir_extension_library(MLIRTensorAllExtensions19 AllExtensions.cpp20 21 ADDITIONAL_HEADER_DIRS22 ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor/Extensions23 24 LINK_LIBS PUBLIC25 MLIRTensorShardingExtensions26 )