Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestLoopFusion.cpp113 static bool testSliceComputation(AffineForOp forOpA, AffineForOp forOpB, in testSliceComputation() argument
118 FusionResult result = mlir::canFuseLoops(forOpA, forOpB, d, &sliceUnion); in testSliceComputation()
120 forOpB->emitRemark("slice (") in testSliceComputation()
124 forOpB->emitRemark("Incorrect slice (") in testSliceComputation()
135 static bool testLoopFusionTransformation(AffineForOp forOpA, AffineForOp forOpB, in testLoopFusionTransformation() argument
141 FusionResult result = mlir::canFuseLoops(forOpA, forOpB, d, &sliceUnion); in testLoopFusionTransformation()
143 mlir::fuseLoops(forOpA, forOpB, sliceUnion); in testLoopFusionTransformation()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopFusionUtils.cpp143 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint() local
146 getFirstDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
148 getLastDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
173 return forOpB.getOperation(); in getFusedLoopNestInsertionPoint()
288 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in canFuseLoops() local
299 if (!gatherLoadsAndStores(forOpB, opsB)) { in canFuseLoops()
H A DLoopUtils.cpp1394 void mlir::interchangeLoops(AffineForOp forOpA, AffineForOp forOpB) { in interchangeLoops() argument
1395 assert(&*forOpA.getBody()->begin() == forOpB.getOperation()); in interchangeLoops()
1397 auto &forOpBBody = forOpB.getBody()->getOperations(); in interchangeLoops()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DLoopUtils.h117 void interchangeLoops(AffineForOp forOpA, AffineForOp forOpB);