Home
last modified time | relevance | path

Searched refs:numDims (Results 1 – 25 of 30) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/IR/
H A DAffineMap.cpp31 : numDims(numDims), operandConsts(operandConsts) {} in AffineExprConstantFolder()
87 unsigned numDims; member in __anon1d589f0a0111::AffineExprConstantFolder
247 dimExprs.reserve(numDims); in getMultiDimIdentityMap()
248 for (unsigned i = 0; i < numDims; ++i) in getMultiDimIdentityMap()
259 for (unsigned i = 0, numDims = getNumDims(); i < numDims; ++i) { in isIdentity() local
296 return map->numDims; in getNumDims()
305 return map->numDims + map->numSymbols; in getNumInputs()
442 unsigned numDims = map.getNumDims(); in compose() local
545 unsigned numDims = 0; in compressDims() local
709 numDims = std::max(m.getNumDims(), numDims); in concatAffineMaps()
[all …]
H A DAffineExpr.cpp113 for (unsigned idx = offset; idx < numDims; ++idx) in shiftDims()
902 for (unsigned j = 0; j < numDims + numSymbols; j++) { in getAffineExprFromFlatForm()
905 auto id = j < numDims ? getAffineDimExpr(j, context) in getAffineExprFromFlatForm()
990 for (unsigned j = 0; j < numDims; ++j) { in getSemiAffineExprFromFlatForm()
1000 for (unsigned j = numDims; j < numDims + numSymbols; ++j) { in getSemiAffineExprFromFlatForm()
1007 std::pair<unsigned, signed> indexEntry(j - numDims, in getSemiAffineExprFromFlatForm()
1010 getAffineSymbolExpr(j - numDims, context)); in getSemiAffineExprFromFlatForm()
1024 if (flatExprs[numDims + numSymbols + it.index()] == 0) in getSemiAffineExprFromFlatForm()
1103 : numDims(numDims), numSymbols(numSymbols), numLocals(0) { in SimpleAffineExprFlattener()
1174 lhs, numDims, numSymbols, localExprs, context); in visitModExpr()
[all …]
H A DAffineMapDetail.h31 unsigned numDims; member
43 return std::get<0>(key) == numDims && std::get<1>(key) == numSymbols &&
56 res->numDims = std::get<0>(key); in construct()
H A DBuiltinTypes.cpp814 unsigned numDims = m.getNumDims(); in getStridesAndOffset() local
816 offset = simplifyAffineExpr(offset, numDims, numSymbols); in getStridesAndOffset()
818 stride = simplifyAffineExpr(stride, numDims, numSymbols); in getStridesAndOffset()
997 unsigned numDims = maps[0].getNumDims(), nSymbols = maps[0].getNumSymbols(); in makeCanonicalStridedLayoutExpr() local
1019 return simplifyAffineExpr(expr, numDims, nSymbols); in makeCanonicalStridedLayoutExpr()
/llvm-project-15.0.7/mlir/lib/AsmParser/
H A DAffineParser.cpp62 unsigned &numDims);
70 ParseResult parseDimIdList(unsigned &numDims);
72 ParseResult parseDimAndOptionalSymbolIdList(unsigned &numDims,
488 ParseResult AffineParser::parseDimIdList(unsigned &numDims) { in parseDimIdList() argument
490 auto dimension = getAffineDimExpr(numDims++, getContext()); in parseDimIdList()
511 if (parseDimIdList(numDims)) { in parseDimAndOptionalSymbolIdList()
524 unsigned numDims = 0, numSymbols = 0; in parseAffineMapOrIntegerSetInline() local
527 if (parseDimAndOptionalSymbolIdList(numDims, numSymbols)) in parseAffineMapOrIntegerSetInline()
531 return parseAffineMapRange(numDims, numSymbols, map); in parseAffineMapOrIntegerSetInline()
535 return parseIntegerSetConstraints(numDims, numSymbols, set); in parseAffineMapOrIntegerSetInline()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/
H A DGPUDialect.h50 MMAMatrixStorageType(unsigned numDims, const int64_t *dimShapes, in MMAMatrixStorageType()
52 : dimShapes(dimShapes), numDims(numDims), elementType(elementType), in MMAMatrixStorageType()
73 return ArrayRef<int64_t>(dimShapes, numDims); in getShape()
82 unsigned numDims; member
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/Analysis/
H A DAffineStructures.h43 unsigned numReservedCols, unsigned numDims,
49 numDims, numSymbols, numLocals)) { in IntegerPolyhedron()
61 FlatAffineValueConstraints(unsigned numDims = 0, unsigned numSymbols = 0,
66 /*numReservedCols=*/numDims + numSymbols +
68 numDims, numSymbols, numLocals, valArgs) {}
115 unsigned numReservedCols, unsigned numDims, unsigned numSymbols,
117 void reset(unsigned numDims = 0, unsigned numSymbols = 0,
120 unsigned numReservedCols, unsigned numDims, unsigned numSymbols,
122 void reset(unsigned numDims, unsigned numSymbols, unsigned numLocals,
593 LogicalResult getFlattenedAffineExpr(AffineExpr expr, unsigned numDims,
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DIntegerSet.cpp35 MlirIntegerSet mlirIntegerSetEmptyGet(MlirContext context, intptr_t numDims, in mlirIntegerSetEmptyGet() argument
37 return wrap(IntegerSet::getEmptySet(static_cast<unsigned>(numDims), in mlirIntegerSetEmptyGet()
42 MlirIntegerSet mlirIntegerSetGet(MlirContext context, intptr_t numDims, in mlirIntegerSetGet() argument
50 static_cast<unsigned>(numDims), static_cast<unsigned>(numSymbols), in mlirIntegerSetGet()
H A DAffineMap.cpp59 intptr_t numDims) { in mlirAffineMapMultiDimIdentityGet() argument
60 return wrap(AffineMap::getMultiDimIdentityMap(numDims, unwrap(ctx))); in mlirAffineMapMultiDimIdentityGet()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DAffineExprVisitor.h258 unsigned numDims; variable
273 SimpleAffineExprFlattener(unsigned numDims, unsigned numSymbols);
336 return numDims + numSymbols + numLocals + 1; in getNumCols()
339 inline unsigned getLocalVarStartIndex() const { return numDims + numSymbols; } in getLocalVarStartIndex()
340 inline unsigned getSymbolStartIndex() const { return numDims; } in getSymbolStartIndex()
H A DIntegerSet.h56 static IntegerSet getEmptySet(unsigned numDims, unsigned numSymbols, in getEmptySet() argument
60 return get(numDims, numSymbols, one, true); in getEmptySet()
H A DAffineMap.h71 static AffineMap getMultiDimIdentityMap(unsigned numDims,
353 unsigned getNumDims() const { return numDims; } in getNumDims()
354 void setNumDims(unsigned d) { numDims = d; } in setNumDims()
376 unsigned numDims = 0; member
H A DAffineExpr.h149 AffineExpr shiftDims(unsigned numDims, unsigned shift,
263 unsigned numDims, unsigned numSymbols,
302 AffineExpr simplifyAffineExpr(AffineExpr expr, unsigned numDims,
/llvm-project-15.0.7/mlir/include/mlir-c/
H A DIntegerSet.h70 intptr_t numDims,
79 mlirIntegerSetGet(MlirContext context, intptr_t numDims, intptr_t numSymbols,
H A DAffineMap.h91 mlirAffineMapMultiDimIdentityGet(MlirContext ctx, intptr_t numDims);
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DAffineStructures.cpp77 localVarCst->reset(numDims, numSymbols); in getFlattenedAffineExprs()
81 AffineExprFlattener flattener(numDims, numSymbols); in getFlattenedAffineExprs()
105 mlir::getFlattenedAffineExpr(AffineExpr expr, unsigned numDims, in getFlattenedAffineExpr() argument
1447 unsigned numDims = cst.getNumDimVars(); in computeLocalVars() local
1451 for (unsigned i = 0; i < numDims; i++) in computeLocalVars()
1453 for (unsigned i = numDims, e = numDims + numSyms; i < e; i++) in computeLocalVars()
1454 memo[i] = getAffineSymbolExpr(i - numDims, context); in computeLocalVars()
1463 if (!memo[numDims + numSyms + i] && in computeLocalVars()
1477 unsigned numDims = getNumDimVars(); in getIneqAsAffineValueMap() local
1480 assert(pos < numDims && "invalid position"); in getIneqAsAffineValueMap()
[all …]
H A DLoopAnalysis.cpp227 unsigned numDims = accessMap.getNumDims(); in isContiguousAccess() local
236 exprOperands.push_back(mapOperands[numDims + symExpr.getPosition()]); in isContiguousAccess()
H A DUtils.cpp86 unsigned numDims = ivs.size(); in getAsConstraints() local
93 cst->reset(numDims, numSymbols, 0, values); in getAsConstraints()
97 for (unsigned i = numDims, end = values.size(); i < end; ++i) { in getAsConstraints()
481 unsigned numDims = accessMap.getNumDims(); in compute() local
503 cst.reset(numDims, numSymbols, 0, operands); in compute()
507 for (unsigned i = 0; i < numDims + numSymbols; ++i) { in compute()
/llvm-project-15.0.7/mlir/unittests/Analysis/Presburger/
H A DUtils.h44 parsePresburgerSetFromPolyStrings(unsigned numDims, ArrayRef<StringRef> strs,
47 PresburgerSpace::getSetSpace(numDims, numSymbols));
H A DPresburgerSetTest.cpp90 static PresburgerSet makeSetFromPoly(unsigned numDims, in makeSetFromPoly() argument
93 PresburgerSet::getEmpty(PresburgerSpace::getSetSpace(numDims)); in makeSetFromPoly()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/
H A DAffineOps.cpp443 if (operands.size() > numDims) in printDimAndSymbolList()
455 numDims = opInfos.size(); in parseDimAndSymbolList()
475 if (opIt++ < numDims) { in verifyDimAndSymbolIdentifiers()
498 unsigned numDims; in parse() local
1575 unsigned numDims; in parseBound() local
2380 unsigned numDims; in parse() local
2389 if (set.getNumDims() != numDims) in parse()
2795 if (operands.size() != numDims) in printAffineMinMaxOp()
3432 auto numDims = getNumDims(); in verify() local
3435 getSteps().size() != numDims || getBody()->getNumArguments() != numDims) { in verify()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Analysis/Presburger/
H A DPresburgerSpace.h86 static PresburgerSpace getSetSpace(unsigned numDims = 0,
89 return PresburgerSpace(/*numDomain=*/0, /*numRange=*/numDims, numSymbols,
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp510 unsigned numDims = map.getNumDims(), numRes = map.getNumResults(); in createLoopRanges() local
512 SmallVector<Range, 4> res(numDims); in createLoopRanges()
528 unsigned numDims = map.getNumDims(), numRes = map.getNumResults(); in computeStaticLoopSizes() local
530 SmallVector<int64_t, 4> res(numDims, 0); in computeStaticLoopSizes()
/llvm-project-15.0.7/mlir/lib/Conversion/AffineToStandard/
H A DAffineToStandard.cpp284 auto numDims = integerSet.getNumDims(); in matchAndRewrite() local
286 operandsRef.take_front(numDims), in matchAndRewrite()
287 operandsRef.drop_front(numDims)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DIRAffine.cpp867 [](intptr_t numDims, intptr_t numSymbols, py::list exprs, in populateIRAffine()
885 context->get(), numDims, numSymbols, exprs.size(), in populateIRAffine()
893 [](intptr_t numDims, intptr_t numSymbols, in populateIRAffine()
896 mlirIntegerSetEmptyGet(context->get(), numDims, numSymbols); in populateIRAffine()

12