Searched refs:forOpB (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLoopFusion.cpp | 106 static bool testSliceComputation(AffineForOp forOpA, AffineForOp forOpB, in testSliceComputation() argument
111 FusionResult result = mlir::canFuseLoops(forOpA, forOpB, d, &sliceUnion); in testSliceComputation()
113 forOpB->emitRemark("slice (") in testSliceComputation()
124 static bool testLoopFusionTransformation(AffineForOp forOpA, AffineForOp forOpB, in testLoopFusionTransformation() argument
130 FusionResult result = mlir::canFuseLoops(forOpA, forOpB, d, &sliceUnion); in testLoopFusionTransformation()
132 mlir::fuseLoops(forOpA, forOpB, sliceUnion); in testLoopFusionTransformation()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopFusionUtils.cpp | 142 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in getFusedLoopNestInsertionPoint() local
145 getFirstDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
147 getLastDependentOpInRange(forOpA.getOperation(), forOpB.getOperation()); in getFusedLoopNestInsertionPoint()
172 return forOpB.getOperation(); in getFusedLoopNestInsertionPoint()
291 auto forOpB = isSrcForOpBeforeDstForOp ? dstForOp : srcForOp; in canFuseLoops() local
302 if (!gatherLoadsAndStores(forOpB, opsB)) { in canFuseLoops()
|
| D | LoopUtils.cpp | 1404 void mlir::interchangeLoops(AffineForOp forOpA, AffineForOp forOpB) { in interchangeLoops() argument
1405 assert(&*forOpA.getBody()->begin() == forOpB.getOperation()); in interchangeLoops()
1407 auto &forOpBBody = forOpB.getBody()->getOperations(); in interchangeLoops()
|
| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | LoopUtils.h | 116 void interchangeLoops(AffineForOp forOpA, AffineForOp forOpB);
|