diff --git a/lib/Target/PTX/PTXParamManager.cpp b/lib/Target/PTX/PTXParamManager.cpp index f4945d94639..7753787ebc5 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 05b0d31f7a8..1a18c74f98c 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 AllParams; - DenseSet ArgumentParams; - DenseSet ReturnParams; - DenseSet LocalParams; + SmallVector ArgumentParams; + SmallVector ReturnParams; + SmallVector LocalParams; public: - typedef DenseSet::const_iterator param_iterator; + typedef SmallVector::const_iterator param_iterator; PTXParamManager();