aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/Target/PTX/PTXParamManager.cpp6
-rw-r--r--lib/Target/PTX/PTXParamManager.h10
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();