13 FailureOr<ConstantOrScalableBound::BoundSize>
19 auto binop = dyn_cast<AffineBinaryOpExpr>(
map.
getResult(0));
22 auto matchConstant = [&](
AffineExpr expr, int64_t &constant) ->
bool {
23 if (
auto cst = dyn_cast<AffineConstantExpr>(expr)) {
24 constant = cst.getValue();
31 auto lhs = binop.getLHS();
32 auto rhs = binop.getRHS();
33 if ((matchConstant(lhs, cst) && isa<AffineSymbolExpr>(rhs)) ||
34 (matchConstant(rhs, cst) && isa<AffineSymbolExpr>(lhs))) {
42 FailureOr<ConstantOrScalableBound>
44 Value value, std::optional<int64_t> dim,
unsigned vscaleMin,
47 using namespace presburger;
48 assert(vscaleMin <= vscaleMax);
52 auto defaultStopCondition = [&](
Value v, std::optional<int64_t> dim,
58 value.
getContext(), stopCondition ? stopCondition : defaultStopCondition,
59 vscaleMin, vscaleMax);
60 int64_t pos = scalableCstr.insert(value, dim,
false);
61 scalableCstr.processWorklist();
64 if (scalableCstr.cstr.isEmpty()) {
70 auto projectOutFn = [&](ValueDim p) {
71 bool isStartingPoint =
74 return p.first != scalableCstr.
getVscaleValue() && !isStartingPoint;
76 scalableCstr.projectOut(projectOutFn);
77 scalableCstr.projectOutAnonymous(pos);
80 for (
unsigned i = 0, e = scalableCstr.cstr.getNumLocalVars(); i < e; ++i) {
81 scalableCstr.cstr.projectOut(scalableCstr.cstr.getNumDimAndSymbolVars());
84 assert(scalableCstr.cstr.getNumDimAndSymbolVars() ==
85 scalableCstr.positionToValueDim.size() &&
86 "inconsistent mapping state");
89 for (int64_t i = 0; i < scalableCstr.cstr.getNumDimAndSymbolVars(); ++i) {
92 if (scalableCstr.positionToValueDim[i] !=
100 scalableCstr.cstr.getSliceBounds(pos, 1, value.
getContext(), &lowerBound,
101 &upperBound, closedUB);
103 auto invalidBound = [](
auto &bound) {
104 return !bound[0] || bound[0].getNumResults() != 1;
108 if (boundType == BoundType::EQ && !invalidBound(lowerBound) &&
109 lowerBound[0] == upperBound[0]) {
110 return lowerBound[0];
111 }
else if (boundType == BoundType::LB && !invalidBound(lowerBound)) {
112 return lowerBound[0];
113 }
else if (boundType == BoundType::UB && !invalidBound(upperBound)) {
114 return upperBound[0];
function_ref< bool(Region *, ArrayRef< bool > visited)> StopConditionFn
Stop condition for traverseRegionGraph.
Base type for affine expression.
A multi-dimensional affine map Affine map's are immutable like Type's, and they are uniqued.
int64_t getSingleConstantResult() const
Returns the constant result of this map.
bool isSingleConstant() const
Returns true if this affine map is a single result constant function.
unsigned getNumResults() const
unsigned getNumInputs() const
AffineExpr getResult(unsigned idx) const
A helper class to be used with ValueBoundsOpInterface.
static constexpr int64_t kIndexValue
Dimension identifier to indicate a value is index-typed.
This class represents an instance of an SSA value in the MLIR system, representing a computable value...
MLIRContext * getContext() const
Utility to get the associated MLIRContext that this value is defined in.
BoundType
The type of bound: equal, lower bound or upper bound.
@ Mul
RHS of mul is always a constant or a symbolic expression.
A thin wrapper over an AffineMap which can represent a constant bound, or a scalable bound (in terms ...
FailureOr< BoundSize > getSize() const
Get the (possibly) scalable size of the bound, returns failure if the bound cannot be represented as ...
A version of ValueBoundsConstraintSet that can solve for scalable bounds.
static FailureOr< ConstantOrScalableBound > computeScalableBound(Value value, std::optional< int64_t > dim, unsigned vscaleMin, unsigned vscaleMax, presburger::BoundType boundType, bool closedUB=true, StopConditionFn stopCondition=nullptr)
Computes a (possibly) scalable bound for a given value.
Value getVscaleValue() const
Get the value of vscale. Returns nullptr vscale as not been encountered.