MLIR  20.0.0git
Classes | Public Types | Public Member Functions | Static Public Member Functions | Protected Member Functions | Protected Attributes | Static Protected Attributes | List of all members
mlir::presburger::IntegerRelation Class Reference

An IntegerRelation represents the set of points from a PresburgerSpace that satisfy a list of affine constraints. More...

#include "mlir/Analysis/Presburger/IntegerRelation.h"

+ Inheritance diagram for mlir::presburger::IntegerRelation:

Classes

struct  CountsSnapshot
 The struct CountsSnapshot stores the count of each VarKind, and also of each constraint type. More...
 

Public Types

enum class  Kind {
  IntegerRelation , IntegerPolyhedron , FlatLinearConstraints , FlatLinearValueConstraints ,
  FlatAffineValueConstraints , FlatAffineRelation
}
 All derived classes of IntegerRelation. More...
 

Public Member Functions

 IntegerRelation (unsigned numReservedInequalities, unsigned numReservedEqualities, unsigned numReservedCols, const PresburgerSpace &space)
 Constructs a relation reserving memory for the specified number of constraints and variables. More...
 
 IntegerRelation (const PresburgerSpace &space)
 Constructs a relation with the specified number of dimensions and symbols. More...
 
virtual ~IntegerRelation ()=default
 
virtual Kind getKind () const
 Return the kind of this IntegerRelation. More...
 
std::unique_ptr< IntegerRelationclone () const
 
const PresburgerSpacegetSpace () const
 Returns a reference to the underlying space. More...
 
void setSpace (const PresburgerSpace &oSpace)
 Set the space to oSpace, which should have the same number of ids as the current space. More...
 
void setSpaceExceptLocals (const PresburgerSpace &oSpace)
 Set the space to oSpace, which should not have any local ids. More...
 
void setId (VarKind kind, unsigned i, Identifier id)
 Set the identifier for the ith variable of the specified kind of the IntegerRelation's PresburgerSpace. More...
 
void resetIds ()
 
ArrayRef< IdentifiergetIds (VarKind kind)
 Get the identifiers for the variables of specified varKind. More...
 
PresburgerSpace getSpaceWithoutLocals () const
 Returns a copy of the space without locals. More...
 
void append (const IntegerRelation &other)
 Appends constraints from other into this. More...
 
IntegerRelation intersect (IntegerRelation other) const
 Return the intersection of the two relations. More...
 
bool isEqual (const IntegerRelation &other) const
 Return whether this and other are equal. More...
 
bool isObviouslyEqual (const IntegerRelation &other) const
 Perform a quick equality check on this and other. More...
 
bool isSubsetOf (const IntegerRelation &other) const
 Return whether this is a subset of the given IntegerRelation. More...
 
DynamicAPInt atEq (unsigned i, unsigned j) const
 Returns the value at the specified equality row and column. More...
 
int64_t atEq64 (unsigned i, unsigned j) const
 The same, but casts to int64_t. More...
 
DynamicAPInt & atEq (unsigned i, unsigned j)
 
DynamicAPInt atIneq (unsigned i, unsigned j) const
 Returns the value at the specified inequality row and column. More...
 
int64_t atIneq64 (unsigned i, unsigned j) const
 The same, but casts to int64_t. More...
 
DynamicAPInt & atIneq (unsigned i, unsigned j)
 
unsigned getNumConstraints () const
 
unsigned getNumDomainVars () const
 
unsigned getNumRangeVars () const
 
unsigned getNumSymbolVars () const
 
unsigned getNumLocalVars () const
 
unsigned getNumDimVars () const
 
unsigned getNumDimAndSymbolVars () const
 
unsigned getNumVars () const
 
unsigned getNumCols () const
 Returns the number of columns in the constraint system. More...
 
unsigned getNumEqualities () const
 
unsigned getNumInequalities () const
 
unsigned getNumReservedEqualities () const
 
unsigned getNumReservedInequalities () const
 
ArrayRef< DynamicAPInt > getEquality (unsigned idx) const
 
ArrayRef< DynamicAPInt > getInequality (unsigned idx) const
 
SmallVector< int64_t, 8 > getEquality64 (unsigned idx) const
 The same, but casts to int64_t. More...
 
SmallVector< int64_t, 8 > getInequality64 (unsigned idx) const
 
IntMatrix getInequalities () const
 
unsigned getNumVarKind (VarKind kind) const
 Get the number of vars of the specified kind. More...
 
unsigned getVarKindOffset (VarKind kind) const
 Return the index at which the specified kind of vars starts. More...
 
unsigned getVarKindEnd (VarKind kind) const
 Return the index at Which the specified kind of vars ends. More...
 
unsigned getVarKindOverlap (VarKind kind, unsigned varStart, unsigned varLimit) const
 Get the number of elements of the specified kind in the range [varStart, varLimit). More...
 
VarKind getVarKindAt (unsigned pos) const
 Return the VarKind of the var at the specified position. More...
 
CountsSnapshot getCounts () const
 
void truncate (const CountsSnapshot &counts)
 
virtual unsigned insertVar (VarKind kind, unsigned pos, unsigned num=1)
 Insert num variables of the specified kind at position pos. More...
 
unsigned appendVar (VarKind kind, unsigned num=1)
 Append num variables of the specified kind after the last variable of that kind. More...
 
void addInequality (ArrayRef< DynamicAPInt > inEq)
 Adds an inequality (>= 0) from the coefficients specified in inEq. More...
 
void addInequality (ArrayRef< int64_t > inEq)
 
void addEquality (ArrayRef< DynamicAPInt > eq)
 Adds an equality from the coefficients specified in eq. More...
 
void addEquality (ArrayRef< int64_t > eq)
 
virtual void eliminateRedundantLocalVar (unsigned posA, unsigned posB)
 Eliminate the posB^th local variable, replacing every instance of it with the posA^th local variable. More...
 
void removeVar (VarKind kind, unsigned pos)
 Removes variables of the specified kind with the specified pos (or within the specified range) from the system. More...
 
virtual void removeVarRange (VarKind kind, unsigned varStart, unsigned varLimit)
 
void removeVar (unsigned pos)
 Removes the specified variable from the system. More...
 
void removeEquality (unsigned pos)
 
void removeInequality (unsigned pos)
 
void removeEqualityRange (unsigned start, unsigned end)
 Remove the (in)equalities at positions [start, end). More...
 
void removeInequalityRange (unsigned start, unsigned end)
 
MaybeOptimum< SmallVector< Fraction, 8 > > findRationalLexMin () const
 Get the lexicographically minimum rational point satisfying the constraints. More...
 
MaybeOptimum< SmallVector< DynamicAPInt, 8 > > findIntegerLexMin () const
 Same as above, but returns lexicographically minimal integer point. More...
 
virtual void swapVar (unsigned posA, unsigned posB)
 Swap the posA^th variable with the posB^th variable. More...
 
void clearConstraints ()
 Removes all equalities and inequalities. More...
 
void setAndEliminate (unsigned pos, ArrayRef< DynamicAPInt > values)
 Sets the values.size() variables starting at pos to the specified values and removes them. More...
 
void setAndEliminate (unsigned pos, ArrayRef< int64_t > values)
 
virtual void clearAndCopyFrom (const IntegerRelation &other)
 Replaces the contents of this IntegerRelation with other. More...
 
void getLowerAndUpperBoundIndices (unsigned pos, SmallVectorImpl< unsigned > *lbIndices, SmallVectorImpl< unsigned > *ubIndices, SmallVectorImpl< unsigned > *eqIndices=nullptr, unsigned offset=0, unsigned num=0) const
 Gather positions of all lower and upper bounds of the variable at pos, and optionally any equalities on it. More...
 
bool isEmpty () const
 Checks for emptiness by performing variable elimination on all variables, running the GCD test on each equality constraint, and checking for invalid constraints. More...
 
bool isObviouslyEmpty () const
 Performs GCD checks and invalid constraint checks. More...
 
bool isEmptyByGCDTest () const
 Runs the GCD test on all equality constraints. More...
 
bool isIntegerEmpty () const
 Returns true if the set of constraints is found to have no solution, false if a solution exists. More...
 
IntMatrix getBoundedDirections () const
 Returns a matrix where each row is a vector along which the polytope is bounded. More...
 
std::optional< SmallVector< DynamicAPInt, 8 > > findIntegerSample () const
 Find an integer sample point satisfying the constraints using a branch and bound algorithm with generalized basis reduction, with some additional processing using Simplex for unbounded sets. More...
 
std::optional< DynamicAPInt > computeVolume () const
 Compute an overapproximation of the number of integer points in the relation. More...
 
bool containsPoint (ArrayRef< DynamicAPInt > point) const
 Returns true if the given point satisfies the constraints, or false otherwise. More...
 
bool containsPoint (ArrayRef< int64_t > point) const
 
std::optional< SmallVector< DynamicAPInt, 8 > > containsPointNoLocal (ArrayRef< DynamicAPInt > point) const
 Given the values of non-local vars, return a satisfying assignment to the local if one exists, or an empty optional otherwise. More...
 
std::optional< SmallVector< DynamicAPInt, 8 > > containsPointNoLocal (ArrayRef< int64_t > point) const
 
DivisionRepr getLocalReprs (std::vector< MaybeLocalRepr > *repr=nullptr) const
 Returns a DivisonRepr representing the division representation of local variables in the constraint system. More...
 
void addBound (BoundType type, unsigned pos, const DynamicAPInt &value)
 Adds a constant bound for the specified variable. More...
 
void addBound (BoundType type, unsigned pos, int64_t value)
 
void addBound (BoundType type, ArrayRef< DynamicAPInt > expr, const DynamicAPInt &value)
 Adds a constant bound for the specified expression. More...
 
void addBound (BoundType type, ArrayRef< int64_t > expr, int64_t value)
 
void addLocalFloorDiv (ArrayRef< DynamicAPInt > dividend, const DynamicAPInt &divisor)
 Adds a new local variable as the floordiv of an affine function of other variables, the coefficients of which are provided in dividend and with respect to a positive constant divisor. More...
 
void addLocalFloorDiv (ArrayRef< int64_t > dividend, int64_t divisor)
 
void projectOut (unsigned pos, unsigned num)
 Projects out (aka eliminates) num variables starting at position pos. More...
 
void projectOut (unsigned pos)
 
LogicalResult constantFoldVar (unsigned pos)
 Tries to fold the specified variable to a constant using a trivial equality detection; if successful, the constant is substituted for the variable everywhere in the constraint system and then removed from the system. More...
 
void constantFoldVarRange (unsigned pos, unsigned num)
 This method calls constantFoldVar for the specified range of variables, num variables starting at position pos. More...
 
LogicalResult unionBoundingBox (const IntegerRelation &other)
 Updates the constraints to be the smallest bounding (enclosing) box that contains the points of this set and that of other, with the symbols being treated specially. More...
 
std::optional< DynamicAPInt > getConstantBoundOnDimSize (unsigned pos, SmallVectorImpl< DynamicAPInt > *lb=nullptr, DynamicAPInt *boundFloorDivisor=nullptr, SmallVectorImpl< DynamicAPInt > *ub=nullptr, unsigned *minLbPos=nullptr, unsigned *minUbPos=nullptr) const
 Returns the smallest known constant bound for the extent of the specified variable (pos^th), i.e., the smallest known constant that is greater than or equal to 'exclusive upper bound' - 'lower bound' of the variable. More...
 
std::optional< int64_t > getConstantBoundOnDimSize64 (unsigned pos, SmallVectorImpl< int64_t > *lb=nullptr, int64_t *boundFloorDivisor=nullptr, SmallVectorImpl< int64_t > *ub=nullptr, unsigned *minLbPos=nullptr, unsigned *minUbPos=nullptr) const
 The same, but casts to int64_t. More...
 
std::optional< DynamicAPInt > getConstantBound (BoundType type, unsigned pos) const
 Returns the constant bound for the pos^th variable if there is one; std::nullopt otherwise. More...
 
std::optional< int64_t > getConstantBound64 (BoundType type, unsigned pos) const
 The same, but casts to int64_t. More...
 
void removeIndependentConstraints (unsigned pos, unsigned num)
 Removes constraints that are independent of (i.e., do not have a coefficient) variables in the range [pos, pos + num). More...
 
bool isHyperRectangular (unsigned pos, unsigned num) const
 Returns true if the set can be trivially detected as being hyper-rectangular on the specified contiguous set of variables. More...
 
void removeTrivialRedundancy ()
 Removes duplicate constraints, trivially true constraints, and constraints that can be detected as redundant as a result of differing only in their constant term part. More...
 
void removeRedundantInequalities ()
 A more expensive check than removeTrivialRedundancy to detect redundant inequalities. More...
 
void removeRedundantConstraints ()
 Removes redundant constraints using Simplex. More...
 
void removeDuplicateDivs ()
 
void simplify ()
 Simplify the constraint system by removing canonicalizing constraints and removing redundant constraints. More...
 
void convertVarKind (VarKind srcKind, unsigned varStart, unsigned varLimit, VarKind dstKind, unsigned pos)
 Converts variables of kind srcKind in the range [varStart, varLimit) to variables of kind dstKind. More...
 
void convertVarKind (VarKind srcKind, unsigned varStart, unsigned varLimit, VarKind dstKind)
 
void convertToLocal (VarKind kind, unsigned varStart, unsigned varLimit)
 
void mergeAndAlignSymbols (IntegerRelation &other)
 Merge and align symbol variables of this and other with respect to identifiers. More...
 
unsigned mergeLocalVars (IntegerRelation &other)
 Adds additional local vars to the sets such that they both have the union of the local vars in each set, without changing the set of points that lie in this and other. More...
 
bool hasOnlyDivLocals () const
 Check whether all local ids have a division representation. More...
 
void setDimSymbolSeparation (unsigned newSymbolCount)
 Changes the partition between dimensions and symbols. More...
 
IntegerPolyhedron getDomainSet () const
 Return a set corresponding to all points in the domain of the relation. More...
 
IntegerPolyhedron getRangeSet () const
 Return a set corresponding to all points in the range of the relation. More...
 
void intersectDomain (const IntegerPolyhedron &poly)
 Intersect the given poly with the domain in-place. More...
 
void intersectRange (const IntegerPolyhedron &poly)
 Intersect the given poly with the range in-place. More...
 
void inverse ()
 Invert the relation i.e., swap its domain and range. More...
 
void compose (const IntegerRelation &rel)
 Let the relation this be R1, and the relation rel be R2. More...
 
void applyDomain (const IntegerRelation &rel)
 Given a relation rel, apply the relation to the domain of this relation. More...
 
void applyRange (const IntegerRelation &rel)
 Given a relation rel, apply the relation to the range of this relation. More...
 
void mergeAndCompose (const IntegerRelation &other)
 Given a relation other: (A -> B), this operation merges the symbol and local variables and then takes the composition of other on this: (B -> C). More...
 
PresburgerRelation computeReprWithOnlyDivLocals () const
 Compute an equivalent representation of the same set, such that all local vars in all disjuncts have division representations. More...
 
SymbolicLexOpt findSymbolicIntegerLexMin () const
 Compute the symbolic integer lexmin of the relation. More...
 
SymbolicLexOpt findSymbolicIntegerLexMax () const
 Same as findSymbolicIntegerLexMin but produces lexmax instead of lexmin. More...
 
PresburgerRelation subtract (const PresburgerRelation &set) const
 Return the set difference of this set and the given set, i.e., return this \ set. More...
 
void removeTrivialEqualities ()
 
bool isFullDim ()
 
void print (raw_ostream &os) const
 
void dump () const
 

Static Public Member Functions

static IntegerRelation getUniverse (const PresburgerSpace &space)
 Return a system with no constraints, i.e., one which is satisfied by all points. More...
 
static IntegerRelation getEmpty (const PresburgerSpace &space)
 Return an empty system containing an invalid equation 0 = 1. More...
 
static bool classof (const IntegerRelation *cst)
 

Protected Member Functions

bool hasInvalidConstraint () const
 Checks all rows of equality/inequality constraints for trivial contradictions (for example: 1 == 0, 0 >= 1), which may have surfaced after elimination. More...
 
template<bool isLower>
std::optional< DynamicAPInt > computeConstantLowerOrUpperBound (unsigned pos)
 Returns the constant lower bound if isLower is true, and the upper bound if isLower is false. More...
 
template<bool isLower>
std::optional< int64_t > computeConstantLowerOrUpperBound64 (unsigned pos)
 The same, but casts to int64_t. More...
 
LogicalResult gaussianEliminateVar (unsigned position)
 Eliminates a single variable at position from equality and inequality constraints. More...
 
void removeRedundantLocalVars ()
 Removes local variables using equalities. More...
 
unsigned gaussianEliminateVars (unsigned posStart, unsigned posLimit)
 Eliminates variables from equality and inequality constraints in column range [posStart, posLimit). More...
 
bool gaussianEliminate ()
 Perform a Gaussian elimination operation to reduce all equations to standard form. More...
 
virtual void fourierMotzkinEliminate (unsigned pos, bool darkShadow=false, bool *isResultIntegerExact=nullptr)
 Eliminates the variable at the specified position using Fourier-Motzkin variable elimination, but uses Gaussian elimination if there is an equality involving that variable. More...
 
void gcdTightenInequalities ()
 Tightens inequalities given that we are dealing with integer spaces. More...
 
void normalizeConstraintsByGCD ()
 Normalized each constraints by the GCD of its coefficients. More...
 
bool findConstraintWithNonZeroAt (unsigned colIdx, bool isEq, unsigned *rowIdx) const
 Searches for a constraint with a non-zero coefficient at colIdx in equality (isEq=true) or inequality (isEq=false) constraints. More...
 
bool isColZero (unsigned pos) const
 Returns true if the pos^th column is all zero for both inequalities and equalities. More...
 
bool removeDuplicateConstraints ()
 Checks for identical inequalities and eliminates redundant inequalities. More...
 
virtual bool hasConsistentState () const
 Returns false if the fields corresponding to various variable counts, or equality/inequality buffer sizes aren't consistent; true otherwise. More...
 
virtual void printSpace (raw_ostream &os) const
 Prints the number of constraints, dimensions, symbols and locals in the IntegerRelation. More...
 
void removeVarRange (unsigned varStart, unsigned varLimit)
 Removes variables in the column range [varStart, varLimit), and copies any remaining valid data into place, updates member variables, and resizes arrays as needed. More...
 
void truncateVarKind (VarKind kind, unsigned num)
 Truncate the vars of the specified kind to the specified number by dropping some vars at the end. More...
 
void truncateVarKind (VarKind kind, const CountsSnapshot &counts)
 Truncate the vars to the number in the space of the specified CountsSnapshot. More...
 

Protected Attributes

PresburgerSpace space
 
IntMatrix equalities
 Coefficients of affine equalities (in == 0 form). More...
 
IntMatrix inequalities
 Coefficients of affine inequalities (in >= 0 form). More...
 

Static Protected Attributes

constexpr static unsigned kExplosionFactor = 32
 A parameter that controls detection of an unrealistic number of constraints. More...
 

Detailed Description

An IntegerRelation represents the set of points from a PresburgerSpace that satisfy a list of affine constraints.

Affine constraints can be inequalities or equalities in the form:

Inequality: c_0*x_0 + c_1*x_1 + .... + c_{n-1}*x_{n-1} + c_n >= 0 Equality : c_0*x_0 + c_1*x_1 + .... + c_{n-1}*x_{n-1} + c_n == 0

where c_0, c_1, ..., c_n are integers and n is the total number of variables in the space.

Such a relation corresponds to the set of integer points lying in a convex polyhedron. For example, consider the relation: (x) -> (y) : (1 <= x <= 7, x = 2y) These can be thought of as points in the polyhedron: (x, y) : (1 <= x <= 7, x = 2y) This relation contains the pairs (2, 1), (4, 2), and (6, 3).

Since IntegerRelation makes a distinction between dimensions, VarKind::Range and VarKind::Domain should be used to refer to dimension variables.

Definition at line 64 of file IntegerRelation.h.

Member Enumeration Documentation

◆ Kind

All derived classes of IntegerRelation.

Enumerator
IntegerRelation 
IntegerPolyhedron 
FlatLinearConstraints 
FlatLinearValueConstraints 
FlatAffineValueConstraints 
FlatAffineRelation 

Definition at line 67 of file IntegerRelation.h.

Constructor & Destructor Documentation

◆ IntegerRelation() [1/2]

mlir::presburger::IntegerRelation::IntegerRelation ( unsigned  numReservedInequalities,
unsigned  numReservedEqualities,
unsigned  numReservedCols,
const PresburgerSpace space 
)
inline

Constructs a relation reserving memory for the specified number of constraints and variables.

Definition at line 78 of file IntegerRelation.h.

References mlir::presburger::PresburgerSpace::getNumVars(), and space.

Referenced by getConstantBound(), and getUniverse().

◆ IntegerRelation() [2/2]

mlir::presburger::IntegerRelation::IntegerRelation ( const PresburgerSpace space)
inlineexplicit

Constructs a relation with the specified number of dimensions and symbols.

Definition at line 89 of file IntegerRelation.h.

◆ ~IntegerRelation()

virtual mlir::presburger::IntegerRelation::~IntegerRelation ( )
virtualdefault

Member Function Documentation

◆ addBound() [1/4]

void IntegerRelation::addBound ( BoundType  type,
ArrayRef< DynamicAPInt >  expr,
const DynamicAPInt &  value 
)

◆ addBound() [2/4]

void mlir::presburger::IntegerRelation::addBound ( BoundType  type,
ArrayRef< int64_t >  expr,
int64_t  value 
)
inline

Definition at line 458 of file IntegerRelation.h.

References addBound(), and mlir::presburger::getDynamicAPIntVec().

◆ addBound() [3/4]

void IntegerRelation::addBound ( BoundType  type,
unsigned  pos,
const DynamicAPInt &  value 
)

◆ addBound() [4/4]

void mlir::presburger::IntegerRelation::addBound ( BoundType  type,
unsigned  pos,
int64_t  value 
)
inline

Definition at line 451 of file IntegerRelation.h.

References addBound().

◆ addEquality() [1/2]

void IntegerRelation::addEquality ( ArrayRef< DynamicAPInt >  eq)

◆ addEquality() [2/2]

void mlir::presburger::IntegerRelation::addEquality ( ArrayRef< int64_t >  eq)
inline

Definition at line 318 of file IntegerRelation.h.

References addEquality(), and mlir::presburger::getDynamicAPIntVec().

◆ addInequality() [1/2]

void IntegerRelation::addInequality ( ArrayRef< DynamicAPInt >  inEq)

◆ addInequality() [2/2]

void mlir::presburger::IntegerRelation::addInequality ( ArrayRef< int64_t >  inEq)
inline

Definition at line 313 of file IntegerRelation.h.

References addInequality(), and mlir::presburger::getDynamicAPIntVec().

◆ addLocalFloorDiv() [1/2]

void IntegerRelation::addLocalFloorDiv ( ArrayRef< DynamicAPInt >  dividend,
const DynamicAPInt &  divisor 
)

Adds a new local variable as the floordiv of an affine function of other variables, the coefficients of which are provided in dividend and with respect to a positive constant divisor.

Adds a new local variable as the floordiv of an affine function of other variables, the coefficients of which are provided in 'dividend' and with respect to a positive constant 'divisor'.

Two constraints are added to the system to capture equivalence with the floordiv: q = dividend floordiv c <=> c*q <= dividend <= c*q + c - 1.

Two constraints are added to the system to capture equivalence with the floordiv. q = expr floordiv c <=> c*q <= expr <= c*q + c - 1.

Definition at line 1514 of file IntegerRelation.cpp.

References addInequality(), appendVar(), mlir::presburger::getDivLowerBound(), mlir::presburger::getDivUpperBound(), getNumCols(), and mlir::presburger::Local.

Referenced by addLocalFloorDiv().

◆ addLocalFloorDiv() [2/2]

void mlir::presburger::IntegerRelation::addLocalFloorDiv ( ArrayRef< int64_t >  dividend,
int64_t  divisor 
)
inline

◆ append()

void IntegerRelation::append ( const IntegerRelation other)

◆ appendVar()

unsigned IntegerRelation::appendVar ( VarKind  kind,
unsigned  num = 1 
)

Append num variables of the specified kind after the last variable of that kind.

The coefficient columns corresponding to the added variables are initialized to zero. Return the absolute column position (i.e., not relative to the kind of variable) of the first appended variable.

Definition at line 363 of file IntegerRelation.cpp.

References getNumVarKind(), and insertVar().

Referenced by addLocalFloorDiv(), mlir::FlatLinearConstraints::appendDimVar(), mlir::FlatLinearConstraints::appendLocalVar(), mlir::FlatLinearConstraints::appendSymbolVar(), compose(), mlir::affine::MemRefAccess::getAccessRelation(), mlir::ValueBoundsConstraintSet::insert(), intersectDomain(), and intersectRange().

◆ applyDomain()

void IntegerRelation::applyDomain ( const IntegerRelation rel)

Given a relation rel, apply the relation to the domain of this relation.

R1: i -> j : (0 <= i < 2, j = i) R2: i -> k : (k = i floordiv 2) R3: k -> j : (0 <= k < 1, 2k <= j <= 2k + 1)

R1 = {(0, 0), (1, 1)}. R2 maps both 0 and 1 to 0. So R3 = {(0, 0), (0, 1)}.

Formally, R1.applyDomain(R2) = R2.inverse().compose(R1).

Definition at line 2495 of file IntegerRelation.cpp.

References compose(), and inverse().

◆ applyRange()

void IntegerRelation::applyRange ( const IntegerRelation rel)

Given a relation rel, apply the relation to the range of this relation.

Formally, R1.applyRange(R2) is the same as R1.compose(R2) but we provide this for uniformity with applyDomain.

Definition at line 2501 of file IntegerRelation.cpp.

References compose().

◆ atEq() [1/2]

DynamicAPInt& mlir::presburger::IntegerRelation::atEq ( unsigned  i,
unsigned  j 
)
inline

Definition at line 185 of file IntegerRelation.h.

References equalities.

◆ atEq() [2/2]

DynamicAPInt mlir::presburger::IntegerRelation::atEq ( unsigned  i,
unsigned  j 
) const
inline

◆ atEq64()

int64_t mlir::presburger::IntegerRelation::atEq64 ( unsigned  i,
unsigned  j 
) const
inline

The same, but casts to int64_t.

This is unsafe and will assert-fail if the value does not fit in an int64_t.

Definition at line 182 of file IntegerRelation.h.

References equalities.

◆ atIneq() [1/2]

DynamicAPInt& mlir::presburger::IntegerRelation::atIneq ( unsigned  i,
unsigned  j 
)
inline

Definition at line 196 of file IntegerRelation.h.

References inequalities.

◆ atIneq() [2/2]

DynamicAPInt mlir::presburger::IntegerRelation::atIneq ( unsigned  i,
unsigned  j 
) const
inline

◆ atIneq64()

int64_t mlir::presburger::IntegerRelation::atIneq64 ( unsigned  i,
unsigned  j 
) const
inline

The same, but casts to int64_t.

This is unsafe and will assert-fail if the value does not fit in an int64_t.

Definition at line 193 of file IntegerRelation.h.

References inequalities.

◆ classof()

static bool mlir::presburger::IntegerRelation::classof ( const IntegerRelation cst)
inlinestatic

Definition at line 114 of file IntegerRelation.h.

◆ clearAndCopyFrom()

void IntegerRelation::clearAndCopyFrom ( const IntegerRelation other)
virtual

Replaces the contents of this IntegerRelation with other.

Definition at line 563 of file IntegerRelation.cpp.

Referenced by getFlattenedAffineExprs().

◆ clearConstraints()

void IntegerRelation::clearConstraints ( )

Removes all equalities and inequalities.

Definition at line 479 of file IntegerRelation.cpp.

References equalities, inequalities, and mlir::presburger::Matrix< T >::resizeVertically().

Referenced by unionBoundingBox().

◆ clone()

std::unique_ptr< IntegerRelation > IntegerRelation::clone ( ) const

Definition at line 50 of file IntegerRelation.cpp.

◆ compose()

void IntegerRelation::compose ( const IntegerRelation rel)

Let the relation this be R1, and the relation rel be R2.

Modifies R1 to be the composition of R1 and R2: R1;R2.

Formally, if R1: A -> B, and R2: B -> C, then this function returns a relation R3: A -> C such that a point (a, c) belongs to R3 iff there exists b such that (a, b) is in R1 and, (b, c) is in R2.

Definition at line 2469 of file IntegerRelation.cpp.

References appendVar(), convertVarKind(), mlir::presburger::Domain, getDomainSet(), getNumRangeVars(), getRangeSet(), getSpace(), intersectRange(), mlir::presburger::Local, and mlir::presburger::Range.

Referenced by applyDomain(), applyRange(), and mlir::presburger::PresburgerRelation::compose().

◆ computeConstantLowerOrUpperBound()

template<bool isLower>
std::optional< DynamicAPInt > IntegerRelation::computeConstantLowerOrUpperBound ( unsigned  pos)
protected

Returns the constant lower bound if isLower is true, and the upper bound if isLower is false.

Definition at line 1721 of file IntegerRelation.cpp.

References atEq(), findEqualityToConstant(), getNumCols(), getNumVars(), and projectOut().

◆ computeConstantLowerOrUpperBound64()

template<bool isLower>
std::optional<int64_t> mlir::presburger::IntegerRelation::computeConstantLowerOrUpperBound64 ( unsigned  pos)
inlineprotected

The same, but casts to int64_t.

This is unsafe and will assert-fail if the value does not fit in an int64_t.

Definition at line 773 of file IntegerRelation.h.

◆ computeReprWithOnlyDivLocals()

PresburgerRelation IntegerRelation::computeReprWithOnlyDivLocals ( ) const

Compute an equivalent representation of the same set, such that all local vars in all disjuncts have division representations.

This representation may involve local vars that correspond to divisions, and may also be a union of convex disjuncts.

Definition at line 223 of file IntegerRelation.cpp.

References mlir::presburger::SymbolicLexSimplex::computeSymbolicIntegerLexMin(), copy(), mlir::presburger::PWMAFunction::getDomain(), getNumLocalVars(), mlir::presburger::PresburgerSpace::getSetSpace(), getSpace(), mlir::presburger::SymbolicLexOpt::lexopt, mlir::presburger::Local, mlir::presburger::PresburgerSpace::removeVarRange(), mlir::presburger::PresburgerRelation::setSpace(), space, mlir::presburger::SymbolicLexOpt::unboundedDomain, and mlir::presburger::PresburgerSet::unionSet().

◆ computeVolume()

std::optional< DynamicAPInt > IntegerRelation::computeVolume ( ) const

Compute an overapproximation of the number of integer points in the relation.

Symbol vars currently not supported. If the computed overapproximation is infinite, an empty optional is returned.

Definition at line 1241 of file IntegerRelation.cpp.

References mlir::presburger::Simplex::computeIntegerBounds(), getNumDimAndSymbolVars(), getNumSymbolVars(), getNumVars(), mlir::presburger::SimplexBase::isEmpty(), max(), and min().

◆ constantFoldVar()

LogicalResult IntegerRelation::constantFoldVar ( unsigned  pos)

Tries to fold the specified variable to a constant using a trivial equality detection; if successful, the constant is substituted for the variable everywhere in the constraint system and then removed from the system.

Definition at line 1559 of file IntegerRelation.cpp.

References atEq(), findEqualityToConstant(), getNumCols(), getNumVars(), and setAndEliminate().

Referenced by constantFoldVarRange().

◆ constantFoldVarRange()

void IntegerRelation::constantFoldVarRange ( unsigned  pos,
unsigned  num 
)

This method calls constantFoldVar for the specified range of variables, num variables starting at position pos.

Definition at line 1572 of file IntegerRelation.cpp.

References constantFoldVar().

◆ containsPoint() [1/2]

bool IntegerRelation::containsPoint ( ArrayRef< DynamicAPInt >  point) const

Returns true if the given point satisfies the constraints, or false otherwise.

A point satisfies an equality iff the value of the equality at the expression is zero, and it satisfies an inequality iff the value of the inequality at that point is non-negative.

Takes the values of all vars including locals.

Definition at line 991 of file IntegerRelation.cpp.

References getEquality(), getInequality(), getNumEqualities(), getNumInequalities(), and valueAt().

Referenced by containsPoint().

◆ containsPoint() [2/2]

bool mlir::presburger::IntegerRelation::containsPoint ( ArrayRef< int64_t >  point) const
inline

Definition at line 426 of file IntegerRelation.h.

References containsPoint(), and mlir::presburger::getDynamicAPIntVec().

◆ containsPointNoLocal() [1/2]

std::optional< SmallVector< DynamicAPInt, 8 > > IntegerRelation::containsPointNoLocal ( ArrayRef< DynamicAPInt >  point) const

Given the values of non-local vars, return a satisfying assignment to the local if one exists, or an empty optional otherwise.

Just substitute the values given and check if an integer sample exists for the local vars.

TODO: this could be made more efficient by handling divisions separately. Instead of finding an integer sample over all the locals, we can first compute the values of the locals that have division representations and only use the integer emptiness check for the locals that don't have this. Handling this correctly requires ordering the divs, though.

Definition at line 1012 of file IntegerRelation.cpp.

References copy(), getNumLocalVars(), getNumVars(), getVarKindOffset(), and mlir::presburger::Local.

Referenced by mlir::presburger::PresburgerRelation::containsPoint(), and containsPointNoLocal().

◆ containsPointNoLocal() [2/2]

std::optional<SmallVector<DynamicAPInt, 8> > mlir::presburger::IntegerRelation::containsPointNoLocal ( ArrayRef< int64_t >  point) const
inline

◆ convertToLocal()

void mlir::presburger::IntegerRelation::convertToLocal ( VarKind  kind,
unsigned  varStart,
unsigned  varLimit 
)
inline

◆ convertVarKind() [1/2]

void mlir::presburger::IntegerRelation::convertVarKind ( VarKind  srcKind,
unsigned  varStart,
unsigned  varLimit,
VarKind  dstKind 
)
inline

Definition at line 604 of file IntegerRelation.h.

References convertVarKind(), and getNumVarKind().

◆ convertVarKind() [2/2]

void IntegerRelation::convertVarKind ( VarKind  srcKind,
unsigned  varStart,
unsigned  varLimit,
VarKind  dstKind,
unsigned  pos 
)

Converts variables of kind srcKind in the range [varStart, varLimit) to variables of kind dstKind.

If pos is given, the variables are placed at position pos of dstKind, otherwise they are placed after all the other variables of kind dstKind. The internal ordering among the moved variables is preserved.

Definition at line 1462 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerSpace::convertVarKind(), equalities, getNumVarKind(), getVarKindOffset(), inequalities, mlir::presburger::Matrix< T >::moveColumns(), and space.

Referenced by mlir::affine::checkMemrefAccessDependence(), compose(), convertToLocal(), convertVarKind(), mlir::affine::MemRefAccess::getAccessRelation(), getDomainSet(), getRangeSet(), and inverse().

◆ dump()

void IntegerRelation::dump ( ) const

◆ eliminateRedundantLocalVar()

void IntegerRelation::eliminateRedundantLocalVar ( unsigned  posA,
unsigned  posB 
)
virtual

Eliminate the posB^th local variable, replacing every instance of it with the posA^th local variable.

This should be used when the two local variables are known to always take the same values.

Definition at line 1297 of file IntegerRelation.cpp.

References mlir::presburger::Matrix< T >::addToColumn(), equalities, getNumLocalVars(), getVarKindOffset(), inequalities, mlir::presburger::Local, and removeVar().

Referenced by mergeLocalVars(), and removeDuplicateDivs().

◆ findConstraintWithNonZeroAt()

bool IntegerRelation::findConstraintWithNonZeroAt ( unsigned  colIdx,
bool  isEq,
unsigned *  rowIdx 
) const
protected

Searches for a constraint with a non-zero coefficient at colIdx in equality (isEq=true) or inequality (isEq=false) constraints.

Returns true and sets row found in search in rowIdx, false otherwise.

Definition at line 570 of file IntegerRelation.cpp.

References atEq(), atIneq(), getNumCols(), getNumEqualities(), and getNumInequalities().

Referenced by gaussianEliminate(), gaussianEliminateVars(), and isColZero().

◆ findIntegerLexMin()

MaybeOptimum< SmallVector< DynamicAPInt, 8 > > IntegerRelation::findIntegerLexMin ( ) const

Same as above, but returns lexicographically minimal integer point.

Note: this should be used only when the lexmin is really required. For a generic integer sampling operation, findIntegerSample is more robust and should be preferred. Note that Domain is minimized first, then range.

Definition at line 162 of file IntegerRelation.cpp.

References mlir::presburger::LexSimplex::findIntegerLexMin(), mlir::presburger::MaybeOptimum< T >::getKind(), getNumDimAndSymbolVars(), getNumSymbolVars(), getNumVars(), and mlir::presburger::MaybeOptimum< T >::isBounded().

◆ findIntegerSample()

std::optional< SmallVector< DynamicAPInt, 8 > > IntegerRelation::findIntegerSample ( ) const

Find an integer sample point satisfying the constraints using a branch and bound algorithm with generalized basis reduction, with some additional processing using Simplex for unbounded sets.

Let this set be S.

Returns an integer sample point if one exists, or an empty Optional otherwise. The returned value also includes values of local ids.

If S is bounded then we directly call into the GBR sampling algorithm. Otherwise, there are some unbounded directions, i.e., vectors v such that S extends to infinity along v or -v. In this case we use an algorithm described in the integer set library (isl) manual and used by the isl_set_sample function in that library. The algorithm is:

1) Apply a unimodular transform T to S to obtain S*T, such that all dimensions in which S*T is bounded lie in the linear span of a prefix of the dimensions.

2) Construct a set B by removing all constraints that involve the unbounded dimensions and then deleting the unbounded dimensions. Note that B is a Bounded set.

3) Try to obtain a sample from B using the GBR sampling algorithm. If no sample is found, return that S is empty.

4) Otherwise, substitute the obtained sample into S*T to obtain a set C. C is a full-dimensional Cone and always contains a sample.

5) Obtain an integer sample from C.

6) Return T*v, where v is the concatenation of the samples from B and C.

The following is a sketch of a proof that a) If the algorithm returns empty, then S is empty. b) If the algorithm returns a sample, it is a valid sample in S.

The algorithm returns empty only if B is empty, in which case S*T is certainly empty since B was obtained by removing constraints and then deleting unconstrained dimensions from S*T. Since T is unimodular, a vector v is in S*T iff T*v is in S. So in this case, since S*T is empty, S is empty too.

Otherwise, the algorithm substitutes the sample from B into S*T. All the constraints of S*T that did not involve unbounded dimensions are satisfied by this substitution. All dimensions in the linear span of the dimensions outside the prefix are unbounded in S*T (step 1). Substituting values for the bounded dimensions cannot make these dimensions bounded, and these are the only remaining dimensions in C, so C is unbounded along every vector (in the positive or negative direction, or both). C is hence a full-dimensional cone and therefore always contains an integer point.

Concatenating the samples from B and C gives a sample v in S*T, so the returned sample T*v is a sample in S.

Definition at line 871 of file IntegerRelation.cpp.

References mlir::presburger::Simplex::findIntegerSample(), mlir::presburger::SimplexBase::isEmpty(), isEmptyByGCDTest(), and mlir::presburger::Simplex::isUnbounded().

Referenced by isIntegerEmpty().

◆ findRationalLexMin()

MaybeOptimum< SmallVector< Fraction, 8 > > IntegerRelation::findRationalLexMin ( ) const

Get the lexicographically minimum rational point satisfying the constraints.

Returns an empty optional if the relation is empty or if the lexmin is unbounded. Symbols are not supported and will result in assert-failure. Note that Domain is minimized first, then range.

Definition at line 142 of file IntegerRelation.cpp.

References mlir::presburger::LexSimplex::findRationalLexMin(), getNumDimAndSymbolVars(), getNumSymbolVars(), getNumVars(), and mlir::presburger::MaybeOptimum< T >::isBounded().

◆ findSymbolicIntegerLexMax()

SymbolicLexOpt IntegerRelation::findSymbolicIntegerLexMax ( ) const

Same as findSymbolicIntegerLexMin but produces lexmax instead of lexmin.

findSymbolicIntegerLexMax is implemented using findSymbolicIntegerLexMin as follows:

  1. A new relation is created which is this relation with the sign of each dimension variable in range flipped;
  2. findSymbolicIntegerLexMin is called on the range negated relation to compute the negated lexmax of this relation;
  3. The sign of the negated lexmax is flipped and returned.

Definition at line 318 of file IntegerRelation.cpp.

References atEq(), atIneq(), mlir::presburger::PWMAFunction::Piece::domain, findSymbolicIntegerLexMin(), mlir::presburger::PWMAFunction::getAllPieces(), getNumDomainVars(), getNumEqualities(), getNumInequalities(), getNumRangeVars(), mlir::presburger::Matrix< T >::getNumRows(), mlir::presburger::PWMAFunction::getSpace(), mlir::presburger::SymbolicLexOpt::lexopt, mlir::presburger::Matrix< T >::negateRow(), and mlir::presburger::SymbolicLexOpt::unboundedDomain.

◆ findSymbolicIntegerLexMin()

SymbolicLexOpt IntegerRelation::findSymbolicIntegerLexMin ( ) const

Compute the symbolic integer lexmin of the relation.

This finds, for every assignment to the symbols and domain, the lexicographically minimum value attained by the range.

For example, the symbolic lexmin of the set

(x, y)[a, b, c] : (a <= x, b <= x, x <= c)

can be written as

x = a if b <= a, a <= c x = b if a < b, b <= c

This function is stored in the lexopt function in the result. Some assignments to the symbols might make the set empty. Such points are not part of the function's domain. In the above example, this happens when max(a, b) > c.

For some values of the symbols, the lexmin may be unbounded. SymbolicLexOpt stores these parts of the symbolic domain in a separate PresburgerSet, unboundedDomain.

Definition at line 284 of file IntegerRelation.cpp.

References mlir::presburger::SymbolicLexSimplex::computeSymbolicIntegerLexMin(), mlir::presburger::Domain, getNumDomainVars(), getNumLocalVars(), mlir::presburger::PWMAFunction::getNumOutputs(), getNumSymbolVars(), getNumVars(), mlir::presburger::PresburgerSpace::getSetSpace(), getVarKindEnd(), getVarKindOffset(), mlir::presburger::SymbolicLexOpt::lexopt, mlir::presburger::PWMAFunction::removeOutputs(), and mlir::presburger::Symbol.

Referenced by findSymbolicIntegerLexMax().

◆ fourierMotzkinEliminate()

void IntegerRelation::fourierMotzkinEliminate ( unsigned  pos,
bool  darkShadow = false,
bool *  isResultIntegerExact = nullptr 
)
protectedvirtual

Eliminates the variable at the specified position using Fourier-Motzkin variable elimination, but uses Gaussian elimination if there is an equality involving that variable.

If the result of the elimination is integer exact, *isResultIntegerExact is set to true. If darkShadow is set to true, a potential under approximation (subset) of the rational shadow / exact integer shadow is computed.

Create the new system which has one variable less.

Definition at line 1940 of file IntegerRelation.cpp.

References dump(), getNumVars(), and hasConsistentState().

Referenced by isEmpty(), projectOut(), and mlir::FlatLinearValueConstraints::projectOut().

◆ gaussianEliminate()

bool IntegerRelation::gaussianEliminate ( )
protected

Perform a Gaussian elimination operation to reduce all equations to standard form.

Returns whether the constraint system was modified.

Definition at line 1125 of file IntegerRelation.cpp.

References atEq(), eliminateFromConstraint(), equalities, findConstraintWithNonZeroAt(), gcdTightenInequalities(), getEmpty(), getNumEqualities(), getNumInequalities(), getNumVars(), getSpace(), inequalities, mlir::presburger::IntMatrix::normalizeRow(), removeEqualityRange(), and mlir::presburger::Matrix< T >::swapRows().

Referenced by simplify().

◆ gaussianEliminateVar()

LogicalResult mlir::presburger::IntegerRelation::gaussianEliminateVar ( unsigned  position)
inlineprotected

Eliminates a single variable at position from equality and inequality constraints.

Returns success if the variable was eliminated, and failure otherwise.

Definition at line 781 of file IntegerRelation.h.

References gaussianEliminateVars().

◆ gaussianEliminateVars()

unsigned IntegerRelation::gaussianEliminateVars ( unsigned  posStart,
unsigned  posLimit 
)
protected

Eliminates variables from equality and inequality constraints in column range [posStart, posLimit).

Returns the number of variables eliminated.

Definition at line 1077 of file IntegerRelation.cpp.

References eliminateFromConstraint(), equalities, findConstraintWithNonZeroAt(), gcdTightenInequalities(), getNumEqualities(), getNumInequalities(), getNumVars(), hasConsistentState(), inequalities, mlir::presburger::IntMatrix::normalizeRow(), removeEquality(), and removeVarRange().

Referenced by gaussianEliminateVar(), isEmpty(), and projectOut().

◆ gcdTightenInequalities()

void IntegerRelation::gcdTightenInequalities ( )
protected

Tightens inequalities given that we are dealing with integer spaces.

This is similar to the GCD test but applied to inequalities. The constant term can be reduced to the preceding multiple of the GCD of the coefficients, i.e., 64*i - 100 >= 0 => 64*i - 128 >= 0 (since 'i' is an integer). This is a fast method (linear in the number of coefficients).

This is analogous to the GCD test but applied to inequalities. The constant term can be reduced to the preceding multiple of the GCD of the coefficients, i.e., 64*i - 100 >= 0 => 64*i - 128 >= 0 (since 'i' is an integer). This is a fast method - linear in the number of coefficients.

Definition at line 1065 of file IntegerRelation.cpp.

References atIneq(), getNumCols(), getNumInequalities(), inequalities, and mlir::presburger::IntMatrix::normalizeRow().

Referenced by gaussianEliminate(), gaussianEliminateVars(), projectOut(), removeRedundantConstraints(), and removeTrivialRedundancy().

◆ getBoundedDirections()

IntMatrix IntegerRelation::getBoundedDirections ( ) const

Returns a matrix where each row is a vector along which the polytope is bounded.

The span of the returned vectors is guaranteed to contain all such vectors. The returned vectors are NOT guaranteed to be linearly independent. This function should not be called on empty sets.

Definition at line 782 of file IntegerRelation.cpp.

References atEq(), atIneq(), getNumCols(), getNumEqualities(), getNumInequalities(), mlir::presburger::Simplex::isBoundedAlongConstraint(), and mlir::presburger::SimplexBase::isEmpty().

◆ getConstantBound()

std::optional< DynamicAPInt > IntegerRelation::getConstantBound ( BoundType  type,
unsigned  pos 
) const

Returns the constant bound for the pos^th variable if there is one; std::nullopt otherwise.

Definition at line 1778 of file IntegerRelation.cpp.

References mlir::presburger::EQ, IntegerRelation(), mlir::presburger::LB, and mlir::presburger::UB.

Referenced by getConstantBound64(), and unionBoundingBox().

◆ getConstantBound64()

std::optional<int64_t> mlir::presburger::IntegerRelation::getConstantBound64 ( BoundType  type,
unsigned  pos 
) const
inline

The same, but casts to int64_t.

This is unsafe and will assert-fail if the value does not fit in an int64_t.

Definition at line 558 of file IntegerRelation.h.

References getConstantBound().

Referenced by mlir::ValueBoundsConstraintSet::computeConstantBound(), computeDirectionVector(), mlir::affine::normalizeMemRefType(), and mlir::affine::simplifyConstrainedMinMaxOp().

◆ getConstantBoundOnDimSize()

std::optional< DynamicAPInt > IntegerRelation::getConstantBoundOnDimSize ( unsigned  pos,
SmallVectorImpl< DynamicAPInt > *  lb = nullptr,
DynamicAPInt *  boundFloorDivisor = nullptr,
SmallVectorImpl< DynamicAPInt > *  ub = nullptr,
unsigned *  minLbPos = nullptr,
unsigned *  minUbPos = nullptr 
) const

Returns the smallest known constant bound for the extent of the specified variable (pos^th), i.e., the smallest known constant that is greater than or equal to 'exclusive upper bound' - 'lower bound' of the variable.

Returns a non-negative constant bound on the extent (upper bound - lower bound) of the specified variable if it is found to be a constant; returns std::nullopt if it's not a constant.

This constant bound is guaranteed to be non-negative. Returns std::nullopt if it's not a constant. This method employs trivial (low complexity / cost) checks and detection. Symbolic variables are treated specially, i.e., it looks for constant differences between affine expressions involving only the symbolic variables. lb and ub (along with the boundFloorDivisor) are set to represent the lower and upper bound associated with the constant difference: lb, ub have the coefficients, and boundFloorDivisor, their divisor. minLbPos and minUbPos if non-null are set to the position of the constant lower bound and upper bound respectively (to the same if they are from an equality). Ex: if the lower bound is [(s0 + s2 - 1) floordiv 32] for a system with three symbolic variables, *lb = [1, 0, 1], lbDivisor = 32. See comments at function definition for examples.

This methods treats symbolic variables specially, i.e., it looks for constant differences between affine expressions involving only the symbolic variables. See comments at function definition for example. 'lb', if provided, is set to the lower bound associated with the constant difference. Note that 'lb' is purely symbolic and thus will contain the coefficients of the symbolic variables and the constant coefficient.

Definition at line 1593 of file IntegerRelation.cpp.

References atEq(), findEqualityToConstant(), getEquality(), getNumDimAndSymbolVars(), getNumDimVars(), and getNumSymbolVars().

Referenced by createFullTiles(), createSeparationCondition(), getConstantBoundOnDimSize64(), and unionBoundingBox().

◆ getConstantBoundOnDimSize64()

std::optional<int64_t> mlir::presburger::IntegerRelation::getConstantBoundOnDimSize64 ( unsigned  pos,
SmallVectorImpl< int64_t > *  lb = nullptr,
int64_t *  boundFloorDivisor = nullptr,
SmallVectorImpl< int64_t > *  ub = nullptr,
unsigned *  minLbPos = nullptr,
unsigned *  minUbPos = nullptr 
) const
inline

The same, but casts to int64_t.

This is unsafe and will assert-fail if the value does not fit in an int64_t.

Definition at line 533 of file IntegerRelation.h.

References getConstantBoundOnDimSize(), and mlir::presburger::getInt64Vec().

Referenced by mlir::affine::MemRefRegion::getConstantBoundOnDimSize().

◆ getCounts()

IntegerRelation::CountsSnapshot IntegerRelation::getCounts ( ) const

◆ getDomainSet()

IntegerPolyhedron IntegerRelation::getDomainSet ( ) const

Return a set corresponding to all points in the domain of the relation.

Definition at line 2346 of file IntegerRelation.cpp.

References convertVarKind(), mlir::presburger::Domain, getNumVarKind(), mlir::presburger::Local, mlir::presburger::Range, and mlir::presburger::SetDim.

Referenced by mlir::affine::checkMemrefAccessDependence(), compose(), and intersectDomain().

◆ getEmpty()

static IntegerRelation mlir::presburger::IntegerRelation::getEmpty ( const PresburgerSpace space)
inlinestatic

Return an empty system containing an invalid equation 0 = 1.

Definition at line 103 of file IntegerRelation.h.

References addEquality(), mlir::presburger::PresburgerSpace::getNumVars(), and space.

Referenced by gaussianEliminate(), and removeDuplicateConstraints().

◆ getEquality()

ArrayRef<DynamicAPInt> mlir::presburger::IntegerRelation::getEquality ( unsigned  idx) const
inline

◆ getEquality64()

SmallVector<int64_t, 8> mlir::presburger::IntegerRelation::getEquality64 ( unsigned  idx) const
inline

The same, but casts to int64_t.

This is unsafe and will assert-fail if the value does not fit in an int64_t.

Definition at line 240 of file IntegerRelation.h.

References equalities, mlir::presburger::getInt64Vec(), and mlir::presburger::Matrix< T >::getRow().

◆ getIds()

ArrayRef< Identifier > IntegerRelation::getIds ( VarKind  kind)

Get the identifiers for the variables of specified varKind.

Calls resetIds on the relations space if identifiers are not enabled.

Definition at line 79 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerSpace::getIds(), mlir::presburger::PresburgerSpace::isUsingIds(), mlir::presburger::PresburgerSpace::resetIds(), and space.

Referenced by mlir::affine::MemRefAccess::getAccessRelation().

◆ getInequalities()

IntMatrix mlir::presburger::IntegerRelation::getInequalities ( ) const
inline

◆ getInequality()

ArrayRef<DynamicAPInt> mlir::presburger::IntegerRelation::getInequality ( unsigned  idx) const
inline

◆ getInequality64()

SmallVector<int64_t, 8> mlir::presburger::IntegerRelation::getInequality64 ( unsigned  idx) const
inline

◆ getKind()

virtual Kind mlir::presburger::IntegerRelation::getKind ( ) const
inlinevirtual

◆ getLocalReprs()

DivisionRepr IntegerRelation::getLocalReprs ( std::vector< MaybeLocalRepr > *  repr = nullptr) const

Returns a DivisonRepr representing the division representation of local variables in the constraint system.

If repr is not nullptr, the equality and pairs of inequality constraints identified by their position indices using which an explicit representation for each local variable can be computed are set in repr in the form of a MaybeLocalRepr struct. If no such inequality pair/equality can be found, the kind attribute in MaybeLocalRepr is set to None.

Definition at line 1023 of file IntegerRelation.cpp.

References mlir::presburger::DivisionRepr::clearRepr(), mlir::presburger::computeSingleVarRepr(), mlir::presburger::DivisionRepr::getDenom(), mlir::presburger::DivisionRepr::getDividend(), getNumDimAndSymbolVars(), getNumLocalVars(), getNumVars(), getVarKindOffset(), and mlir::presburger::Local.

Referenced by mlir::getMultiAffineFunctionFromMap(), getSetDifference(), hasOnlyDivLocals(), mlir::presburger::mergeLocalVars(), and removeDuplicateDivs().

◆ getLowerAndUpperBoundIndices()

void IntegerRelation::getLowerAndUpperBoundIndices ( unsigned  pos,
SmallVectorImpl< unsigned > *  lbIndices,
SmallVectorImpl< unsigned > *  ubIndices,
SmallVectorImpl< unsigned > *  eqIndices = nullptr,
unsigned  offset = 0,
unsigned  num = 0 
) const

Gather positions of all lower and upper bounds of the variable at pos, and optionally any equalities on it.

Gather all lower and upper bounds of the variable at pos, and optionally any equalities on it.

In addition, the bounds are to be independent of variables in position range [offset, offset + num).

Definition at line 487 of file IntegerRelation.cpp.

References getNumCols(), and getNumVars().

Referenced by mlir::presburger::computeSingleVarRepr(), and createSeparationCondition().

◆ getNumCols()

unsigned mlir::presburger::IntegerRelation::getNumCols ( ) const
inline

◆ getNumConstraints()

unsigned mlir::presburger::IntegerRelation::getNumConstraints ( ) const
inline

Definition at line 200 of file IntegerRelation.h.

References getNumEqualities(), and getNumInequalities().

Referenced by isEmpty(), and printSpace().

◆ getNumDimAndSymbolVars()

unsigned mlir::presburger::IntegerRelation::getNumDimAndSymbolVars ( ) const
inline

◆ getNumDimVars()

unsigned mlir::presburger::IntegerRelation::getNumDimVars ( ) const
inline

◆ getNumDomainVars()

unsigned mlir::presburger::IntegerRelation::getNumDomainVars ( ) const
inline

◆ getNumEqualities()

unsigned mlir::presburger::IntegerRelation::getNumEqualities ( ) const
inline

◆ getNumInequalities()

unsigned mlir::presburger::IntegerRelation::getNumInequalities ( ) const
inline

◆ getNumLocalVars()

unsigned mlir::presburger::IntegerRelation::getNumLocalVars ( ) const
inline

◆ getNumRangeVars()

unsigned mlir::presburger::IntegerRelation::getNumRangeVars ( ) const
inline

◆ getNumReservedEqualities()

unsigned mlir::presburger::IntegerRelation::getNumReservedEqualities ( ) const
inline

◆ getNumReservedInequalities()

unsigned mlir::presburger::IntegerRelation::getNumReservedInequalities ( ) const
inline

◆ getNumSymbolVars()

unsigned mlir::presburger::IntegerRelation::getNumSymbolVars ( ) const
inline

◆ getNumVarKind()

unsigned mlir::presburger::IntegerRelation::getNumVarKind ( VarKind  kind) const
inline

◆ getNumVars()

unsigned mlir::presburger::IntegerRelation::getNumVars ( ) const
inline

◆ getRangeSet()

IntegerPolyhedron IntegerRelation::getRangeSet ( ) const

Return a set corresponding to all points in the range of the relation.

Definition at line 2420 of file IntegerRelation.cpp.

References convertVarKind(), mlir::presburger::Domain, getNumVarKind(), and mlir::presburger::Local.

Referenced by compose(), and intersectRange().

◆ getSpace()

const PresburgerSpace& mlir::presburger::IntegerRelation::getSpace ( ) const
inline

◆ getSpaceWithoutLocals()

PresburgerSpace mlir::presburger::IntegerRelation::getSpaceWithoutLocals ( ) const
inline

◆ getUniverse()

static IntegerRelation mlir::presburger::IntegerRelation::getUniverse ( const PresburgerSpace space)
inlinestatic

Return a system with no constraints, i.e., one which is satisfied by all points.

Definition at line 98 of file IntegerRelation.h.

References IntegerRelation(), and space.

Referenced by mlir::presburger::PresburgerRelation::complement(), and mlir::presburger::PresburgerRelation::getUniverse().

◆ getVarKindAt()

VarKind mlir::presburger::IntegerRelation::getVarKindAt ( unsigned  pos) const
inline

◆ getVarKindEnd()

unsigned mlir::presburger::IntegerRelation::getVarKindEnd ( VarKind  kind) const
inline

◆ getVarKindOffset()

unsigned mlir::presburger::IntegerRelation::getVarKindOffset ( VarKind  kind) const
inline

◆ getVarKindOverlap()

unsigned mlir::presburger::IntegerRelation::getVarKindOverlap ( VarKind  kind,
unsigned  varStart,
unsigned  varLimit 
) const
inline

Get the number of elements of the specified kind in the range [varStart, varLimit).

Definition at line 266 of file IntegerRelation.h.

References mlir::presburger::PresburgerSpace::getVarKindOverlap(), and space.

◆ hasConsistentState()

bool IntegerRelation::hasConsistentState ( ) const
protectedvirtual

Returns false if the fields corresponding to various variable counts, or equality/inequality buffer sizes aren't consistent; true otherwise.

This is meant to be used within an assert internally.

Definition at line 538 of file IntegerRelation.cpp.

References equalities, mlir::presburger::Matrix< T >::hasConsistentState(), and inequalities.

Referenced by fourierMotzkinEliminate(), gaussianEliminateVars(), hasInvalidConstraint(), isEmptyByGCDTest(), and print().

◆ hasInvalidConstraint()

bool IntegerRelation::hasInvalidConstraint ( ) const
protected

Checks all rows of equality/inequality constraints for trivial contradictions (for example: 1 == 0, 0 >= 1), which may have surfaced after elimination.

Returns true if an invalid constraint is found; false otherwise.

Definition at line 592 of file IntegerRelation.cpp.

References atEq(), atIneq(), getNumCols(), getNumEqualities(), getNumInequalities(), and hasConsistentState().

Referenced by isEmpty(), and isObviouslyEmpty().

◆ hasOnlyDivLocals()

bool IntegerRelation::hasOnlyDivLocals ( ) const

Check whether all local ids have a division representation.

Definition at line 1387 of file IntegerRelation.cpp.

References getLocalReprs(), and mlir::presburger::DivisionRepr::hasAllReprs().

Referenced by mlir::presburger::PresburgerRelation::hasOnlyDivLocals().

◆ insertVar()

unsigned IntegerRelation::insertVar ( VarKind  kind,
unsigned  pos,
unsigned  num = 1 
)
virtual

Insert num variables of the specified kind at position pos.

Positions are relative to the kind of variable. The coefficient columns corresponding to the added variables are initialized to zero. Return the absolute column position (i.e., not relative to the kind of variable) of the first added variable.

Reimplemented in mlir::presburger::IntegerPolyhedron, and mlir::FlatLinearValueConstraints.

Definition at line 354 of file IntegerRelation.cpp.

References equalities, getNumVarKind(), inequalities, mlir::presburger::Matrix< T >::insertColumns(), mlir::presburger::PresburgerSpace::insertVar(), and space.

Referenced by appendVar(), mlir::affine::MemRefAccess::getAccessRelation(), mlir::presburger::MultiAffineFunction::getAsRelation(), mlir::presburger::IntegerPolyhedron::insertVar(), mergeAndAlignSymbols(), mergeAndCompose(), and mlir::presburger::mergeLocalVars().

◆ intersect()

IntegerRelation IntegerRelation::intersect ( IntegerRelation  other) const

Return the intersection of the two relations.

If there are locals, they will be merged.

Definition at line 100 of file IntegerRelation.cpp.

References append(), and mergeLocalVars().

Referenced by mlir::presburger::IntegerPolyhedron::intersect(), and mlir::presburger::PresburgerRelation::intersect().

◆ intersectDomain()

void IntegerRelation::intersectDomain ( const IntegerPolyhedron poly)

Intersect the given poly with the domain in-place.

Formally, let the relation this be R: A -> B and poly is C, then this operation modifies R to be (A intersection C) -> B.

Definition at line 2433 of file IntegerRelation.cpp.

References append(), appendVar(), getDomainSet(), getNumRangeVars(), getSpace(), inverse(), mergeLocalVars(), and mlir::presburger::Range.

Referenced by mlir::presburger::MultiAffineFunction::isEqual().

◆ intersectRange()

void IntegerRelation::intersectRange ( const IntegerPolyhedron poly)

Intersect the given poly with the range in-place.

Formally, let the relation this be R: A -> B and poly is C, then this operation modifies R to be A -> (B intersection C).

Definition at line 2449 of file IntegerRelation.cpp.

References append(), appendVar(), mlir::presburger::Domain, getNumDomainVars(), getRangeSet(), getSpace(), and mergeLocalVars().

Referenced by compose().

◆ inverse()

void IntegerRelation::inverse ( )

Invert the relation i.e., swap its domain and range.

Formally, let the relation this be R: A -> B, then this operation modifies R to be B -> A.

Definition at line 2462 of file IntegerRelation.cpp.

References convertVarKind(), mlir::presburger::Domain, getNumVarKind(), getVarKindEnd(), and mlir::presburger::Range.

Referenced by applyDomain(), mlir::affine::checkMemrefAccessDependence(), and intersectDomain().

◆ isColZero()

bool IntegerRelation::isColZero ( unsigned  pos) const
protected

Returns true if the pos^th column is all zero for both inequalities and equalities.

Definition at line 2292 of file IntegerRelation.cpp.

References findConstraintWithNonZeroAt().

◆ isEmpty()

bool IntegerRelation::isEmpty ( ) const

Checks for emptiness by performing variable elimination on all variables, running the GCD test on each equality constraint, and checking for invalid constraints.

Returns true if the GCD test fails for any equality, or if any invalid constraints are discovered on any row. Returns false otherwise.

Definition at line 695 of file IntegerRelation.cpp.

References fourierMotzkinEliminate(), gaussianEliminateVars(), getBestVarToEliminate(), getNumConstraints(), getNumVars(), hasInvalidConstraint(), isEmptyByGCDTest(), kExplosionFactor, and removeRedundantLocalVars().

Referenced by mlir::affine::checkMemrefAccessDependence(), mlir::ValueBoundsConstraintSet::comparePos(), mlir::presburger::PresburgerRelation::compose(), mlir::presburger::PresburgerRelation::intersect(), isFullDim(), mlir::affine::simplifyConstrainedMinMaxOp(), and mlir::affine::simplifyIntegerSet().

◆ isEmptyByGCDTest()

bool IntegerRelation::isEmptyByGCDTest ( ) const

Runs the GCD test on all equality constraints.

Returns true if this test fails on any equality. Returns false otherwise. This test can be used to disprove the existence of a solution. If it returns true, no integer solution to the equality constraints can exist.

Definition at line 759 of file IntegerRelation.cpp.

References mlir::presburger::abs(), atEq(), getNumCols(), getNumEqualities(), and hasConsistentState().

Referenced by findIntegerSample(), getSetDifference(), isEmpty(), and isObviouslyEmpty().

◆ isEqual()

bool IntegerRelation::isEqual ( const IntegerRelation other) const

Return whether this and other are equal.

This is integer-exact and somewhat expensive, since it uses the integer emptiness check (see IntegerRelation::findIntegerSample()).

Definition at line 107 of file IntegerRelation.cpp.

References getSpace(), mlir::presburger::PresburgerSpace::isCompatible(), mlir::presburger::PresburgerRelation::isEqual(), and space.

Referenced by mlir::presburger::MultiAffineFunction::isEqual().

◆ isFullDim()

bool IntegerRelation::isFullDim ( )

◆ isHyperRectangular()

bool IntegerRelation::isHyperRectangular ( unsigned  pos,
unsigned  num 
) const

Returns true if the set can be trivially detected as being hyper-rectangular on the specified contiguous set of variables.

Definition at line 1798 of file IntegerRelation.cpp.

References getNumCols().

Referenced by checkIfHyperRectangular().

◆ isIntegerEmpty()

bool IntegerRelation::isIntegerEmpty ( ) const

Returns true if the set of constraints is found to have no solution, false if a solution exists.

Uses the same algorithm as findIntegerSample.

Definition at line 823 of file IntegerRelation.cpp.

References findIntegerSample().

Referenced by mlir::presburger::PresburgerRelation::isIntegerEmpty().

◆ isObviouslyEmpty()

bool IntegerRelation::isObviouslyEmpty ( ) const

Performs GCD checks and invalid constraint checks.

Definition at line 741 of file IntegerRelation.cpp.

References hasInvalidConstraint(), and isEmptyByGCDTest().

Referenced by simplify(), and mlir::presburger::PresburgerRelation::simplify().

◆ isObviouslyEqual()

bool IntegerRelation::isObviouslyEqual ( const IntegerRelation other) const

Perform a quick equality check on this and other.

The relations are equal if the check return true, but may or may not be equal if the check returns false. The equality check is performed in a plain manner, by comparing if all the equalities and inequalities in this and other are the same.

Definition at line 112 of file IntegerRelation.cpp.

References atEq(), atIneq(), cols, getNumCols(), getNumEqualities(), getNumInequalities(), getSpace(), mlir::presburger::PresburgerSpace::isEqual(), and space.

◆ isSubsetOf()

bool IntegerRelation::isSubsetOf ( const IntegerRelation other) const

Return whether this is a subset of the given IntegerRelation.

This is integer-exact and somewhat expensive, since it uses the integer emptiness check (see IntegerRelation::findIntegerSample()).

Definition at line 136 of file IntegerRelation.cpp.

References getSpace(), mlir::presburger::PresburgerSpace::isCompatible(), mlir::presburger::PresburgerRelation::isSubsetOf(), and space.

◆ mergeAndAlignSymbols()

void IntegerRelation::mergeAndAlignSymbols ( IntegerRelation other)

Merge and align symbol variables of this and other with respect to identifiers.

mergeAndAlignSymbols's implementation can be broken down into two steps:

After this operation the symbol variables of both relations have the same identifiers in the same order.

  1. Merge and align identifiers into other from this. If an identifier fromthisexists inotherthen we align it. Otherwise, we assume it is a new identifier and insert it intootherin the same position asthis.
  2. Add identifiers that are inotherbut notthis to this.

Definition at line 1314 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerSpace::getId(), mlir::presburger::PresburgerSpace::getIds(), getNumVarKind(), getVarKindOffset(), insertVar(), mlir::presburger::PresburgerSpace::isUsingIds(), mlir::presburger::PresburgerSpace::setId(), space, swapVar(), and mlir::presburger::Symbol.

Referenced by mlir::affine::MemRefAccess::getAccessRelation(), and mergeAndCompose().

◆ mergeAndCompose()

void IntegerRelation::mergeAndCompose ( const IntegerRelation other)

Given a relation other: (A -> B), this operation merges the symbol and local variables and then takes the composition of other on this: (B -> C).

The resulting relation represents tuples of the form: A -> C.

Definition at line 2533 of file IntegerRelation.cpp.

References append(), convertToLocal(), mlir::presburger::Domain, mlir::presburger::PresburgerSpace::getId(), getNumDomainVars(), getNumRangeVars(), getSpace(), mlir::presburger::Identifier::hasValue(), insertVar(), mergeAndAlignSymbols(), mergeLocalVars(), mlir::presburger::Range, removeRedundantLocalVars(), removeVarRange(), mlir::presburger::PresburgerSpace::setId(), and space.

Referenced by mlir::affine::checkMemrefAccessDependence().

◆ mergeLocalVars()

unsigned IntegerRelation::mergeLocalVars ( IntegerRelation other)

Adds additional local vars to the sets such that they both have the union of the local vars in each set, without changing the set of points that lie in this and other.

Adds additional local ids to the sets such that they both have the union of the local ids in each set, without changing the set of points that lie in this and other.

While taking union, if a local var in other has a division representation which is a duplicate of division representation, of another local var, it is not added to the final union of local vars and is instead merged. The new ordering of local vars is:

[Local vars of this] [Non-merged local vars of other]

The relative ordering of local vars is same as before.

After merging, if the i^th local variable in one set has a known division representation, then the i^th local variable in the other set either has the same division representation or no known division representation.

The spaces of both relations should be compatible.

Returns the number of non-merged local vars of other, i.e. the number of locals that have been added to this.

To detect local ids that always take the same value, each local id is represented as a floordiv with constant denominator in terms of other ids. After extracting these divisions, local ids in other with the same division representation as some other local id in any set are considered duplicate and are merged.

It is possible that division representation for some local id cannot be obtained, and thus these local ids are not considered for detecting duplicates.

Definition at line 1356 of file IntegerRelation.cpp.

References eliminateRedundantLocalVar(), getNumLocalVars(), and mlir::presburger::mergeLocalVars().

Referenced by mlir::affine::MemRefAccess::getAccessRelation(), getSetDifference(), intersect(), intersectDomain(), intersectRange(), mergeAndAlignVars(), and mergeAndCompose().

◆ normalizeConstraintsByGCD()

void IntegerRelation::normalizeConstraintsByGCD ( )
protected

Normalized each constraints by the GCD of its coefficients.

Definition at line 585 of file IntegerRelation.cpp.

References equalities, getNumEqualities(), getNumInequalities(), inequalities, and mlir::presburger::IntMatrix::normalizeRow().

Referenced by projectOut(), removeTrivialRedundancy(), and simplify().

◆ print()

void IntegerRelation::print ( raw_ostream &  os) const

◆ printSpace()

void IntegerRelation::printSpace ( raw_ostream &  os) const
protectedvirtual

Prints the number of constraints, dimensions, symbols and locals in the IntegerRelation.

Reimplemented in mlir::FlatLinearValueConstraints, and mlir::FlatLinearConstraints.

Definition at line 2503 of file IntegerRelation.cpp.

References getNumConstraints(), mlir::presburger::PresburgerSpace::print(), and space.

Referenced by print().

◆ projectOut() [1/2]

void mlir::presburger::IntegerRelation::projectOut ( unsigned  pos)
inline

Definition at line 479 of file IntegerRelation.h.

References projectOut().

Referenced by projectOut().

◆ projectOut() [2/2]

void IntegerRelation::projectOut ( unsigned  pos,
unsigned  num 
)

Projects out (aka eliminates) num variables starting at position pos.

The resulting constraint system is the shadow along the dimensions that still exist. This method may not always be integer exact.

Definition at line 2096 of file IntegerRelation.cpp.

References fourierMotzkinEliminate(), gaussianEliminateVars(), gcdTightenInequalities(), getBestVarToEliminate(), getNumCols(), getNumVars(), and normalizeConstraintsByGCD().

Referenced by mlir::ValueBoundsConstraintSet::computeBound(), computeConstantLowerOrUpperBound(), computeDirectionVector(), and mlir::ValueBoundsConstraintSet::projectOut().

◆ removeDuplicateConstraints()

bool IntegerRelation::removeDuplicateConstraints ( )
protected

Checks for identical inequalities and eliminates redundant inequalities.

Returns whether the constraint system was modified.

Definition at line 2360 of file IntegerRelation.cpp.

References addEquality(), atIneq(), cols, getEmpty(), getInequality(), getNumCols(), getNumInequalities(), getSpace(), inequalities, removeInequality(), and mlir::presburger::Matrix< T >::swapRows().

Referenced by simplify().

◆ removeDuplicateDivs()

void IntegerRelation::removeDuplicateDivs ( )

◆ removeEquality()

void IntegerRelation::removeEquality ( unsigned  pos)

◆ removeEqualityRange()

void IntegerRelation::removeEqualityRange ( unsigned  start,
unsigned  end 
)

Remove the (in)equalities at positions [start, end).

Definition at line 450 of file IntegerRelation.cpp.

References equalities, and mlir::presburger::Matrix< T >::removeRows().

Referenced by gaussianEliminate(), and truncate().

◆ removeIndependentConstraints()

void IntegerRelation::removeIndependentConstraints ( unsigned  pos,
unsigned  num 
)

Removes constraints that are independent of (i.e., do not have a coefficient) variables in the range [pos, pos + num).

Definition at line 2330 of file IntegerRelation.cpp.

References getIndependentConstraints(), getNumVars(), removeEquality(), and removeInequality().

Referenced by createSeparationCondition().

◆ removeInequality()

void IntegerRelation::removeInequality ( unsigned  pos)

◆ removeInequalityRange()

void IntegerRelation::removeInequalityRange ( unsigned  start,
unsigned  end 
)

Definition at line 456 of file IntegerRelation.cpp.

References inequalities, and mlir::presburger::Matrix< T >::removeRows().

Referenced by truncate().

◆ removeRedundantConstraints()

void IntegerRelation::removeRedundantConstraints ( )

Removes redundant constraints using Simplex.

Although the algorithm can theoretically take exponential time in the worst case (rare), it is known to perform much better in the average case. If V is the number of vertices in the polytope and C is the number of constraints, the algorithm takes O(VC) time.

Definition at line 1210 of file IntegerRelation.cpp.

References mlir::presburger::Simplex::detectRedundant(), gcdTightenInequalities(), and getNumInequalities().

◆ removeRedundantInequalities()

void IntegerRelation::removeRedundantInequalities ( )

A more expensive check than removeTrivialRedundancy to detect redundant inequalities.

Definition at line 1178 of file IntegerRelation.cpp.

References getNumInequalities().

◆ removeRedundantLocalVars()

void IntegerRelation::removeRedundantLocalVars ( )
protected

Removes local variables using equalities.

Each equality is checked if it can be reduced to the form: e = affine-expr, where e is a local variable and affine-expr is an affine expression not containing e. If an equality satisfies this form, the local variable is replaced in each constraint and then removed. The equality used to replace this local variable is also removed.

Definition at line 1420 of file IntegerRelation.cpp.

References mlir::presburger::abs(), atEq(), eliminateFromConstraint(), equalities, getNumDimAndSymbolVars(), getNumEqualities(), getNumInequalities(), getNumVars(), mlir::presburger::IntMatrix::normalizeRow(), removeEquality(), and removeVar().

Referenced by mlir::affine::FlatAffineRelation::compose(), isEmpty(), and mergeAndCompose().

◆ removeTrivialEqualities()

void IntegerRelation::removeTrivialEqualities ( )

Definition at line 2508 of file IntegerRelation.cpp.

References getEquality(), getNumEqualities(), rangeIsZero(), and removeEquality().

Referenced by isFullDim().

◆ removeTrivialRedundancy()

void IntegerRelation::removeTrivialRedundancy ( )

Removes duplicate constraints, trivially true constraints, and constraints that can be detected as redundant as a result of differing only in their constant term part.

A constraint of the form <non-negative constant> >= 0 is considered trivially true. This method is a linear time method on the constraints, does a single scan, and updates in place. It also normalizes constraints by their GCD and performs GCD tightening on inequalities.

A constraint of the form <non-negative constant> >= 0 is considered trivially true.

Definition at line 1828 of file IntegerRelation.cpp.

References gcdTightenInequalities(), and normalizeConstraintsByGCD().

Referenced by createSeparationCondition(), mlir::affine::simplifyIntegerSet(), and unionBoundingBox().

◆ removeVar() [1/2]

void IntegerRelation::removeVar ( unsigned  pos)

Removes the specified variable from the system.

Definition at line 386 of file IntegerRelation.cpp.

References removeVarRange().

◆ removeVar() [2/2]

void IntegerRelation::removeVar ( VarKind  kind,
unsigned  pos 
)

Removes variables of the specified kind with the specified pos (or within the specified range) from the system.

The specified location is relative to the first variable of the specified kind.

Definition at line 382 of file IntegerRelation.cpp.

References removeVarRange().

Referenced by createSeparationCondition(), eliminateRedundantLocalVar(), and removeRedundantLocalVars().

◆ removeVarRange() [1/2]

void IntegerRelation::removeVarRange ( unsigned  varStart,
unsigned  varLimit 
)
protected

Removes variables in the column range [varStart, varLimit), and copies any remaining valid data into place, updates member variables, and resizes arrays as needed.

Definition at line 404 of file IntegerRelation.cpp.

References mlir::presburger::Domain, getNumVarKind(), getNumVars(), getVarKindOffset(), mlir::presburger::Local, min(), mlir::presburger::Range, removeVarRange(), and mlir::presburger::Symbol.

◆ removeVarRange() [2/2]

void IntegerRelation::removeVarRange ( VarKind  kind,
unsigned  varStart,
unsigned  varLimit 
)
virtual

◆ resetIds()

void mlir::presburger::IntegerRelation::resetIds ( )
inline

◆ setAndEliminate() [1/2]

void IntegerRelation::setAndEliminate ( unsigned  pos,
ArrayRef< DynamicAPInt >  values 
)

Sets the values.size() variables starting at pos to the specified values and removes them.

Definition at line 546 of file IntegerRelation.cpp.

References mlir::presburger::Matrix< T >::addToColumn(), equalities, getNumCols(), getNumVars(), inequalities, and removeVarRange().

Referenced by constantFoldVar(), and setAndEliminate().

◆ setAndEliminate() [2/2]

void mlir::presburger::IntegerRelation::setAndEliminate ( unsigned  pos,
ArrayRef< int64_t >  values 
)
inline

Definition at line 366 of file IntegerRelation.h.

References mlir::presburger::getDynamicAPIntVec(), and setAndEliminate().

◆ setDimSymbolSeparation()

void mlir::presburger::IntegerRelation::setDimSymbolSeparation ( unsigned  newSymbolCount)
inline

Changes the partition between dimensions and symbols.

Depending on the new symbol count, either a chunk of dimensional variables immediately before the split become symbols, or some of the symbols immediately after the split become dimensions.

Definition at line 649 of file IntegerRelation.h.

References mlir::presburger::PresburgerSpace::setVarSymbolSeparation(), and space.

Referenced by createFullTiles(), createSeparationCondition(), and turnSymbolIntoDim().

◆ setId()

void IntegerRelation::setId ( VarKind  kind,
unsigned  i,
Identifier  id 
)

Set the identifier for the ith variable of the specified kind of the IntegerRelation's PresburgerSpace.

The index is relative to the kind of the variable.

Definition at line 71 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerSpace::getNumVarKind(), mlir::presburger::PresburgerSpace::isUsingIds(), mlir::presburger::Local, mlir::presburger::PresburgerSpace::setId(), and space.

Referenced by mlir::affine::MemRefAccess::getAccessRelation(), and mlir::affine::getRelationFromMap().

◆ setSpace()

void IntegerRelation::setSpace ( const PresburgerSpace oSpace)

Set the space to oSpace, which should have the same number of ids as the current space.

Definition at line 58 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerSpace::getNumVars(), and space.

◆ setSpaceExceptLocals()

void IntegerRelation::setSpaceExceptLocals ( const PresburgerSpace oSpace)

Set the space to oSpace, which should not have any local ids.

oSpace can have fewer ids than the current space; in that case, the the extra ids in this that are not accounted for by oSpace will be considered as local ids. oSpace should not have more ids than the current space; this will result in an assert failure.

Definition at line 63 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerSpace::getNumLocalVars(), getNumVars(), mlir::presburger::PresburgerSpace::getNumVars(), mlir::presburger::PresburgerSpace::insertVar(), mlir::presburger::Local, and space.

◆ simplify()

void IntegerRelation::simplify ( )

Simplify the constraint system by removing canonicalizing constraints and removing redundant constraints.

Definition at line 1400 of file IntegerRelation.cpp.

References gaussianEliminate(), isObviouslyEmpty(), normalizeConstraintsByGCD(), and removeDuplicateConstraints().

Referenced by mlir::presburger::PresburgerRelation::simplify().

◆ subtract()

PresburgerRelation IntegerRelation::subtract ( const PresburgerRelation set) const

Return the set difference of this set and the given set, i.e., return this \ set.

Definition at line 350 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerRelation::subtract().

Referenced by mlir::presburger::IntegerPolyhedron::subtract().

◆ swapVar()

void IntegerRelation::swapVar ( unsigned  posA,
unsigned  posB 
)
virtual

◆ truncate()

void IntegerRelation::truncate ( const CountsSnapshot counts)

◆ truncateVarKind() [1/2]

void IntegerRelation::truncateVarKind ( VarKind  kind,
const CountsSnapshot counts 
)
protected

Truncate the vars to the number in the space of the specified CountsSnapshot.

Definition at line 209 of file IntegerRelation.cpp.

References mlir::presburger::PresburgerSpace::getNumVarKind(), mlir::presburger::IntegerRelation::CountsSnapshot::getSpace(), and truncateVarKind().

◆ truncateVarKind() [2/2]

void IntegerRelation::truncateVarKind ( VarKind  kind,
unsigned  num 
)
protected

Truncate the vars of the specified kind to the specified number by dropping some vars at the end.

num must be less than the current number.

Definition at line 203 of file IntegerRelation.cpp.

References getNumVarKind(), and removeVarRange().

Referenced by truncate(), and truncateVarKind().

◆ unionBoundingBox()

LogicalResult IntegerRelation::unionBoundingBox ( const IntegerRelation other)

Updates the constraints to be the smallest bounding (enclosing) box that contains the points of this set and that of other, with the symbols being treated specially.

For each of the dimensions, the min of the lower bounds (symbolic) and the max of the upper bounds (symbolic) is computed to determine such a bounding box. other is expected to have the same dimensional variables as this constraint system (in the same order).

E.g.: 1) this = {0 <= d0 <= 127}, other = {16 <= d0 <= 192}, output = {0 <= d0 <= 192} 2) this = {s0 + 5 <= d0 <= s0 + 20}, other = {s0 + 1 <= d0 <= s0 + 9}, output = {s0 + 1 <= d0 <= s0 + 20} 3) this = {0 <= d0 <= 5, 1 <= d1 <= 9} other = {2 <= d0 <= 6, 5 <= d1 <= 15}, output = {0 <= d0 <= 6, 1 <= d1 <= 15}

Definition at line 2181 of file IntegerRelation.cpp.

References addInequality(), append(), clearConstraints(), copy(), getCommonConstraints(), getConstantBound(), getConstantBoundOnDimSize(), getNumCols(), getNumDimVars(), getNumLocalVars(), getNumSymbolVars(), mlir::presburger::PresburgerSpace::getRelationSpace(), getSpace(), mlir::presburger::PresburgerSpace::isEqual(), mlir::presburger::LB, max(), min(), removeTrivialRedundancy(), space, and mlir::presburger::UB.

Member Data Documentation

◆ equalities

IntMatrix mlir::presburger::IntegerRelation::equalities
protected

◆ inequalities

IntMatrix mlir::presburger::IntegerRelation::inequalities
protected

◆ kExplosionFactor

constexpr static unsigned mlir::presburger::IntegerRelation::kExplosionFactor = 32
staticconstexprprotected

A parameter that controls detection of an unrealistic number of constraints.

If the number of constraints is this many times the number of variables, we consider such a system out of line with the intended use case of IntegerRelation.

Definition at line 869 of file IntegerRelation.h.

Referenced by isEmpty().

◆ space

PresburgerSpace mlir::presburger::IntegerRelation::space
protected

The documentation for this class was generated from the following files: