Home
last modified time | relevance | path

Searched refs:forOpA (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()
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()
147 forOpA.erase(); in testLoopFusionTransformation()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopFusionUtils.cpp142 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in getFusedLoopNestInsertionPoint() local
146 getFirstDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
148 getLastDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
287 auto forOpA = isSrcForOpBeforeDstForOp ? srcForOp : dstForOp; in canFuseLoops() local
292 if (!gatherLoadsAndStores(forOpA, opsA)) { 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()
1396 auto &forOpABody = forOpA.getBody()->getOperations(); in interchangeLoops()
1402 forOpA->getBlock()->getOperations().splice(Block::iterator(forOpA), in interchangeLoops()
1410 forOpBBody.splice(forOpBBody.begin(), forOpA->getBlock()->getOperations(), in interchangeLoops()
1411 Block::iterator(forOpA)); in interchangeLoops()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DLoopUtils.h117 void interchangeLoops(AffineForOp forOpA, AffineForOp forOpB);