Lines Matching refs:cst

69 ComputationSliceState::getSourceAsConstraints(FlatAffineValueConstraints &cst) {  in getSourceAsConstraints()  argument
71 cst.reset(/*numDims=*/ivs.size(), /*numSymbols=*/0, /*numLocals=*/0, ivs); in getSourceAsConstraints()
75 if (failed(cst.addAffineForOpDomain(loop))) in getSourceAsConstraints()
83 ComputationSliceState::getAsConstraints(FlatAffineValueConstraints *cst) { in getAsConstraints() argument
93 cst->reset(numDims, numSymbols, 0, values); in getAsConstraints()
99 assert(cst->containsVar(value) && "value expected to be present"); in getAsConstraints()
103 cst->addBound(FlatAffineValueConstraints::EQ, value, cOp.value()); in getAsConstraints()
105 if (failed(cst->addAffineForOpDomain(loop))) in getAsConstraints()
111 LogicalResult ret = cst->addSliceBounds(ivs, lbs, ubs, lbOperands[0]); in getAsConstraints()
353 assert(rank == cst.getNumDimVars() && "inconsistent memref region"); in getConstantBoundingSizeAndShape()
360 FlatAffineValueConstraints cstWithShapeBounds(cst); in getConstantBoundingSizeAndShape()
408 assert(pos < cst.getNumDimVars() && "invalid position"); in getLowerAndUpperBound()
412 assert(rank == cst.getNumDimVars() && "inconsistent memref region"); in getLowerAndUpperBound()
414 auto boundPairs = cst.getLowerAndUpperBound( in getLowerAndUpperBound()
415 pos, /*offset=*/0, /*num=*/rank, cst.getNumDimAndSymbolVars(), in getLowerAndUpperBound()
421 assert(lbMap.getNumInputs() == cst.getNumDimAndSymbolVars() - rank); in getLowerAndUpperBound()
422 assert(ubMap.getNumInputs() == cst.getNumDimAndSymbolVars() - rank); in getLowerAndUpperBound()
427 return cst.unionBoundingBox(*other.getConstraints()); in unionBoundingBox()
472 cst.reset(rank, loopDepth, /*numLocals=*/0, regionSymbols); in compute()
503 cst.reset(numDims, numSymbols, 0, operands); in compute()
514 if (failed(cst.addAffineForOpDomain(loop))) in compute()
523 cst.addBound(FlatAffineValueConstraints::EQ, symbol, constOp.value()); in compute()
533 cst.addInductionVarOrTerminalSymbol(operand); in compute()
537 cst.addSliceBounds(sliceState->ivs, sliceState->lbs, sliceState->ubs, in compute()
545 if (failed(cst.composeMap(&accessValueMap))) { in compute()
554 cst.setDimSymbolSeparation(cst.getNumDimAndSymbolVars() - rank); in compute()
563 cst.getValues(cst.getNumDimVars(), cst.getNumDimAndSymbolVars(), &vars); in compute()
568 cst.projectOut(var); in compute()
574 cst.projectOut(cst.getNumDimAndSymbolVars(), cst.getNumLocalVars()); in compute()
577 cst.constantFoldVarRange(/*pos=*/cst.getNumDimVars(), in compute()
578 /*num=*/cst.getNumSymbolVars()); in compute()
580 assert(cst.getNumDimVars() == rank && "unexpected MemRefRegion format"); in compute()
588 cst.addBound(FlatAffineValueConstraints::LB, /*pos=*/r, /*value=*/0); in compute()
591 cst.addBound(FlatAffineValueConstraints::UB, /*pos=*/r, in compute()
595 cst.removeTrivialRedundancy(); in compute()
598 LLVM_DEBUG(cst.dump()); in compute()
764 FlatAffineValueConstraints *cst) { in addMissingLoopIVBounds() argument
765 for (unsigned i = 0, e = cst->getNumDimVars(); i < e; ++i) { in addMissingLoopIVBounds()
766 auto value = cst->getValue(i); in addMissingLoopIVBounds()
770 if (failed(cst->addAffineForOpDomain(loop))) in addMissingLoopIVBounds()