NiccoloN
59 lines
1.4 KiB
CMake
59 lines
1.4 KiB
CMake
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
|
|
Patterns.cpp
|
|
CompileTime.cpp
|
|
ONNXToSpatialVerifier.cpp
|
|
Patterns/Pre.cpp
|
|
Patterns/Post.cpp
|
|
Patterns/GeneratedConversion.cpp
|
|
Patterns/Math/Conv.cpp
|
|
Patterns/Math/ConvGeometry.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/Flatten.cpp
|
|
Patterns/Tensor/Gather.cpp
|
|
Patterns/Tensor/Resize.cpp
|
|
Patterns/Tensor/Reshape.cpp
|
|
Patterns/Tensor/Slice.cpp
|
|
Patterns/Tensor/Split.cpp
|
|
Patterns/Tensor/Transpose.cpp
|
|
ONNXToSpatialPass.cpp
|
|
SpatialLayoutPlanningPass.cpp
|
|
LowerSpatialPlansPass.cpp
|
|
Common/AttributeUtils.cpp
|
|
Common/BiasAddUtils.cpp
|
|
Common/ComputeRegionBuilder.cpp
|
|
Common/MatrixProductLowering.cpp
|
|
Common/RowStripLayoutUtils.cpp
|
|
Common/ShapeTilingUtils.cpp
|
|
Common/WeightMaterialization.cpp
|
|
|
|
EXCLUDE_FROM_OM_LIBS
|
|
|
|
DEPENDS
|
|
ONNXToSpatialIncGen
|
|
|
|
LINK_LIBS PUBLIC
|
|
MLIRLinalgDialect
|
|
MLIRSCFDialect
|
|
MLIRTosaDialect
|
|
OMCompilerOptions
|
|
OMPimCompilerOptions
|
|
OMONNXOps
|
|
SpatialOps
|
|
OMPimCommon
|
|
|
|
ACCEL_INCLUDE_DIRS PRIVATE
|
|
${PIM_GENERATED_INCLUDE_DIRS}
|
|
)
|