From 379494fb6e0002955c7bced843a26f34e78e3b24 Mon Sep 17 00:00:00 2001
From: Akash Kothari <akashk4@tyler.cs.illinois.edu>
Date: Wed, 15 Jan 2020 05:51:23 -0600
Subject: [PATCH] Clean some code up in DFG2LLVM_NVPTX source

---
 .../DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp         | 53 +++++++++----------
 1 file changed, 26 insertions(+), 27 deletions(-)

diff --git a/hpvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp b/hpvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
index f1f1a3983c..97e0f3411b 100644
--- a/hpvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
+++ b/hpvm/lib/Transforms/DFG2LLVM_NVPTX/DFG2LLVM_NVPTX.cpp
@@ -103,21 +103,21 @@ public:
   std::vector<Value*> localWGSize;
   std::vector<int> localDimMap;
 
-  std::map<unsigned, unsigned> getInArgMap() {
+  std::map<unsigned, unsigned> &getInArgMap() {
     return inArgMap;
   }
   void setInArgMap(std::map<unsigned, unsigned> map) {
     inArgMap = map;
   }
 
-  std::map<unsigned, std::pair<Value*, unsigned> > getSharedInArgMap() {
+  std::map<unsigned, std::pair<Value*, unsigned> > &getSharedInArgMap() {
     return sharedInArgMap;
   }
   void setSharedInArgMap(std::map<unsigned, std::pair<Value*, unsigned> > map) {
     sharedInArgMap = map;
   }
 
-  std::vector<unsigned> getOutArgMap() {
+  std::vector<unsigned> &getOutArgMap() {
     return outArgMap;
   }
   void setOutArgMap(std::vector<unsigned> map) {
@@ -128,7 +128,7 @@ public:
     localWGSize = V;
   }
 
-  bool hasLocalWG() {
+  bool hasLocalWG() const {
     return blockDim != 0;
   }
 };
@@ -210,37 +210,37 @@ public:
 
     // Copying instead of creating new, in order to preserve required info (metadata)
     // Remove functions, global variables and aliases
-    std::vector<GlobalVariable*> gvv = std::vector<GlobalVariable*>();
+    std::vector<GlobalVariable*> GVVect;
     for (Module::global_iterator mi = KernelM->global_begin(),
          me = KernelM->global_end(); (mi != me); ++mi) {
-      GlobalVariable* gv = &*mi;
-      gvv.push_back(gv);
+      GlobalVariable* GV = &*mi;
+      GVVect.push_back(GV);
     }
-    for (std::vector<GlobalVariable*>::iterator vi = gvv.begin(); vi != gvv.end(); ++vi) {
-      (*vi)->replaceAllUsesWith(UndefValue::get((*vi)->getType()));
-      (*vi)->eraseFromParent();
+    for (auto *GV : GVVect) {
+      GV->replaceAllUsesWith(UndefValue::get(GV->getType()));
+      GV->eraseFromParent();
     }
 
-    std::vector<Function*> fv = std::vector<Function*>();
+    std::vector<Function*> FuncVect;
     for (Module::iterator mi = KernelM->begin(),
          me = KernelM->end(); (mi != me); ++mi) {
-      Function* f = &*mi;
-      fv.push_back(f);
+      Function* F = &*mi;
+      FuncVect.push_back(F);
     }
-    for (std::vector<Function*>::iterator vi = fv.begin(); vi != fv.end(); ++vi) {
-      (*vi)->replaceAllUsesWith(UndefValue::get((*vi)->getType()));
-      (*vi)->eraseFromParent();
+    for (auto *F : FuncVect) {
+      F->replaceAllUsesWith(UndefValue::get(F->getType()));
+      F->eraseFromParent();
     }
 
-    std::vector<GlobalAlias*> av = std::vector<GlobalAlias*>();
+    std::vector<GlobalAlias*> GAVect;
     for (Module::alias_iterator mi = KernelM->alias_begin(),
          me = KernelM->alias_end(); (mi != me); ++mi) {
-      GlobalAlias* a = &*mi;
-      av.push_back(a);
+      GlobalAlias* GA = &*mi;
+      GAVect.push_back(GA);
     }
-    for (std::vector<GlobalAlias*>::iterator vi = av.begin(); vi != av.end(); ++vi) {
-      (*vi)->replaceAllUsesWith(UndefValue::get((*vi)->getType()));
-      (*vi)->eraseFromParent();
+    for (auto *GA : GAVect) {
+      GA->replaceAllUsesWith(UndefValue::get(GA->getType()));
+      GA->eraseFromParent();
     }
 
     changeDataLayout(*KernelM);
@@ -458,7 +458,7 @@ void CGT_NVPTX::insertRuntimeCalls(DFInternalNode* N, Kernel* K, const Twine& Fi
   // context
   std::vector<Value*> DevicePointers;
 
-  std::map<unsigned, unsigned> kernelInArgMap = K->getInArgMap();
+  std::map<unsigned, unsigned> &kernelInArgMap = K->getInArgMap();
   /*
     for(unsigned i=0; i<KF->getFunctionType()->getNumParams(); i++) {
 
@@ -468,10 +468,9 @@ void CGT_NVPTX::insertRuntimeCalls(DFInternalNode* N, Kernel* K, const Twine& Fi
 
   */
 
-  for(std::map<unsigned, unsigned>::iterator ib = kernelInArgMap.begin(),
-      ie = kernelInArgMap.end(); ib != ie; ++ib) {
-    unsigned i = ib->first;
-    Value* inputVal = getArgumentAt(F_X86, ib->second);
+  for(auto &InArgMapPair  : kernelInArgMap) {
+    unsigned i = InArgMapPair.first;
+    Value* inputVal = getArgumentAt(F_X86, InArgMapPair.second);
     DEBUG(errs() << "\tArgument "<< i<< " = "  << *inputVal << "\n");
 
     // input value has been obtained.
-- 
GitLab