| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 201 LogicalResult visitDimExpr(AffineDimExpr dimExpr) { in visitDimExpr()
202 unsigned position = dimExpr.getPosition(); in visitDimExpr()
223 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in isDimExprOrMulExpr() local
224 unsigned dim = dimExpr.getPosition(); in isDimExprOrMulExpr()
242 auto dimExpr = getAffineExprOfType<AffineDimExpr>(lhsExpr, rhsExpr); in isDimExprOrMulExpr() local
243 if (!mulExpr || !dimExpr) in isDimExprOrMulExpr()
245 unsigned dim = dimExpr.getPosition(); in isDimExprOrMulExpr()
550 bool visitDimExpr(AffineDimExpr dimExpr) { in visitDimExpr()
551 return positions.test(dimExpr.getPosition()); in visitDimExpr()
|
| H A D | LinalgOps.cpp | 1827 AffineExpr dimExpr = sourceMap.getResult(i); in createNewOperandWithStaticSizes() local
1828 if (affineExprToSize.find(dimExpr) == affineExprToSize.end() || in createNewOperandWithStaticSizes()
1836 newShape.push_back(affineExprToSize[dimExpr]); in createNewOperandWithStaticSizes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorTransferPermutationMapRewritePatterns.cpp | 215 auto dimExpr = expr.dyn_cast<AffineConstantExpr>(); in matchAndRewrite() local
216 if (!dimExpr || dimExpr.getValue() != 0) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | AffineMap.cpp | 133 } else if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in isMinorIdentityWithBroadcasting() local
135 if (dimExpr.getPosition() != suffixStart + resIdx) in isMinorIdentityWithBroadcasting()
180 } else if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in isPermutationOfMinorIdentityWithBroadcasting() local
181 if (dimExpr.getPosition() < projectionStart) in isPermutationOfMinorIdentityWithBroadcasting()
184 dimExpr.getPosition() - projectionStart + leadingBroadcast; in isPermutationOfMinorIdentityWithBroadcasting()
|
| H A D | BuiltinTypes.cpp | 1007 AffineExpr dimExpr = std::get<0>(en); in makeCanonicalStridedLayoutExpr() local
1011 expr = expr ? expr + dimExpr * stride : dimExpr * stride; in makeCanonicalStridedLayoutExpr()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | LoopAnalysis.cpp | 233 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) in isContiguousAccess() local
234 exprOperands.push_back(mapOperands[dimExpr.getPosition()]); in isContiguousAccess()
|
| H A D | AffineStructures.cpp | 824 auto dimExpr = dividendExpr.dyn_cast<AffineDimExpr>(); in detectAsMod() local
825 if (!dimExpr) in detectAsMod()
831 dimExpr.getPosition()); in detectAsMod()
835 memo[pos] = dimExpr; in detectAsMod()
837 memo[pos] = dimExpr % divisor; in detectAsMod()
841 memo[quotientPosition] = dimExpr.floorDiv(divisor) * quotientSign; in detectAsMod()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | AffineMap.h | 554 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in applyPermutationMap() local
555 result.push_back(source[dimExpr.getPosition()]); in applyPermutationMap()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Fusion.cpp | 91 auto dimExpr = en.value().dyn_cast<AffineDimExpr>(); in getShapeDefiningLoopRange() local
92 if (!dimExpr) in getShapeDefiningLoopRange()
|
| H A D | DropUnitDims.cpp | 190 if (AffineDimExpr dimExpr = expr.value().dyn_cast<AffineDimExpr>()) in matchAndRewrite() local
191 if (dims[dimExpr.getPosition()] == 1) in matchAndRewrite()
|
| H A D | ElementwiseOpFusion.cpp | 110 if (auto dimExpr = result.dyn_cast<AffineDimExpr>()) in areElementwiseOpsFusable() local
111 coveredDims[dimExpr.getPosition()] = true; in areElementwiseOpsFusable()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | LoopFusion.cpp | 968 auto dimExpr = b.getAffineDimExpr(outerIVs.size() + i); in createPrivateMemRef() local
971 simplifyAffineExpr(dimExpr - offsets[i], outerIVs.size() + rank, 0); in createPrivateMemRef()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 996 if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) in canonicalizeMapOrSetAndOperands() local
997 usedDims[dimExpr.getPosition()] = true; in canonicalizeMapOrSetAndOperands()
2924 } else if (auto dimExpr = expr.dyn_cast<AffineDimExpr>()) { in matchAndRewrite() local
2925 Value dimValue = dimOperands[dimExpr.getPosition()]; in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 2298 auto dimExpr = b.getAffineDimExpr(regionSymbols.size() + i); in generateCopy() local
2299 remapExprs.push_back(dimExpr - fastBufOffsets[i]); in generateCopy()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 3128 auto dimExpr = permutationMap.getResult(i).dyn_cast<AffineDimExpr>(); in foldTransferInBoundsAttribute() local
3129 assert(dimExpr && "Broadcast dims must be in-bounds"); in foldTransferInBoundsAttribute()
3131 isInBounds(op, /*resultIdx=*/i, /*indicesIdx=*/dimExpr.getPosition()); in foldTransferInBoundsAttribute()
|