| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | LoopRangeFolding.cpp | 57 BlockAndValueMapping ubMap; in runOnOperation() local
58 ubMap.map(indVar, op.getUpperBound()); in runOnOperation()
64 Operation *ubFold = b.clone(*user, ubMap); in runOnOperation()
70 Operation *ubFold = b.clone(*user, ubMap); in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | Utils.cpp | 159 AffineMap ubMap = ubs[i]; in isSliceMaximalFastCheck() local
162 if (!lbMap || !ubMap || lbMap.getNumResults() != 1 || in isSliceMaximalFastCheck()
163 ubMap.getNumResults() != 1 || in isSliceMaximalFastCheck()
164 lbMap.getResult(0) + 1 != ubMap.getResult(0) || in isSliceMaximalFastCheck()
418 ubMap = boundPairs.second; in getLowerAndUpperBound()
420 assert(ubMap && "upper bound for a region must exist"); in getLowerAndUpperBound()
977 assert(lbMap.getNumDims() == ubMap.getNumDims()); in getConstDifference()
980 AffineExpr ubExpr(ubMap.getResult(0)); in getConstDifference()
1002 AffineMap ubMap = slice.ubs[i]; in buildSliceTripCountMap() local
1007 ubMap.getNumResults() == 0) { in buildSliceTripCountMap()
[all …]
|
| H A D | AffineStructures.cpp | 653 AffineMap ubMap = ubMaps[i]; in addDomainFromSliceMaps() local
655 assert(!ubMap || ubMap.getNumInputs() == operands.size()); in addDomainFromSliceMaps()
660 ubMap.getNumResults() == 1 && in addDomainFromSliceMaps()
688 if (ubMap && failed(addBound(BoundType::UB, i, ubMap, operands))) in addDomainFromSliceMaps()
967 return {lbMap, ubMap}; in getLowerAndUpperBound()
1128 if (!ubMap || ubMap.getNumResults() > 1) { in getSliceBounds()
1133 ubMap = AffineMap::get( in getSliceBounds()
1144 LLVM_DEBUG(ubMap.dump();); in getSliceBounds()
1298 AffineMap ubMap = ubMaps[i]; in addSliceBounds() local
1300 assert(!ubMap || ubMap.getNumInputs() == operands.size()); in addSliceBounds()
[all …]
|
| H A D | LoopAnalysis.cpp | 52 auto ubMap = forOp.getUpperBoundMap(); in getTripCountMapAndOperands() local
61 AffineValueMap ubValueMap(ubMap, forOp.getUpperBoundOperands()); in getTripCountMapAndOperands()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 182 AffineMap lbMap, AffineMap ubMap, in generateShiftedLoop() argument
189 assert(ubMap.getNumInputs() == ubOperands.size()); in generateShiftedLoop()
633 AffineMap ubMap = in setIntraTileBoundsParametric() local
636 newIntraTileLoop.setUpperBound(ubOperands, ubMap); in setIntraTileBoundsParametric()
730 AffineMap ubMap = in setInterTileBoundsParametric() local
733 newLoop.setUpperBound(ubOperands, ubMap); in setInterTileBoundsParametric()
807 AffineMap ubMap = b.getSingleDimShiftAffineMap( in constructTiledIndexSetHyperRect() local
841 AffineMap ubMap = in constructTiledIndexSetHyperRect() local
848 AffineMap ubMap = in constructTiledIndexSetHyperRect() local
1651 auto ubMap = forOp.getUpperBoundMap(); in stripmineSink() local
[all …]
|
| H A D | LoopFusionUtils.cpp | 453 if (AffineMap ubMap = srcSlice.ubs[i]) { in fuseLoops() local
455 canonicalizeMapAndOperands(&ubMap, &ubOperands); in fuseLoops()
456 forOp.setUpperBound(ubOperands, ubMap); in fuseLoops()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 1824 auto ubMap = forOp.getUpperBoundMap(); in canonicalizeLoopBounds() local
1826 auto prevUbMap = ubMap; in canonicalizeLoopBounds()
1834 ubMap = removeDuplicateExprs(ubMap); in canonicalizeLoopBounds()
1842 if (ubMap != prevUbMap) in canonicalizeLoopBounds()
1994 auto ubMap = getUpperBoundMap(); in getUpperBound() local
1996 lbMap.getNumInputs() + ubMap.getNumInputs(), ubMap); in getUpperBound()
2037 auto ubMap = getUpperBoundMap(); in setUpperBoundMap() local
2041 (void)ubMap; in setUpperBoundMap()
2086 auto ubMap = getUpperBoundMap(); in matchingBoundOperandList() local
2251 auto ubMap = loop.getUpperBoundMap(); in replaceForOpWithNewYields() local
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/ |
| H A D | Utils.h | 309 AffineMap &ubMap) const;
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/ |
| H A D | LoopUtils.h | 276 AffineMap ubMap, int64_t step = 1);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | SuperVectorize.cpp | 1038 AffineMap ubMap = vecForOp.getUpperBoundMap(); in createMask() local
1040 if (ubMap.getNumResults() == 1) in createMask()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineOps.td | 233 "ValueRange":$ubOperands, "AffineMap":$ubMap, CArg<"int64_t", "1">:$step,
|