NiccoloN
909c4acfdd
Validate Operations / validate-operations (push) Has been cancelled
remove Spatial many ops in favor of tensor ops like in pim
47 lines
1.0 KiB
CMake
47 lines
1.0 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
|
|
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}
|
|
)
|