This is an archive of the discontinued LLVM Phabricator instance.

Differential D103715

Add memref.dim canonicalization patterns to TilingCanonicalizationPatterns
ClosedPublic

Authored by asaadaldien on Jun 4 2021, 1:33 PM.

Details

Reviewers
mravishankar
nicolasvasilache
Commits
rGa52959401d95: Add memref.dim canonicalization patterns to TilingCanonicalizationPatterns
Summary

Otherwise tiled and padded linalg op will be alive (after distribution).

Diff Detail

Revision Contents

PathSize
mlir/
lib/
Dialect/
Linalg/
Transforms/
1 line

Diff 349958

mlir/lib/Dialect/Linalg/Transforms/Tiling.cpp

Show First 20 Lines • Show All 408 Lines • ▼ Show 20 Linesvoid mlir::linalg::populateLinalgTilingCanonicalizationPatterns(
AffineMaxOp::getCanonicalizationPatterns(patterns, ctx); AffineMaxOp::getCanonicalizationPatterns(patterns, ctx);
scf::ForOp::getCanonicalizationPatterns(patterns, ctx); scf::ForOp::getCanonicalizationPatterns(patterns, ctx);
scf::ParallelOp::getCanonicalizationPatterns(patterns, ctx); scf::ParallelOp::getCanonicalizationPatterns(patterns, ctx);
ConstantIndexOp::getCanonicalizationPatterns(patterns, ctx); ConstantIndexOp::getCanonicalizationPatterns(patterns, ctx);
SubTensorOp::getCanonicalizationPatterns(patterns, ctx); SubTensorOp::getCanonicalizationPatterns(patterns, ctx);
memref::SubViewOp::getCanonicalizationPatterns(patterns, ctx); memref::SubViewOp::getCanonicalizationPatterns(patterns, ctx);
tensor::CastOp::getCanonicalizationPatterns(patterns, ctx); tensor::CastOp::getCanonicalizationPatterns(patterns, ctx);
memref::ViewOp::getCanonicalizationPatterns(patterns, ctx); memref::ViewOp::getCanonicalizationPatterns(patterns, ctx);
memref::DimOp::getCanonicalizationPatterns(patterns, ctx);
CanonicalizationPatternList< CanonicalizationPatternList<
#define GET_OP_LIST #define GET_OP_LIST
#include "mlir/Dialect/Linalg/IR/LinalgStructuredOps.cpp.inc" #include "mlir/Dialect/Linalg/IR/LinalgStructuredOps.cpp.inc"
>::insert(patterns); >::insert(patterns);
} }
/// Populate the given list with patterns that apply Linalg tiling. /// Populate the given list with patterns that apply Linalg tiling.
static void insertTilingPatterns(RewritePatternSet &patterns, static void insertTilingPatterns(RewritePatternSet &patterns,
▲ Show 20 Lines • Show All 88 Lines • Show Last 20 Lines