Home
last modified time | relevance | path

Searched refs:tripCountMap (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DLoopAnalysis.cpp36 AffineForOp forOp, AffineMap *tripCountMap, in getTripCountMapAndOperands() argument
47 *tripCountMap = AffineMap::getConstantMap(ceilDiv(loopSpan, step), context); in getTripCountMapAndOperands()
54 *tripCountMap = AffineMap(); in getTripCountMapAndOperands()
75 *tripCountMap = tripCountValueMap.getAffineMap(); in getTripCountMapAndOperands()
H A DUtils.cpp995 llvm::SmallDenseMap<Operation *, uint64_t, 8> *tripCountMap) { in buildSliceTripCountMap() argument
1010 (*tripCountMap)[op] = in buildSliceTripCountMap()
1016 (*tripCountMap)[op] = maybeConstTripCount.value(); in buildSliceTripCountMap()
1025 (*tripCountMap)[op] = tripCount.value(); in buildSliceTripCountMap()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DLoopFusionUtils.h134 DenseMap<Operation *, uint64_t> tripCountMap; member
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/
H A DUtils.h199 llvm::SmallDenseMap<Operation *, uint64_t, 8> *tripCountMap);
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp59 AffineMap tripCountMap; in getCleanupLoopLowerBound() local
61 getTripCountMapAndOperands(forOp, &tripCountMap, &tripCountOperands); in getCleanupLoopLowerBound()
63 if (!tripCountMap) { in getCleanupLoopLowerBound()
78 SmallVector<AffineExpr, 4> bumpExprs(tripCountMap.getNumResults()); in getCleanupLoopLowerBound()
79 SmallVector<Value, 4> bumpValues(tripCountMap.getNumResults()); in getCleanupLoopLowerBound()
81 for (unsigned i = 0, e = tripCountMap.getNumResults(); i < e; i++) { in getCleanupLoopLowerBound()
82 auto tripCountExpr = tripCountMap.getResult(i); in getCleanupLoopLowerBound()
84 auto bumpMap = AffineMap::get(tripCountMap.getNumDims(), in getCleanupLoopLowerBound()
85 tripCountMap.getNumSymbols(), bumpExprs[i]); in getCleanupLoopLowerBound()
90 SmallVector<AffineExpr, 4> newUbExprs(tripCountMap.getNumResults()); in getCleanupLoopLowerBound()
[all …]
H A DLoopFusionUtils.cpp512 stats->tripCountMap[childForOp] = *maybeConstTripCount; in getLoopNestStats()
554 int64_t tripCount = stats.tripCountMap[forOp]; in getComputeCostHelper()