Home
last modified time | relevance | path

Searched refs:srcForOp (Results 1 – 4 of 4) sorted by relevance

/external/llvm-project/mlir/lib/Transforms/Utils/
DLoopFusionUtils.cpp137 static Operation *getFusedLoopNestInsertionPoint(AffineForOp srcForOp, in getFusedLoopNestInsertionPoint() argument
140 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in getFusedLoopNestInsertionPoint()
141 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in getFusedLoopNestInsertionPoint()
142 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint()
263 FusionResult mlir::canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp, in canFuseLoops() argument
273 auto *block = srcForOp->getBlock(); in canFuseLoops()
281 if (!getFusedLoopNestInsertionPoint(srcForOp, dstForOp)) { in canFuseLoops()
288 srcForOp->isBeforeInBlock(dstForOp.getOperation()); in canFuseLoops()
290 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in canFuseLoops()
291 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in canFuseLoops()
[all …]
DLoopUtils.cpp253 unsigned offset, AffineForOp srcForOp, OpBuilder b) { in generateShiftedLoop() argument
254 auto lbOperands = srcForOp.getLowerBoundOperands(); in generateShiftedLoop()
255 auto ubOperands = srcForOp.getUpperBoundOperands(); in generateShiftedLoop()
260 auto loopChunk = b.create<AffineForOp>(srcForOp.getLoc(), lbOperands, lbMap, in generateShiftedLoop()
261 ubOperands, ubMap, srcForOp.getStep()); in generateShiftedLoop()
263 auto srcIV = srcForOp.getInductionVar(); in generateShiftedLoop()
278 srcForOp.getLoc(), in generateShiftedLoop()
280 -static_cast<int64_t>(srcForOp.getStep() * shift)), in generateShiftedLoop()
/external/llvm-project/mlir/include/mlir/Transforms/
DLoopFusionUtils.h89 FusionResult canFuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,
97 void fuseLoops(AffineForOp srcForOp, AffineForOp dstForOp,
132 bool getFusionComputeCost(AffineForOp srcForOp, LoopNestStats &srcStats,
/external/llvm-project/mlir/test/lib/Transforms/
DTestLoopFusion.cpp54 static bool testDependenceCheck(AffineForOp srcForOp, AffineForOp dstForOp, in testDependenceCheck() argument
60 mlir::canFuseLoops(srcForOp, dstForOp, d, &sliceUnion); in testDependenceCheck()
62 srcForOp->emitRemark("block-level dependence preventing" in testDependenceCheck()