| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | AffineMap.cpp | 110 return getNumDims() >= getNumResults() && in isMinorIdentity()
121 if (getNumDims() < getNumResults()) in isMinorIdentityWithBroadcasting()
256 if (getNumDims() != getNumResults()) in isIdentity()
294 unsigned AffineMap::getNumDims() const { in getNumDims() function in AffineMap
442 unsigned numDims = map.getNumDims(); in compose()
509 if (getNumDims() != getNumResults()) in isPermutation()
547 dimReplacements.reserve(map.getNumDims()); in compressDims()
573 unsigned numDims = maps.front().getNumDims(), in compressUnusedImpl()
668 seenExprs.reserve(map.getNumDims()); in inversePermutation()
721 unsigned numDims = maps[0].getNumDims(); in getUnusedDimsBitVector()
[all …]
|
| H A D | BuiltinAttributeInterfaces.cpp | 84 if (m.getNumDims() != shape.size()) in verifyAffineMapAsLayout()
86 << shape.size() << " != " << m.getNumDims(); in verifyAffineMapAsLayout()
|
| H A D | BuiltinTypes.cpp | 806 simplifyAffineExpr(m.getResult(0), m.getNumDims(), m.getNumSymbols()); in getStridesAndOffset()
814 unsigned numDims = m.getNumDims(); in getStridesAndOffset()
961 if (m.getNumDims() == 0 && m.getNumSymbols() == 0) { in canonicalizeStridedLayout()
980 simplifyAffineExpr(m.getResult(0), m.getNumDims(), m.getNumSymbols()); in canonicalizeStridedLayout()
983 m.getNumDims(), m.getNumSymbols(), simplifiedLayoutExpr))); in canonicalizeStridedLayout()
997 unsigned numDims = maps[0].getNumDims(), nSymbols = maps[0].getNumSymbols(); in makeCanonicalStridedLayoutExpr()
|
| H A D | IntegerSet.cpp | 15 unsigned IntegerSet::getNumDims() const { return set->dimCount; } in getNumDims() function in IntegerSet
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | AffineMap.h | 157 unsigned getNumDims() const;
221 assert(offset <= getNumDims());
222 return AffineMap::get(getNumDims() + shift, getNumSymbols(),
226 return e.shiftDims(getNumDims(), shift, offset);
234 return AffineMap::get(getNumDims(), getNumSymbols() + shift,
249 return AffineMap::get(getNumDims(), getNumSymbols(), exprs, getContext()); in dropResult()
257 return AffineMap::get(getNumDims(), getNumSymbols(), exprs, getContext()); in insertResult()
353 unsigned getNumDims() const { return numDims; } in getNumDims() function
|
| H A D | IntegerSet.h | 80 unsigned getNumDims() const;
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineValueMap.cpp | 32 auto aDims = a.getOperands().take_front(a.getNumDims()); in difference()
33 auto bDims = b.getOperands().take_front(b.getNumDims()); in difference()
43 .shiftDims(a.getNumDims()) in difference()
53 auto diffMap = AffineMap::get(bMap.getNumDims(), bMap.getNumSymbols(), in difference()
|
| H A D | AffineOps.cpp | 505 if (map.getNumDims() != numDims || in parse()
1580 if (map.getNumDims() != numDims) in parseBound()
2029 assert(lbMap.getNumDims() == map.getNumDims() && in setLowerBoundMap()
2038 assert(ubMap.getNumDims() == map.getNumDims() && in setUpperBoundMap()
2087 if (lbMap.getNumDims() != ubMap.getNumDims() || in matchingBoundOperandList()
2389 if (set.getNumDims() != numDims) in parse()
3252 return m.getNumDims() == lbMaps[0].getNumDims() && in build()
3259 return m.getNumDims() == ubMaps[0].getNumDims() && in build()
3413 assert(lbMap.getNumDims() == map.getNumDims() && in setLowerBoundsMap()
3421 assert(ubMap.getNumDims() == map.getNumDims() && in setUpperBoundsMap()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineValueMap.h | 69 inline unsigned getNumDims() const { return map.getNumDims(); } in getNumDims() function
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 227 auto numDims = affineMap.getNumDims(); in expandAffineMap()
523 unsigned nDims = lbMap.getNumDims(); in normalizeAffineParallel()
540 SmallVector<int64_t, 8> newSteps(op.getNumDims(), 1); in normalizeAffineParallel()
585 for (unsigned j = 0, e = origUbMap.getNumDims(); j < e; ++j) in normalizeAffineFor()
587 for (unsigned j = 0, e = origLbMap.getNumDims(); j < e; ++j) in normalizeAffineFor()
592 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in normalizeAffineFor()
620 AffineMap::get(origLbMap.getNumDims() + origUbMap.getNumDims(), in normalizeAffineFor()
627 lb.getMap().getNumDims()); in normalizeAffineFor()
645 AffineMap ivMap = AffineMap::get(origLbMap.getNumDims() + 1, in normalizeAffineFor()
1148 auto singleResMap = AffineMap::get(oldMap.getNumDims(), in replaceAllMemRefUsesWith()
[all …]
|
| H A D | LoopUtils.cpp | 84 auto bumpMap = AffineMap::get(tripCountMap.getNumDims(), in getCleanupLoopLowerBound()
577 for (unsigned j = 0, e = origLbMap.getNumDims(); j < e; ++j) in setIntraTileBoundsParametric()
579 for (unsigned j = 0, e = origUbMap.getNumDims(); j < e; ++j) in setIntraTileBoundsParametric()
594 lbOperands.push_back(lb.getOperand(origLbMap.getNumDims() + j)); in setIntraTileBoundsParametric()
596 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in setIntraTileBoundsParametric()
674 for (unsigned j = 0, e = origUbMap.getNumDims(); j < e; ++j) in setInterTileBoundsParametric()
679 ubOperands.push_back(ub.getOperand(origUbMap.getNumDims() + j)); in setInterTileBoundsParametric()
823 for (unsigned j = 0, e = origUbMap.getNumDims(); j < e; ++j) in constructTiledIndexSetHyperRect()
835 AffineExpr dim = b.getAffineDimExpr(origUbMap.getNumDims()); in constructTiledIndexSetHyperRect()
1621 bounds.push_back(b.getAffineDimExpr(map->getNumDims()) + offset); in augmentMapAndBounds()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | IntegerSet.cpp | 61 (void)unwrapList(unwrap(set).getNumDims(), dimReplacements, mlirDims); in mlirIntegerSetReplaceGet()
74 return static_cast<intptr_t>(unwrap(set).getNumDims()); in mlirIntegerSetGetNumDims()
|
| H A D | AffineMap.cpp | 95 return unwrap(affineMap).getNumDims(); in mlirAffineMapGetNumDims()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 142 AffineMap::get(map.getNumDims() + 1, 0, exprs, op.getContext())); in splitReduction()
191 newMaps.push_back(AffineMap::get(oldOutputMap.getNumDims() + 1, 0, outputExpr, in splitReduction()
252 AffineMap::getMultiDimIdentityMap(map.getNumDims(), op.getContext()); in scaleReductionDim()
256 shiftedIdMap.getNumDims(), /*numSymbols=*/0); in scaleReductionDim()
265 AffineMap::getMultiDimIdentityMap(map.getNumDims(), op.getContext()); in insertParallelDim()
|
| H A D | Split.cpp | 40 if (!type || dimension >= indexing.getNumDims() || in getOperandSlices()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorDropLeadUnitDim.cpp | 214 AffineMap::get(oldMap.getNumDims(), oldMap.getNumSymbols(), newResults, in matchAndRewrite()
262 AffineMap::get(oldMap.getNumDims(), oldMap.getNumSymbols(), newResults, in matchAndRewrite()
360 map = AffineMap::get(map.getNumDims(), 0, transposeResults, in matchAndRewrite()
382 newIndexingMaps.push_back(AffineMap::get(map.getNumDims() - 1, 0, results, in matchAndRewrite()
|
| H A D | VectorTransferPermutationMapRewritePatterns.cpp | 186 map.getNumDims(), map.getNumResults(), rewriter.getContext()); in matchAndRewrite()
227 map.getNumDims(), 0, map.getResults().take_back(reducedShapeRank), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineStructures.cpp | 122 localVarCst->reset(map.getNumDims(), map.getNumSymbols()); in getFlattenedAffineExprs()
125 return ::getFlattenedAffineExprs(map.getResults(), map.getNumDims(), in getFlattenedAffineExprs()
134 localVarCst->reset(set.getNumDims(), set.getNumSymbols()); in getFlattenedAffineExprs()
137 return ::getFlattenedAffineExprs(set.getConstraints(), set.getNumDims(), in getFlattenedAffineExprs()
154 set.getNumDims() + set.getNumSymbols() + 1, in FlatAffineValueConstraints()
155 PresburgerSpace::getSetSpace(set.getNumDims(), in FlatAffineValueConstraints()
460 assert(other.getNumDims() == getNumDimVars() && "dim mismatch"); in composeMatchingMap()
1175 assert(boundMap.getNumDims() == getNumDimVars() && "dim mismatch"); in addBound()
1580 SmallVector<AffineExpr> dimReplacements(map.getNumDims(), {}); in alignAffineMapWithValues()
1607 if (operand.index() < map.getNumDims()) { in alignAffineMapWithValues()
[all …]
|
| H A D | LoopAnalysis.cpp | 65 auto lbMapSplat = AffineMap::get(lbMap.getNumDims(), lbMap.getNumSymbols(), in getTripCountMapAndOperands()
227 unsigned numDims = accessMap.getNumDims(); in isContiguousAccess()
|
| H A D | Utils.cpp | 481 unsigned numDims = accessMap.getNumDims(); in compute()
977 assert(lbMap.getNumDims() == ubMap.getNumDims()); in getConstDifference()
981 auto loopSpanExpr = simplifyAffineExpr(ubExpr - lbExpr, lbMap.getNumDims(), in getConstDifference()
1360 return IntegerSet::getEmptySet(set.getNumDims(), set.getNumSymbols(), in simplifyIntegerSet()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/AffineToStandard/ |
| H A D | AffineToStandard.cpp | 181 lowerBoundTuple.reserve(op.getNumDims()); in matchAndRewrite()
182 upperBoundTuple.reserve(op.getNumDims()); in matchAndRewrite()
183 for (unsigned i = 0, e = op.getNumDims(); i < e; ++i) { in matchAndRewrite()
284 auto numDims = integerSet.getNumDims(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.cpp | 200 unsigned nDims = reassociation[0].getNumDims(); in isReassociationValid()
204 if (m.getNumDims() != nDims || m.getNumSymbols() != 0) { in isReassociationValid()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUDialect.h | 148 unsigned getNumDims() const;
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 510 unsigned numDims = map.getNumDims(), numRes = map.getNumResults(); in createLoopRanges()
528 unsigned numDims = map.getNumDims(), numRes = map.getNumResults(); in computeStaticLoopSizes()
590 llvm::SmallBitVector outputDims(resultShapesFromInputShapesMap.getNumDims()); in reifyResultShapes()
665 if (indexingMap.getNumDims() != numLoops) in verifyStructuredOpInterface()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | ReshapeOpsUtils.h | 131 if (it.value().getNumDims() != expandedRank) in verifyReshapeLikeTypes()
134 << expandedRank << "), but got " << it.value().getNumDims(); in verifyReshapeLikeTypes()
|