| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 307 struct LinalgMarker { struct
308 explicit LinalgMarker(ArrayRef<Identifier> matchDisjunction = {},
310 LinalgMarker(LinalgMarker &&) = default;
311 LinalgMarker(const LinalgMarker &) = default;
396 LinalgMarker marker = LinalgMarker(),
401 LinalgMarker marker = LinalgMarker(),
409 LinalgMarker marker;
417 LinalgMarker marker = LinalgMarker(),
449 LinalgMarker marker = LinalgMarker(),
450 LinalgMarker fusedOpMarker = LinalgMarker(),
[all …]
|
| D | CodegenStrategy.h | 26 buildRewritePatterns(MLIRContext *context, linalg::LinalgMarker m) = 0;
27 linalg::LinalgMarker marker;
37 buildRewritePatterns(MLIRContext *context, linalg::LinalgMarker m) override { in buildRewritePatterns()
55 buildRewritePatterns(MLIRContext *context, linalg::LinalgMarker m) override { in buildRewritePatterns()
72 buildRewritePatterns(MLIRContext *context, linalg::LinalgMarker m) override { in buildRewritePatterns()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgTransforms.cpp | 94 LinalgMarker(Identifier::get("MEM", ctx), Identifier::get("L3", ctx))); in applyPatterns()
97 LinalgMarker(Identifier::get("L3", ctx), Identifier::get("L2", ctx))); in applyPatterns()
100 LinalgMarker(Identifier::get("L2", ctx), Identifier::get("L1", ctx))); in applyPatterns()
103 LinalgMarker(Identifier::get("L1", ctx), Identifier::get("REG", ctx))); in applyPatterns()
109 LinalgMarker({}, Identifier::get("L1", ctx))); in applyPatterns()
113 LinalgMarker(ArrayRef<Identifier>{Identifier::get("MEM", ctx), in applyPatterns()
126 LinalgMarker(Identifier::get("__with_perm__", ctx), in applyPatterns()
133 LinalgMarker(Identifier::get("L2__with_perm__", ctx), in applyPatterns()
137 LinalgMarker(Identifier::get("L1__with_perm__", ctx), in applyPatterns()
142 LinalgMarker(Identifier::get("__with_perm__", ctx), in applyPatterns()
[all …]
|
| D | TestLinalgFusionTransforms.cpp | 48 LinalgMarker(Identifier::get("basic_fusion", context), in fillFusionPatterns()
50 LinalgMarker(ArrayRef<Identifier>(), in fillFusionPatterns()
52 LinalgMarker(ArrayRef<Identifier>(), in fillFusionPatterns()
61 LinalgMarker(Identifier::get("lhs_fusion", context), in fillFusionPatterns()
63 LinalgMarker(ArrayRef<Identifier>(), in fillFusionPatterns()
65 LinalgMarker(ArrayRef<Identifier>(), in fillFusionPatterns()
74 LinalgMarker(Identifier::get("rhs_fusion", context), in fillFusionPatterns()
76 LinalgMarker(ArrayRef<Identifier>(), in fillFusionPatterns()
78 LinalgMarker(ArrayRef<Identifier>(), in fillFusionPatterns()
87 LinalgMarker(Identifier::get("two_operand_fusion", context), in fillFusionPatterns()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Generalization.cpp | 66 LinalgGeneralizationPattern(MLIRContext *context, linalg::LinalgMarker marker, in LinalgGeneralizationPattern()
89 linalg::LinalgMarker marker;
103 linalg::LinalgMarker marker, in LinalgNamedOpGeneralizationPattern()
131 linalg::LinalgMarker marker;
168 linalg::LinalgMarker marker) { in populateLinalgConvGeneralizationPatterns()
174 linalg::LinalgMarker marker) { in populateLinalgNamedOpsGeneralizationPatterns()
|
| D | Transforms.cpp | 47 mlir::linalg::LinalgMarker::LinalgMarker(ArrayRef<Identifier> matchDisjunction, in LinalgMarker() function in mlir::linalg::LinalgMarker
53 mlir::linalg::LinalgMarker::checkAndNotify(PatternRewriter &rewriter, in checkAndNotify()
82 void mlir::linalg::LinalgMarker::replaceLinalgMarker(PatternRewriter &rewriter, in replaceLinalgMarker()
110 LinalgMarker marker, PatternBenefit benefit) in LinalgBaseTilingPattern()
115 LinalgTilingOptions options, LinalgMarker marker, PatternBenefit benefit) in LinalgBaseTilingPattern()
156 LinalgMarker marker, LinalgMarker fusedOpMarker, in LinalgBaseTileAndFusePattern()
157 LinalgMarker originalOpMarker, PatternBenefit benefit) in LinalgBaseTileAndFusePattern()
244 ArrayRef<unsigned> interchangeVector, LinalgMarker marker, in LinalgBaseInterchangePattern()
271 LinalgMarker marker, PatternBenefit benefit) in LinalgBasePromotionPattern()
298 StringRef opName, MLIRContext *context, LinalgMarker marker, in LinalgBaseVectorizationPattern()
|
| D | CodegenStrategy.cpp | 40 ? linalg::LinalgMarker({}, nextState) in transform()
41 : linalg::LinalgMarker(currentState, nextState); in transform()
|
| D | Vectorization.cpp | 617 LinalgMarker({}, Identifier::get(kTiledMarker, context))); in populateVectorizationPatterns()
621 LinalgMarker(Identifier::get(kTiledMarker, context), in populateVectorizationPatterns()
|
| D | Tiling.cpp | 548 ctx, options, LinalgMarker({}, Identifier::get("tiled", ctx))); in insert()
|