| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopFusionUtils.cpp | 387 if (AffineMap ubMap = srcSlice.ubs[i]) { in fuseLoops() local
389 canonicalizeMapAndOperands(&ubMap, &ubOperands); in fuseLoops()
390 forOp.setUpperBound(ubOperands, ubMap); in fuseLoops()
487 static Optional<uint64_t> getConstDifference(AffineMap lbMap, AffineMap ubMap) { in getConstDifference() argument
489 assert(ubMap.getNumResults() == 1 && "expected single result bound map"); in getConstDifference()
490 assert(lbMap.getNumDims() == ubMap.getNumDims()); in getConstDifference()
491 assert(lbMap.getNumSymbols() == ubMap.getNumSymbols()); in getConstDifference()
493 AffineExpr ubExpr(ubMap.getResult(0)); in getConstDifference()
526 AffineMap ubMap = slice.ubs[i]; in buildSliceTripCountMap() local
527 if (lbMap == AffineMap() || ubMap == AffineMap()) { in buildSliceTripCountMap()
[all …]
|
| D | LoopUtils.cpp | 251 AffineMap lbMap, AffineMap ubMap, in generateShiftedLoop() argument
258 assert(ubMap.getNumInputs() == ubOperands.size()); in generateShiftedLoop()
261 ubOperands, ubMap, srcForOp.getStep()); in generateShiftedLoop()
682 AffineMap ubMap = in setIntraTileBoundsParametric() local
685 newIntraTileLoop.setUpperBound(ubOperands, ubMap); in setIntraTileBoundsParametric()
779 AffineMap ubMap = in setInterTileBoundsParametric() local
782 newLoop.setUpperBound(ubOperands, ubMap); in setInterTileBoundsParametric()
852 AffineMap ubMap = in constructTiledIndexSetHyperRect() local
855 /*operands=*/newLoops[i].getInductionVar(), ubMap); in constructTiledIndexSetHyperRect()
885 AffineMap ubMap = in constructTiledIndexSetHyperRect() local
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 1212 ValueRange ubOperands, AffineMap ubMap, int64_t step, in build() argument
1217 assert(((!ubMap && ubOperands.empty()) || in build()
1218 ubOperands.size() == ubMap.getNumInputs()) && in build()
1234 result.addAttribute(getUpperBoundAttrName(), AffineMapAttr::get(ubMap)); in build()
1264 auto ubMap = AffineMap::getConstantMap(ub, builder.getContext()); in build() local
1265 return build(builder, result, {}, lbMap, {}, ubMap, step, iterArgs, in build()
1513 AffineMap ubMap = getUpperBoundMapAttr().getValue(); in getNumIterOperands() local
1515 return getNumOperands() - lbMap.getNumInputs() - ubMap.getNumInputs(); in getNumIterOperands()
1602 auto ubMap = forOp.getUpperBoundMap(); in canonicalizeLoopBounds() local
1604 auto prevUbMap = ubMap; in canonicalizeLoopBounds()
[all …]
|
| /external/llvm-project/mlir/lib/Analysis/ |
| D | AffineStructures.cpp | 1561 auto ubMap = AffineMap::get(dimCount, symCount, ubExprs, context); in getLowerAndUpperBound() local
1563 return {lbMap, ubMap}; in getLowerAndUpperBound()
1686 AffineMap &ubMap = (*ubMaps)[pos]; in getSliceBounds() local
1690 ubMap = AffineMap::get(numMapDims, numMapSymbols, expr + 1); in getSliceBounds()
1703 std::tie(lbMap, ubMap) = tmpClone->getLowerAndUpperBound( in getSliceBounds()
1722 if (!ubMap || ubMap.getNumResults() > 1) { in getSliceBounds()
1727 (ubMap) = AffineMap::get( in getSliceBounds()
1738 LLVM_DEBUG(ubMap.dump();); in getSliceBounds()
1847 AffineMap ubMap = ubMaps[i]; in addSliceBounds() local
1849 assert(!ubMap || ubMap.getNumInputs() == operands.size()); in addSliceBounds()
[all …]
|
| D | Utils.cpp | 195 AffineMap &ubMap) const { in getLowerAndUpperBound()
206 ubMap = boundPairs.second; in getLowerAndUpperBound()
208 assert(ubMap && "upper bound for a region must exist"); in getLowerAndUpperBound()
210 assert(ubMap.getNumInputs() == cst.getNumDimAndSymbolIds() - rank); in getLowerAndUpperBound()
903 if (AffineMap ubMap = sliceState->ubs[i]) in insertBackwardComputationSlice() local
904 forOp.setUpperBound(sliceState->ubOperands[i], ubMap); in insertBackwardComputationSlice()
|
| D | LoopAnalysis.cpp | 52 auto ubMap = forOp.getUpperBoundMap(); in buildTripCountMapAndOperands() local
61 AffineValueMap ubValueMap(ubMap, forOp.getUpperBoundOperands()); in buildTripCountMapAndOperands()
|
| /external/llvm-project/mlir/include/mlir/Analysis/ |
| D | Utils.h | 245 AffineMap &ubMap) const;
|
| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | LoopUtils.h | 295 AffineMap ubMap, int64_t step = 1);
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineOps.td | 221 "ValueRange":$ubOperands, "AffineMap":$ubMap, CArg<"int64_t", "1">:$step,
657 "ValueRange":$lbArgs, "AffineMap":$ubMap, "ValueRange":$ubArgs)>,
660 "ValueRange":$lbArgs, "AffineMap":$ubMap, "ValueRange":$ubArgs,
|