add_mlir_dialect_library(MLIRShardTransforms
  Simplifications.cpp
  ShardingPropagation.cpp
  Partition.cpp
  Transforms.cpp

  ADDITIONAL_HEADER_DIRS
  ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/Shard

  DEPENDS
  MLIRShardPassIncGen
  MLIRShardingInterface

  LINK_LIBS PUBLIC
  MLIRAffineDialect
  MLIRAffineUtils
  MLIRArithDialect
  MLIRArithUtils
  MLIRControlFlowDialect
  MLIRControlFlowInterfaces
  MLIRFuncDialect
  MLIRFunctionInterfaces
  MLIRIR
  MLIRShardDialect
  MLIRPass
  MLIRSupport
  MLIRTensorDialect
  MLIRTosaShardingInterfaceImpl
)