Add outline-shape-computation pass. This pass his pass outlines the shape computation part in high level IR by adding shape.func and populate corresponding mapping information into ShapeMappingAnalysis. Reviewed By: jpienaar Differential Revision: https://reviews.llvm.org/D131810
27 lines
538 B
CMake
27 lines
538 B
CMake
add_mlir_dialect_library(MLIRShapeOpsTransforms
|
|
BufferizableOpInterfaceImpl.cpp
|
|
Bufferize.cpp
|
|
OutlineShapeComputation.cpp
|
|
RemoveShapeConstraints.cpp
|
|
ShapeToShapeLowering.cpp
|
|
|
|
ADDITIONAL_HEADER_DIRS
|
|
${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/ShapeOps/Transforms
|
|
|
|
DEPENDS
|
|
MLIRShapeTransformsIncGen
|
|
)
|
|
|
|
target_link_libraries(MLIRShapeOpsTransforms
|
|
PUBLIC
|
|
MLIRArithDialect
|
|
MLIRBufferizationDialect
|
|
MLIRBufferizationTransforms
|
|
MLIRIR
|
|
MLIRMemRefDialect
|
|
MLIRPass
|
|
MLIRShapeDialect
|
|
MLIRSupport
|
|
MLIRTransforms
|
|
)
|