Searched refs:boundMap (Results 1 – 7 of 7) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineStructures.cpp | 1173 AffineMap boundMap, in addBound() argument
1175 assert(boundMap.getNumDims() == getNumDimVars() && "dim mismatch"); in addBound()
1176 assert(boundMap.getNumSymbols() == getNumSymbolVars() && "symbol mismatch"); in addBound()
1183 assert((type != BoundType::EQ || boundMap.getNumResults() == 1) && in addBound()
1188 if (failed(flattenAlignedMapAndMergeLocals(boundMap, &flatExprs))) in addBound()
1190 assert(flatExprs.size() == boundMap.getNumResults()); in addBound()
1196 for (unsigned j = 0, e = boundMap.getNumInputs(); j < e; j++) { in addBound()
1208 for (unsigned i = boundMap.getNumInputs(); i < end; i++, j++) { in addBound()
1225 AffineMap boundMap) { in addBound() argument
1263 AffineMap boundMap, in addBound() argument
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 215 void getUpperBoundForIndex(Value value, AffineMap &boundMap, in getUpperBoundForIndex() argument
222 boundMap = AffineMap::getMultiDimIdentityMap(1, value.getContext()); in getUpperBoundForIndex()
224 canonicalizeMapAndOperands(&boundMap, &boundOperands); in getUpperBoundForIndex()
296 boundMap = AffineMap::getConstantMap(*ubConst, value.getContext()); in getUpperBoundForIndex()
309 boundMap = upperBounds[0]; in getUpperBoundForIndex()
312 canonicalizeMapAndOperands(&boundMap, &boundOperands); in getUpperBoundForIndex()
317 AffineMap boundMap; in getConstantUpperBoundForIndex() local
319 getUpperBoundForIndex(value, boundMap, boundOperands, in getConstantUpperBoundForIndex()
324 for (AffineExpr result : boundMap.getResults()) in getConstantUpperBoundForIndex()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/ |
| H A D | AffineStructures.h | 177 LogicalResult addBound(BoundType type, unsigned pos, AffineMap boundMap,
188 LogicalResult addBound(BoundType type, unsigned pos, AffineMap boundMap);
195 LogicalResult addBound(BoundType type, unsigned pos, AffineMap boundMap,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.cpp | 353 AffineMap boundMap; in getPackedTensorSizes() local
355 getUpperBoundForIndex(forOp.getUpperBound(), boundMap, boundOperands); in getPackedTensorSizes()
356 Value ubVal = b.createOrFold<AffineMinOp>(boundMap, boundOperands); in getPackedTensorSizes()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | AffineCanonicalizationUtils.cpp | 134 AffineMap boundMap = isMin ? addConstToResults(sliceBound, -1) : sliceBound; in canonicalizeMinMaxOp() local
138 AffineMap alignedBoundMap = boundMap.shiftDims(/*shift=*/1, /*offset=*/dimOp); in canonicalizeMinMaxOp()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 76 void getUpperBoundForIndex(Value value, AffineMap &boundMap,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 1690 static void printBound(AffineMapAttr boundMap, in printBound() argument
1693 AffineMap map = boundMap.getValue(); in printBound()
1726 p << boundMap; in printBound()
1786 AffineMap boundMap = in foldLoopBounds() local
1788 assert(boundMap.getNumResults() >= 1 && in foldLoopBounds()
1791 if (failed(boundMap.constantFold(operandConstants, foldedResults))) in foldLoopBounds()
|