| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopFusionUtils.cpp | 382 if (AffineMap lbMap = srcSlice.lbs[i]) { in fuseLoops() local
384 canonicalizeMapAndOperands(&lbMap, &lbOperands); in fuseLoops()
385 forOp.setLowerBound(lbOperands, lbMap); in fuseLoops()
487 static Optional<uint64_t> getConstDifference(AffineMap lbMap, AffineMap ubMap) { in getConstDifference() argument
488 assert(lbMap.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()
492 AffineExpr lbExpr(lbMap.getResult(0)); in getConstDifference()
494 auto loopSpanExpr = simplifyAffineExpr(ubExpr - lbExpr, lbMap.getNumDims(), in getConstDifference()
495 lbMap.getNumSymbols()); in getConstDifference()
[all …]
|
| D | LoopUtils.cpp | 61 auto lbMap = forOp.getLowerBoundMap(); in getCleanupLoopLowerBound() local
64 if (lbMap.getNumResults() != 1) { in getCleanupLoopLowerBound()
80 auto lb = b.create<AffineApplyOp>(forOp.getLoc(), lbMap, in getCleanupLoopLowerBound()
172 auto lbMap = forOp.getLowerBoundMap(); in promoteIfSingleIteration() local
174 if (lbMap == builder.getDimIdentityMap()) { in promoteIfSingleIteration()
179 builder.create<AffineApplyOp>(forOp.getLoc(), lbMap, lbOperands); in promoteIfSingleIteration()
251 AffineMap lbMap, AffineMap ubMap, in generateShiftedLoop() argument
257 assert(lbMap.getNumInputs() == lbOperands.size()); in generateShiftedLoop()
260 auto loopChunk = b.create<AffineForOp>(srcForOp.getLoc(), lbOperands, lbMap, in generateShiftedLoop()
677 AffineMap lbMap = in setIntraTileBoundsParametric() local
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| D | AffineLoopNormalize.cpp | 24 AffineMap lbMap = op.lowerBoundsMap(); in normalizeAffineParallel() local
28 llvm::all_of(llvm::zip(steps, lbMap.getResults()), [](auto tuple) { in normalizeAffineParallel()
54 AffineExpr lbExpr = lbMap.getResult(i); in normalizeAffineParallel()
55 unsigned nDims = lbMap.getNumDims(); in normalizeAffineParallel()
58 /*symbolCount=*/lbMap.getNumSymbols(), expr); in normalizeAffineParallel()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 1211 ValueRange lbOperands, AffineMap lbMap, in build() argument
1214 assert(((!lbMap && lbOperands.empty()) || in build()
1215 lbOperands.size() == lbMap.getNumInputs()) && in build()
1230 result.addAttribute(getLowerBoundAttrName(), AffineMapAttr::get(lbMap)); in build()
1263 auto lbMap = AffineMap::getConstantMap(lb, builder.getContext()); in build() local
1265 return build(builder, result, {}, lbMap, {}, ubMap, step, iterArgs, in build()
1512 AffineMap lbMap = getLowerBoundMapAttr().getValue(); in getNumIterOperands() local
1515 return getNumOperands() - lbMap.getNumInputs() - ubMap.getNumInputs(); in getNumIterOperands()
1601 auto lbMap = forOp.getLowerBoundMap(); in canonicalizeLoopBounds() local
1603 auto prevLbMap = lbMap; in canonicalizeLoopBounds()
[all …]
|
| /external/llvm-project/mlir/lib/Analysis/ |
| D | LoopAnalysis.cpp | 51 auto lbMap = forOp.getLowerBoundMap(); in buildTripCountMapAndOperands() local
53 if (lbMap.getNumResults() != 1) { in buildTripCountMapAndOperands()
64 lbMap.getResult(0)); in buildTripCountMapAndOperands()
65 auto lbMapSplat = AffineMap::get(lbMap.getNumDims(), lbMap.getNumSymbols(), in buildTripCountMapAndOperands()
|
| D | AffineStructures.cpp | 1560 auto lbMap = AffineMap::get(dimCount, symCount, lbExprs, context); in getLowerAndUpperBound() local
1563 return {lbMap, ubMap}; in getLowerAndUpperBound()
1685 AffineMap &lbMap = (*lbMaps)[pos]; in getSliceBounds() local
1689 lbMap = AffineMap::get(numMapDims, numMapSymbols, expr); in getSliceBounds()
1703 std::tie(lbMap, ubMap) = tmpClone->getLowerAndUpperBound( in getSliceBounds()
1712 if (!lbMap || lbMap.getNumResults() > 1) { in getSliceBounds()
1717 lbMap = AffineMap::get( in getSliceBounds()
1735 LLVM_DEBUG(lbMap.dump();); in getSliceBounds()
1846 AffineMap lbMap = lbMaps[i]; in addSliceBounds() local
1848 assert(!lbMap || lbMap.getNumInputs() == operands.size()); in addSliceBounds()
[all …]
|
| D | Utils.cpp | 194 void MemRefRegion::getLowerAndUpperBound(unsigned pos, AffineMap &lbMap, in getLowerAndUpperBound() argument
205 lbMap = boundPairs.first; in getLowerAndUpperBound()
207 assert(lbMap && "lower bound for a region must exist"); in getLowerAndUpperBound()
209 assert(lbMap.getNumInputs() == cst.getNumDimAndSymbolIds() - rank); in getLowerAndUpperBound()
901 if (AffineMap lbMap = sliceState->lbs[i]) in insertBackwardComputationSlice() local
902 forOp.setLowerBound(sliceState->lbOperands[i], lbMap); in insertBackwardComputationSlice()
|
| /external/llvm-project/mlir/include/mlir/Analysis/ |
| D | Utils.h | 244 void getLowerAndUpperBound(unsigned pos, AffineMap &lbMap,
|
| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | LoopUtils.h | 293 AffineMap lbMap,
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineOps.td | 220 OpBuilderDAG<(ins "ValueRange":$lbOperands, "AffineMap":$lbMap,
656 "ArrayRef<AtomicRMWKind>":$reductions, "AffineMap":$lbMap,
659 "ArrayRef<AtomicRMWKind>":$reductions, "AffineMap":$lbMap,
|