diff --git a/mlir/include/mlir/Analysis/Presburger/PresburgerSet.h b/mlir/include/mlir/Analysis/Presburger/PresburgerSet.h index 5963e60071cc..e6be93902759 100644 --- a/mlir/include/mlir/Analysis/Presburger/PresburgerSet.h +++ b/mlir/include/mlir/Analysis/Presburger/PresburgerSet.h @@ -36,10 +36,10 @@ public: unsigned getNumPolys() const; /// Return the number of real dimensions. - unsigned getNumDims() const; + unsigned getNumDimIds() const; /// Return the number of symbolic dimensions. - unsigned getNumSyms() const; + unsigned getNumSymbolIds() const; /// Return a reference to the list of IntegerPolyhedrons. ArrayRef getAllIntegerPolyhedron() const; @@ -82,9 +82,11 @@ public: bool isEqual(const PresburgerSet &set) const; /// Return a universe set of the specified type that contains all points. - static PresburgerSet getUniverse(unsigned nDim = 0, unsigned nSym = 0); + static PresburgerSet getUniverse(unsigned numDims = 0, + unsigned numSymbols = 0); /// Return an empty set of the specified type that contains no points. - static PresburgerSet getEmptySet(unsigned nDim = 0, unsigned nSym = 0); + static PresburgerSet getEmptySet(unsigned numDims = 0, + unsigned numSymbols = 0); /// Return true if all the sets in the union are known to be integer empty /// false otherwise. @@ -102,19 +104,19 @@ public: private: /// Construct an empty PresburgerSet. - PresburgerSet(unsigned nDim = 0, unsigned nSym = 0) - : nDim(nDim), nSym(nSym) {} + PresburgerSet(unsigned numDims = 0, unsigned numSymbols = 0) + : numDims(numDims), numSymbols(numSymbols) {} /// Return the set difference poly \ set. static PresburgerSet getSetDifference(IntegerPolyhedron poly, const PresburgerSet &set); /// Number of identifiers corresponding to real dimensions. - unsigned nDim; + unsigned numDims; /// Number of symbolic dimensions, unknown but constant for analysis, as in /// IntegerPolyhedron. - unsigned nSym; + unsigned numSymbols; /// The list of integerPolyhedrons that this set is the union of. SmallVector integerPolyhedrons; diff --git a/mlir/lib/Analysis/Presburger/PresburgerSet.cpp b/mlir/lib/Analysis/Presburger/PresburgerSet.cpp index 829d786c8883..08d15bc88c74 100644 --- a/mlir/lib/Analysis/Presburger/PresburgerSet.cpp +++ b/mlir/lib/Analysis/Presburger/PresburgerSet.cpp @@ -16,7 +16,7 @@ using namespace mlir; using namespace presburger_utils; PresburgerSet::PresburgerSet(const IntegerPolyhedron &poly) - : nDim(poly.getNumDimIds()), nSym(poly.getNumSymbolIds()) { + : numDims(poly.getNumDimIds()), numSymbols(poly.getNumSymbolIds()) { unionPolyInPlace(poly); } @@ -24,9 +24,9 @@ unsigned PresburgerSet::getNumPolys() const { return integerPolyhedrons.size(); } -unsigned PresburgerSet::getNumDims() const { return nDim; } +unsigned PresburgerSet::getNumDimIds() const { return numDims; } -unsigned PresburgerSet::getNumSyms() const { return nSym; } +unsigned PresburgerSet::getNumSymbolIds() const { return numSymbols; } ArrayRef PresburgerSet::getAllIntegerPolyhedron() const { return integerPolyhedrons; @@ -42,10 +42,10 @@ PresburgerSet::getIntegerPolyhedron(unsigned index) const { /// compatible spaces. static void assertDimensionsCompatible(const IntegerPolyhedron &poly, const PresburgerSet &set) { - assert(poly.getNumDimIds() == set.getNumDims() && + assert(poly.getNumDimIds() == set.getNumDimIds() && "Number of dimensions of the IntegerPolyhedron and PresburgerSet" "do not match!"); - assert(poly.getNumSymbolIds() == set.getNumSyms() && + assert(poly.getNumSymbolIds() == set.getNumSymbolIds() && "Number of symbols of the IntegerPolyhedron and PresburgerSet" "do not match!"); } @@ -53,9 +53,9 @@ static void assertDimensionsCompatible(const IntegerPolyhedron &poly, /// Assert that the two PresburgerSets live in compatible spaces. static void assertDimensionsCompatible(const PresburgerSet &setA, const PresburgerSet &setB) { - assert(setA.getNumDims() == setB.getNumDims() && + assert(setA.getNumDimIds() == setB.getNumDimIds() && "Number of dimensions of the PresburgerSets do not match!"); - assert(setA.getNumSyms() == setB.getNumSyms() && + assert(setA.getNumSymbolIds() == setB.getNumSymbolIds() && "Number of symbols of the PresburgerSets do not match!"); } @@ -91,14 +91,16 @@ bool PresburgerSet::containsPoint(ArrayRef point) const { }); } -PresburgerSet PresburgerSet::getUniverse(unsigned nDim, unsigned nSym) { - PresburgerSet result(nDim, nSym); - result.unionPolyInPlace(IntegerPolyhedron::getUniverse(nDim, nSym)); +PresburgerSet PresburgerSet::getUniverse(unsigned numDims, + unsigned numSymbols) { + PresburgerSet result(numDims, numSymbols); + result.unionPolyInPlace(IntegerPolyhedron::getUniverse(numDims, numSymbols)); return result; } -PresburgerSet PresburgerSet::getEmptySet(unsigned nDim, unsigned nSym) { - return PresburgerSet(nDim, nSym); +PresburgerSet PresburgerSet::getEmptySet(unsigned numDims, + unsigned numSymbols) { + return PresburgerSet(numDims, numSymbols); } // Return the intersection of this set with the given set. @@ -111,7 +113,7 @@ PresburgerSet PresburgerSet::getEmptySet(unsigned nDim, unsigned nSym) { PresburgerSet PresburgerSet::intersect(const PresburgerSet &set) const { assertDimensionsCompatible(set, *this); - PresburgerSet result(nDim, nSym); + PresburgerSet result(getNumDimIds(), getNumSymbolIds()); for (const IntegerPolyhedron &csA : integerPolyhedrons) { for (const IntegerPolyhedron &csB : set.integerPolyhedrons) { IntegerPolyhedron csACopy = csA, csBCopy = csB; @@ -336,14 +338,14 @@ PresburgerSet PresburgerSet::getSetDifference(IntegerPolyhedron poly, /// Return the complement of this set. PresburgerSet PresburgerSet::complement() const { return getSetDifference( - IntegerPolyhedron::getUniverse(getNumDims(), getNumSyms()), *this); + IntegerPolyhedron::getUniverse(getNumDimIds(), getNumSymbolIds()), *this); } /// Return the result of subtract the given set from this set, i.e., /// return `this \ set`. PresburgerSet PresburgerSet::subtract(const PresburgerSet &set) const { assertDimensionsCompatible(set, *this); - PresburgerSet result(nDim, nSym); + PresburgerSet result(getNumDimIds(), getNumSymbolIds()); // We compute (U_i t_i) \ (U_i set_i) as U_i (t_i \ V_i set_i). for (const IntegerPolyhedron &poly : integerPolyhedrons) result.unionSetInPlace(getSetDifference(poly, set)); @@ -386,7 +388,8 @@ bool PresburgerSet::findIntegerSample(SmallVectorImpl &sample) { } PresburgerSet PresburgerSet::coalesce() const { - PresburgerSet newSet = PresburgerSet::getEmptySet(getNumDims(), getNumSyms()); + PresburgerSet newSet = + PresburgerSet::getEmptySet(getNumDimIds(), getNumSymbolIds()); llvm::SmallBitVector isRedundant(getNumPolys()); for (unsigned i = 0, e = integerPolyhedrons.size(); i < e; ++i) {