diff options
-rw-r--r-- | lib/Target/PTX/PTXParamManager.cpp | 6 | ||||
-rw-r--r-- | lib/Target/PTX/PTXParamManager.h | 10 |
2 files changed, 8 insertions, 8 deletions
diff --git a/lib/Target/PTX/PTXParamManager.cpp b/lib/Target/PTX/PTXParamManager.cpp index f4945d9463..7753787ebc 100644 --- a/lib/Target/PTX/PTXParamManager.cpp +++ b/lib/Target/PTX/PTXParamManager.cpp @@ -32,7 +32,7 @@ unsigned PTXParamManager::addArgumentParam(unsigned Size) { unsigned Index = AllParams.size(); AllParams[Index] = Param; - ArgumentParams.insert(Index); + ArgumentParams.push_back(Index); return Index; } @@ -49,7 +49,7 @@ unsigned PTXParamManager::addReturnParam(unsigned Size) { unsigned Index = AllParams.size(); AllParams[Index] = Param; - ReturnParams.insert(Index); + ReturnParams.push_back(Index); return Index; } @@ -66,7 +66,7 @@ unsigned PTXParamManager::addLocalParam(unsigned Size) { unsigned Index = AllParams.size(); AllParams[Index] = Param; - LocalParams.insert(Index); + LocalParams.push_back(Index); return Index; } diff --git a/lib/Target/PTX/PTXParamManager.h b/lib/Target/PTX/PTXParamManager.h index 05b0d31f7a..1a18c74f98 100644 --- a/lib/Target/PTX/PTXParamManager.h +++ b/lib/Target/PTX/PTXParamManager.h @@ -16,7 +16,7 @@ #define PTX_PARAM_MANAGER_H #include "llvm/ADT/DenseMap.h" -#include "llvm/ADT/DenseSet.h" +#include "llvm/ADT/SmallVector.h" namespace llvm { @@ -40,13 +40,13 @@ private: }; DenseMap<unsigned, PTXParam> AllParams; - DenseSet<unsigned> ArgumentParams; - DenseSet<unsigned> ReturnParams; - DenseSet<unsigned> LocalParams; + SmallVector<unsigned, 4> ArgumentParams; + SmallVector<unsigned, 4> ReturnParams; + SmallVector<unsigned, 4> LocalParams; public: - typedef DenseSet<unsigned>::const_iterator param_iterator; + typedef SmallVector<unsigned, 4>::const_iterator param_iterator; PTXParamManager(); |