PTX: Fix parameter order bug

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@140394 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/Target/PTX/PTXParamManager.cpp b/lib/Target/PTX/PTXParamManager.cpp
index f4945d9463910c517ab72133317ae53d4e4ae59c..7753787ebc519f421b578195a74dd06c3262effc 100644 (file)
--- 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 05b0d31f7a8a9777fcc4efb91fb9a62250bab644..1a18c74f98c14feccf8eabc678e520823d93b6f6 100644 (file)
--- 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();