add_onnx_mlir_dialect(Spatial spat)
add_onnx_mlir_dialect_doc(spat Spatial.td)

set(LLVM_TARGET_DEFINITIONS Spatial.td)
mlir_tablegen(SpatialEnums.hpp.inc -gen-enum-decls "-I${ONNX_MLIR_SRC_ROOT}")
mlir_tablegen(SpatialEnums.cpp.inc -gen-enum-defs "-I${ONNX_MLIR_SRC_ROOT}")
add_public_tablegen_target(OMSpatialEnumsIncGen)

mlir_tablegen(SpatialLayoutInterface.hpp.inc -gen-op-interface-decls "-I${ONNX_MLIR_SRC_ROOT}")
mlir_tablegen(SpatialLayoutInterface.cpp.inc -gen-op-interface-defs "-I${ONNX_MLIR_SRC_ROOT}")
add_public_tablegen_target(OMSpatialLayoutInterfaceIncGen)

add_pim_library(SpatialOps
  SpatialOps.cpp
  SpatialOpsAsm.cpp
  SpatialOpsVerify.cpp
  SpatialOpsCanonicalization.cpp
  ${PIM_SRC_ROOT}/Conversion/ONNXToSpatial/CompileTime.cpp
  Passes/Transforms/MergeComputeNodes/Scheduling/ComputeGraph.cpp
  Passes/Transforms/MergeComputeNodes/Scheduling/ComputeInstanceUtils.cpp
  Passes/Transforms/MergeComputeNodes/DeferredCommunicationPlanning.cpp
  Passes/Transforms/MergeComputeNodes/DeferredProjectionAnalysis.cpp
  Passes/Transforms/MergeComputeNodes/DeferredTransferPlanning.cpp
  Passes/Transforms/MergeComputeNodes/DeferredCommunicationScheduling.cpp
  Passes/Transforms/MergeComputeNodes/DeferredBoundaryPlanning.cpp
  Passes/Transforms/MergeComputeNodes/DeferredCommunicationDeadlock.cpp
  Passes/Transforms/MergeComputeNodes/DeferredBoundaryRealization.cpp
  Passes/Transforms/MergeComputeNodes/DeferredResultRealization.cpp
  Passes/Transforms/MergeComputeNodes/DeferredCommunicationRealization.cpp
  Passes/Transforms/MergeComputeNodes/ScheduledSpatialPasses.cpp
  Passes/Transforms/MergeComputeNodes/ScheduledComputeMaterialization.cpp
  Passes/Transforms/MergeComputeNodes/ScheduledComputePlanning.cpp
  Passes/Transforms/MergeComputeNodes/ScheduledComputeReport.cpp
  Passes/Transforms/MergeComputeNodes/ScheduledComputeVerification.cpp
  Passes/Transforms/MergeComputeNodes/SpatialDataflowCsvExporter.cpp
  Passes/Transforms/MergeComputeNodes/Scheduling/MergeSchedulingAnalysis.cpp
  Passes/Transforms/MergeComputeNodes/Scheduling/PeftScheduler.cpp
  Passes/Transforms/MergeComputeNodes/Scheduling/PipelineScheduling.cpp
  Passes/Transforms/TrivialGraphComputeMergePass.cpp

  EXCLUDE_FROM_OM_LIBS

  DEPENDS
  OMONNXIncGen
  OMSpatialIncGen
  OMSpatialEnumsIncGen
  OMSpatialLayoutInterfaceIncGen

  LINK_LIBS PUBLIC
  MLIRIR
  MLIRBufferizationDialect
  MLIRBufferizationTransforms
  OMMlirDialects
  OMONNXOps
  PimOps
)