PTX: Fix parameter order bug

llvm-svn: 140394

2 files changed, 8 insertions, 8 deletions

diff --git a/llvm/lib/Target/PTX/PTXParamManager.cpp b/llvm/lib/Target/PTX/PTXParamManager.cpp
index f4945d94639..7753787ebc5 100644
--- a/llvm/lib/Target/PTX/PTXParamManager.cpp
+++ b/llvm/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/llvm/lib/Target/PTX/PTXParamManager.h b/llvm/lib/Target/PTX/PTXParamManager.h
index 05b0d31f7a8..1a18c74f98c 100644
--- a/llvm/lib/Target/PTX/PTXParamManager.h
+++ b/llvm/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();