Home
last modified time | relevance | path

Searched refs:getAffineSymbolExpr (Results 1 – 21 of 21) sorted by relevance

/llvm-project-15.0.7/flang/lib/Optimizer/Transforms/
H A DAffinePromotion.cpp228 return {mlir::getAffineSymbolExpr(symCount++, value.getContext())}; in toAffineExpr()
317 lowerBound = mlir::getAffineSymbolExpr(i * 3, context), in createArrayIndexAffineMap()
319 mlir::getAffineSymbolExpr(i * 3 + 1, context), in createArrayIndexAffineMap()
320 stride = mlir::getAffineSymbolExpr(i * 3 + 2, context), in createArrayIndexAffineMap()
496 mlir::getAffineSymbolExpr(0, op.getContext())), in positiveConstantStep()
499 1 + mlir::getAffineSymbolExpr(0, op.getContext())), in positiveConstantStep()
506 auto lowerBound = mlir::getAffineSymbolExpr(0, op.getContext()); in genericBounds()
507 auto upperBound = mlir::getAffineSymbolExpr(1, op.getContext()); in genericBounds()
508 auto step = mlir::getAffineSymbolExpr(2, op.getContext()); in genericBounds()
517 mlir::getAffineSymbolExpr(1, op.getContext())); in genericBounds()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp52 AffineExpr currExpr = builder.getAffineSymbolExpr(dim - startPos); in getCollapsedOutputDimFromInputShape()
106 0, 1, builder.getAffineSymbolExpr(0).floorDiv(linearizedStaticDim)), in getExpandedOutputDimFromInputShape()
186 expr = expr + b.getAffineSymbolExpr(numSymbols++); in reifyResultShapes()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp450 rewriter.getAffineDimExpr(0) * rewriter.getAffineSymbolExpr(0) + in processParallelLoop()
451 rewriter.getAffineSymbolExpr(1)); in processParallelLoop()
494 ((rewriter.getAffineDimExpr(0) - rewriter.getAffineSymbolExpr(0)) in processParallelLoop()
495 .ceilDiv(rewriter.getAffineSymbolExpr(1)))); in processParallelLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DFoldSubViewOps.cpp76 expr = expr * rewriter.getAffineSymbolExpr(numSymbols++); in resolveSourceIndices()
84 expr = expr + rewriter.getAffineSymbolExpr(numSymbols++); in resolveSourceIndices()
H A DComposeSubView.cpp108 expr + rewriter.getAffineSymbolExpr(affineApplyOperands.size()); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DAffineExpr.h252 AffineExpr getAffineSymbolExpr(unsigned position, MLIRContext *context);
320 e = getAffineSymbolExpr(N, ctx); in bindSymbols()
H A DBuilders.h152 AffineExpr getAffineSymbolExpr(unsigned position);
/llvm-project-15.0.7/mlir/lib/IR/
H A DBuilders.cpp289 AffineExpr Builder::getAffineSymbolExpr(unsigned position) { in getAffineSymbolExpr() function in Builder
290 return mlir::getAffineSymbolExpr(position, context); in getAffineSymbolExpr()
319 getAffineSymbolExpr(0)); in getSymbolIdentityMap()
H A DAffineExpr.cpp124 symbols.push_back(getAffineSymbolExpr(idx, getContext())); in shiftSymbols()
126 symbols.push_back(getAffineSymbolExpr(idx + shift, getContext())); in shiftSymbols()
293 return *this == mlir::getAffineSymbolExpr(position, getContext()); in isFunctionOfSymbol()
499 AffineExpr mlir::getAffineSymbolExpr(unsigned position, MLIRContext *context) { in getAffineSymbolExpr() function in mlir
906 : getAffineSymbolExpr(j - numDims, context); in getAffineExprFromFlatForm()
1010 getAffineSymbolExpr(j - numDims, context)); in getSemiAffineExprFromFlatForm()
H A DBuiltinTypes.cpp921 auto sym = getAffineSymbolExpr(nSymbols++, context); in makeStridedLinearLayoutMap()
937 mult = getAffineSymbolExpr(nSymbols++, context); in makeStridedLinearLayoutMap()
1009 ? getAffineSymbolExpr(nSymbols++, context) in makeCanonicalStridedLayoutExpr()
H A DAffineMap.cpp452 getAffineSymbolExpr(idx, getContext()); in compose()
611 symReplacements.push_back(getAffineSymbolExpr(numSymbols++, context)); in compressSymbols()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DTiling.cpp135 AffineExpr s0 = b.getAffineSymbolExpr(0); in computeMultiTileSizes()
136 AffineExpr s1 = b.getAffineSymbolExpr(1); in computeMultiTileSizes()
137 AffineExpr s2 = b.getAffineSymbolExpr(2); in computeMultiTileSizes()
159 AffineExpr s3 = builder.getAffineSymbolExpr(3); in computeMultiTileSizes()
H A DDropUnitDims.cpp122 symReplacements.push_back(getAffineSymbolExpr(symbol, context)); in replaceUnitDims()
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DAffineExpr.cpp89 return wrap(getAffineSymbolExpr(position, unwrap(ctx))); in mlirAffineSymbolExprGet()
/llvm-project-15.0.7/mlir/lib/AsmParser/
H A DAffineParser.cpp312 ? getAffineSymbolExpr(numSymbolOperands++, getContext()) in parseSSAIdExpr()
500 auto symbol = getAffineSymbolExpr(numSymbols++, getContext()); in parseSymbolIdList()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp608 AffineExpr lbTileParameter = b.getAffineSymbolExpr(origLbMap.getNumSymbols()); in setIntraTileBoundsParametric()
609 AffineExpr ubTileParameter = b.getAffineSymbolExpr(origUbMap.getNumSymbols()); in setIntraTileBoundsParametric()
685 AffineExpr tileParameter = b.getAffineSymbolExpr(origUbMap.getNumSymbols()); in setInterTileBoundsParametric()
1765 builder.getAffineSymbolExpr(0)), in coalesceLoops()
1771 /*numSymbols=*/1, builder.getAffineSymbolExpr(0), builder.getContext()); in coalesceLoops()
1795 builder.getAffineSymbolExpr(0))), in coalesceLoops()
1811 builder.getAffineDimExpr(0) % builder.getAffineSymbolExpr(0)), in coalesceLoops()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DAffineStructures.cpp998 memo[i] = getAffineSymbolExpr(i - getNumDimVars(), context); in getSliceBounds()
1454 memo[i] = getAffineSymbolExpr(i - numDims, context); in computeLocalVars()
1598 builder.getAffineSymbolExpr(std::distance(syms.begin(), symIt)); in alignAffineMapWithValues()
1602 replacement = builder.getAffineSymbolExpr(numSymbols++); in alignAffineMapWithValues()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp493 AffineExpr s0 = builder.getAffineSymbolExpr(0); in applyToOne()
494 AffineExpr s1 = builder.getAffineSymbolExpr(1); in applyToOne()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp595 AffineExpr s0 = getAffineSymbolExpr(0, b.getContext()); in updateBoundsForCyclicDistribution()
834 AffineExpr s0 = getAffineSymbolExpr(0, builder.getContext()); in makeTiledShape()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp627 : getAffineSymbolExpr(pos, ctx); in replaceDimOrSym()
688 : getAffineSymbolExpr(nSyms++, ctx)); in composeAffineMapAndOperands()
957 dimRemapping[i] = getAffineSymbolExpr(oldNumSyms + nextSym++, context); in canonicalizePromotedSymbols()
1042 symRemapping[i] = getAffineSymbolExpr(nextSym++, context); in canonicalizeMapOrSetAndOperands()
3610 : getAffineSymbolExpr(pos, parser.getContext())); in deduplicateAndResolveOperands()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp602 {builder.getAffineSymbolExpr(0), in genVectorMask()