Searched refs:combinerOps (Results 1 – 8 of 8) sorted by relevance
| /llvm-project-15.0.7/mlir/test/lib/Analysis/ |
| H A D | TestMatchReduction.cpp | 23 ArrayRef<Operation *> combinerOps) { in printReductionResult() argument
27 for (Operation *combOp : combinerOps) in printReductionResult()
74 SmallVector<Operation *, 4> combinerOps; in runOnOperation() local
75 Value reducedValue = matchReduction(outputs, i, combinerOps); in runOnOperation()
76 printReductionResult(op, i, reducedValue, combinerOps); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 105 SmallVector<Operation *, 4> combinerOps; in splitReduction() local
106 if (!matchReduction(op.getRegionOutputArgs(), 0, combinerOps) || in splitReduction()
107 combinerOps.size() != 1) in splitReduction()
110 Operation *reductionOp = combinerOps[0]; in splitReduction()
298 SmallVector<Operation *> combinerOps; in splitReductionByScaling() local
299 if (!matchReduction(op.getRegionOutputArgs(), 0, combinerOps)) in splitReductionByScaling()
303 llvm::map_range(combinerOps, [&](Operation *reductionOp) { in splitReductionByScaling()
408 llvm::zip(genericOp->getResults(), op.outputs(), combinerOps)) { in splitReductionByScaling()
|
| H A D | Vectorization.cpp | 149 SmallVector<Operation *, 4> combinerOps; in matchLinalgReduction() local
150 if (!matchReduction(linalgOp.getRegionOutputArgs(), outputPos, combinerOps) || in matchLinalgReduction()
151 combinerOps.size() != 1) in matchLinalgReduction()
155 return combinerOps[0]; in matchLinalgReduction()
|
| /llvm-project-15.0.7/mlir/lib/Analysis/ |
| H A D | SliceAnalysis.cpp | 272 SmallVectorImpl<Operation *> &combinerOps) { in matchReduction() argument
301 combinerOps.push_back(combinerOp); in matchReduction()
307 if (combinerOps.size() != 1) in matchReduction()
313 if (terminatorOp->getOperand(redPos) != combinerOps.back()->getResults()[0]) in matchReduction()
|
| /llvm-project-15.0.7/mlir/include/mlir/Analysis/ |
| H A D | SliceAnalysis.h | 237 SmallVectorImpl<Operation *> &combinerOps);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineAnalysis.cpp | 42 SmallVector<Operation *> combinerOps; in getSupportedReduction() local
44 matchReduction(forOp.getRegionIterArgs(), pos, combinerOps); in getSupportedReduction()
49 if (combinerOps.size() > 1) in getSupportedReduction()
52 Operation *combinerOp = combinerOps.back(); in getSupportedReduction()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToOpenMP/ |
| H A D | SCFToOpenMP.cpp | 44 SmallVector<Operation *, 4> combinerOps; in matchSimpleReduction() local
46 /*redPos=*/0, combinerOps); in matchSimpleReduction()
49 combinerOps.size() != 1) in matchSimpleReduction()
52 return isa<OpTy...>(combinerOps[0]) && in matchSimpleReduction()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | SuperVectorize.cpp | 1440 SmallVector<Operation *> combinerOps; in vectorizeAffineYieldOp() local
1442 cast<AffineForOp>(newParentOp).getRegionIterArgs(), i, combinerOps); in vectorizeAffineYieldOp()
1444 assert(combinerOps.size() == 1 && "expect only one combiner op"); in vectorizeAffineYieldOp()
1447 state.builder.setInsertionPoint(combinerOps.back()); in vectorizeAffineYieldOp()
1454 combinerOps.back()->replaceUsesOfWith(reducedVal, maskedReducedVal); in vectorizeAffineYieldOp()
|