35 lines
789 B
CMake
35 lines
789 B
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_onnx_mlir_library(OMONNXToSpatial
|
|
Math/Gemm.cpp
|
|
Math/Conv.cpp
|
|
Math/ExperimentalConv.cpp
|
|
Math/ExperimentalGemm.cpp
|
|
NN/Pooling.cpp
|
|
NN/ExperimentalPooling.cpp
|
|
NN/ReduceMean.cpp
|
|
Tensor/ONNXConcatToTensorConcat.cpp
|
|
Tensor/RemoveUnusedHelperOps.cpp
|
|
Utils/SpatialReducer.cpp
|
|
Utils/WeightSubdivider.cpp
|
|
Utils/AnnotateReplication.cpp
|
|
ONNXToSpatialPass.hpp
|
|
ONNXToSpatialPass.cpp
|
|
ONNXToSpatialCommon.cpp
|
|
|
|
DEPENDS
|
|
ONNXToSpatialIncGen
|
|
|
|
LINK_LIBS PUBLIC
|
|
OMCompilerOptions
|
|
OMPimCompilerOptions
|
|
OMONNXOps
|
|
SpatialOps
|
|
OMPIMCommon
|
|
|
|
ACCEL_INCLUDE_DIRS PRIVATE
|
|
${PIM_INCLUDE_PATH}
|
|
)
|