15 FailureOr<ConstantOrScalableBound::BoundSize>
21 auto binop = dyn_cast<AffineBinaryOpExpr>(
map.
getResult(0));
24 auto matchConstant = [&](
AffineExpr expr, int64_t &constant) ->
bool {
25 if (
auto cst = dyn_cast<AffineConstantExpr>(expr)) {
26 constant = cst.getValue();
33 auto lhs = binop.getLHS();
34 auto rhs = binop.getRHS();
35 if ((matchConstant(lhs, cst) && isa<AffineSymbolExpr>(rhs)) ||
36 (matchConstant(rhs, cst) && isa<AffineSymbolExpr>(lhs))) {
46 Value value, std::optional<int64_t> dim,
unsigned vscaleMin,
49 using namespace presburger;
50 assert(vscaleMin <= vscaleMax);
54 auto defaultStopCondition = [&](
Value v, std::optional<int64_t> dim,
60 value.
getContext(), stopCondition ? stopCondition : defaultStopCondition,
61 vscaleMin, vscaleMax);
62 int64_t pos = scalableCstr.insert(value, dim,
false);
63 scalableCstr.processWorklist();
67 auto projectOutFn = [&](ValueDim p) {
68 bool isStartingPoint =
71 return p.first != scalableCstr.
getVscaleValue() && !isStartingPoint;
73 scalableCstr.projectOut(projectOutFn);
75 assert(scalableCstr.cstr.getNumDimAndSymbolVars() ==
76 scalableCstr.positionToValueDim.size() &&
77 "inconsistent mapping state");
80 for (int64_t i = 0; i < scalableCstr.cstr.getNumDimAndSymbolVars(); ++i) {
83 if (scalableCstr.positionToValueDim[i] !=
91 scalableCstr.cstr.getSliceBounds(pos, 1, value.
getContext(), &lowerBound,
92 &upperBound, closedUB);
94 auto invalidBound = [](
auto &bound) {
95 return !bound[0] || bound[0].getNumResults() != 1;
99 if (boundType == BoundType::EQ && !invalidBound(lowerBound) &&
100 lowerBound[0] == lowerBound[0]) {
101 return lowerBound[0];
102 }
else if (boundType == BoundType::LB && !invalidBound(lowerBound)) {
103 return lowerBound[0];
104 }
else if (boundType == BoundType::UB && !invalidBound(upperBound)) {
105 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
This class provides support for representing a failure result, or a valid value of type T.
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.
LogicalResult failure(bool isFailure=true)
Utility function to generate a LogicalResult.
@ 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.