scaleVector function changed to include std::transform

Author: nishant-sachdeva

src/IR2Vec.cpp

@@ -155,7 +155,7 @@ void generateSYMEncodings() {
   o.close();
 }
 
-void collectIR() {
+void collectIRfunc() {
   auto M = getLLVMIR();
   CollectIR cir(M);
   std::ofstream o;
@@ -230,7 +230,7 @@ int main(int argc, char **argv) {
   } else if (sym) {
     generateSYMEncodings();
   } else if (collectIR) {
-    collectIR();
+    collectIRfunc();
   }
   return 0;
 }

src/include/utils.h

@@ -16,6 +16,7 @@
 
 #include <cxxabi.h>
 
+#include <algorithm>
 #include <map>
 
 namespace IR2Vec {
@@ -46,7 +47,10 @@ extern float WT;
 extern bool debug;
 extern std::unordered_map<std::string, Vector> opcMap;
 std::unique_ptr<llvm::Module> getLLVMIR();
-void scaleVector(Vector &vec, float factor);
+inline void scaleVector(Vector &vec, float factor) {
+  std::transform(vec.begin(), vec.end(), vec.begin(),
+                 [factor](const auto &val) { return val * factor; });
+}
 // newly added
 std::string getDemagledName(const llvm::Function *function);
 char *getActualName(llvm::Function *function);

src/utils.cpp

@@ -41,12 +41,6 @@ std::unique_ptr<Module> IR2Vec::getLLVMIR() {
   return M;
 }
 
-void IR2Vec::scaleVector(Vector &vec, float factor) {
-  for (unsigned i = 0; i < vec.size(); i++) {
-    vec[i] = vec[i] * factor;
-  }
-}
-
 // Function to get demangled function name
 std::string IR2Vec::getDemagledName(const llvm::Function *function) {
   auto functionName = function->getName().str();