set(LLVM_TARGET_DEFINITIONS ONNXToSpatial.td)
mlir_tablegen(ONNXToSpatial.hpp.inc -gen-rewriters "-I${ONNX_MLIR_SRC_ROOT}")
add_public_tablegen_target(ONNXToSpatialIncGen)

add_pim_library(OMONNXToSpatial
  ConversionPatterns.cpp
  HostFoldability.cpp
  HostLegality.cpp
  PrePatterns.cpp
  PostPatterns.cpp
  Patterns/Math/Conv.cpp
  Patterns/Math/Elementwise.cpp
  Patterns/Math/Gemm.cpp
  Patterns/Math/MatMul.cpp
  Patterns/Math/ReduceMean.cpp
  Patterns/NN/Pool.cpp
  Patterns/NN/Relu.cpp
  Patterns/NN/Sigmoid.cpp
  Patterns/NN/Softmax.cpp
  Patterns/Tensor/Concat.cpp
  Patterns/Tensor/Gather.cpp
  Patterns/Tensor/Resize.cpp
  Patterns/Tensor/Reshape.cpp
  Patterns/Tensor/Split.cpp
  ONNXToSpatialPass.cpp
  Common/ComputeRegionBuilder.cpp
  Common/ShapeTilingUtils.cpp
  Common/WeightMaterialization.cpp

  EXCLUDE_FROM_OM_LIBS

  DEPENDS
  ONNXToSpatialIncGen

  LINK_LIBS PUBLIC
  MLIRSCFDialect
  MLIRTosaDialect
  OMCompilerOptions
  OMPimCompilerOptions
  OMONNXOps
  SpatialOps
  OMPimCommon

  ACCEL_INCLUDE_DIRS PRIVATE
  ${PIM_GENERATED_INCLUDE_DIRS}
)