scaleVector function changed to include std::transform

IITH-Compilers · Jul 31, 2024 · 90ffa85 · 90ffa85
1 parent ac7fce7
commit 90ffa85
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 9 deletions.
diff --git a/src/IR2Vec.cpp b/src/IR2Vec.cpp
@@ -157,7 +157,7 @@ void generateSYMEncodings() {
   o.close();
 }
 
-void collectIR() {
+void collectIRfunc() {
   auto M = getLLVMIR();
   CollectIR cir(M);
   std::ofstream o;
@@ -232,7 +232,7 @@ int main(int argc, char **argv) {
   } else if (sym) {
     generateSYMEncodings();
   } else if (collectIR) {
-    collectIR();
+    collectIRfunc();
   }
   return 0;
 }
diff --git a/src/include/utils.h b/src/include/utils.h
@@ -18,6 +18,7 @@
 
 #include <cxxabi.h>
 
+#include <algorithm>
 #include <map>
 
 namespace IR2Vec {
@@ -48,7 +49,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);

diff --git a/src/utils.cpp b/src/utils.cpp
@@ -43,12 +43,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();