RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-11-25 12:12:47 +01:00
Code Issues Projects Releases Wiki Activity

[OpenMPOpt] Add support for BooleanStateWithSetVector

Browse Source 
D101977 added `BooleanStateWithPtrSetVector` to store pointers to a set meanwhile
tracking boolean state. One of the limitation is that it can only store pointer.
We might want it to store other types of values, such as integer for parallel
level. This patch generalizes the idea and create `BooleanStateWithSetVector`.
`BooleanStateWithPtrSetVector` therefore becomes a type alias of `BooleanStateWithSetVector`.

Reviewed By: jdoerfert

Differential Revision: https://reviews.llvm.org/D106149
This commit is contained in:
Shilei Tian 2021-07-22 12:37:43 -04:00
parent 86e8114938
commit 04b998f247
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
lib/Transforms/IPO/OpenMPOpt.cpp
View File

@ -447,20 +447,19 @@ struct OMPInformationCache : public InformationCache {
};
template <typename Ty, bool InsertInvalidates = true>
struct BooleanStateWithPtrSetVector : public BooleanState {
bool contains(Ty *Elem) const { return Set.contains(Elem); }
bool insert(Ty *Elem) {
struct BooleanStateWithSetVector : public BooleanState {
bool contains(const Ty &Elem) const { return Set.contains(Elem); }
bool insert(const Ty &Elem) {
if (InsertInvalidates)
BooleanState::indicatePessimisticFixpoint();
return Set.insert(Elem);
}
Ty *operator[](int Idx) const { return Set[Idx]; }
bool operator==(const BooleanStateWithPtrSetVector &RHS) const {
const Ty &operator[](int Idx) const { return Set[Idx]; }
bool operator==(const BooleanStateWithSetVector &RHS) const {
return BooleanState::operator==(RHS) && Set == RHS.Set;
}
bool operator!=(const BooleanStateWithPtrSetVector &RHS) const {
bool operator!=(const BooleanStateWithSetVector &RHS) const {
return !(*this == RHS);
}
@ -468,8 +467,7 @@ struct BooleanStateWithPtrSetVector : public BooleanState {
size_t size() const { return Set.size(); }
/// "Clamp" this state with \p RHS.
BooleanStateWithPtrSetVector &
operator^=(const BooleanStateWithPtrSetVector &RHS) {
BooleanStateWithSetVector &operator^=(const BooleanStateWithSetVector &RHS) {
BooleanState::operator^=(RHS);
Set.insert(RHS.Set.begin(), RHS.Set.end());
return *this;
@ -477,7 +475,7 @@ struct BooleanStateWithPtrSetVector : public BooleanState {
private:
/// A set to keep track of elements.
SetVector<Ty *> Set;
SetVector<Ty> Set;
public:
typename decltype(Set)::iterator begin() { return Set.begin(); }
@ -486,6 +484,10 @@ public:
typename decltype(Set)::const_iterator end() const { return Set.end(); }
};
template <typename Ty, bool InsertInvalidates = true>
using BooleanStateWithPtrSetVector =
BooleanStateWithSetVector<Ty *, InsertInvalidates>;
struct KernelInfoState : AbstractState {
/// Flag to track if we reached a fixpoint.
bool IsAtFixpoint = false;