681ae09722
Follow-up to #102598 : as discussed, move tensor sharding implementation into separate tensor extension lib. @sogartar @yaochengji, could you take a look at this PR?
26 lines
544 B
CMake
26 lines
544 B
CMake
set(LLVM_OPTIONAL_SOURCES
|
|
AllExtensions.cpp
|
|
MeshShardingExtensions.cpp
|
|
)
|
|
|
|
add_mlir_extension_library(MLIRTensorMeshShardingExtensions
|
|
MeshShardingExtensions.cpp
|
|
|
|
ADDITIONAL_HEADER_DIRS
|
|
${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor/Extensions
|
|
|
|
LINK_LIBS PUBLIC
|
|
MLIRTensorDialect
|
|
MLIRIR
|
|
MLIRShardingInterface
|
|
)
|
|
|
|
add_mlir_extension_library(MLIRTensorAllExtensions
|
|
AllExtensions.cpp
|
|
|
|
ADDITIONAL_HEADER_DIRS
|
|
${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Tensor/Extensions
|
|
|
|
LINK_LIBS PUBLIC
|
|
MLIRTensorMeshShardingExtensions
|
|
) |