Move mathematical functions to utils.h

Author: Theodoros Theodoridis

include/tc/autotuner/utils/utils.h

@@ -21,6 +21,7 @@
 #include "tc/core/cuda/cuda.h"
 #include "tc/core/cuda/cuda_mapping_options.h"
 #include "tc/core/utils/dlpack.h"
+#include "tc/lang/tree.h"
 
 #include <llvm/ADT/Optional.h>
 
@@ -59,6 +60,11 @@ llvm::Optional<CudaMappingOptions> getBestOptions(
     const std::vector<const DLTensor*>& inputs,
     const std::vector<const DLTensor*>& outputs);
 
+double mean(std::vector<double>& v);
+double stdv(std::vector<double>& v, double mean);
+void normalizeVector(std::vector<double>& v);
+void sigmaScale(std::vector<double>& v);
+
 } // namespace autotune
 } // namespace tc
 

src/autotuner/genetic_search.cc

@@ -21,6 +21,8 @@
 #include <random>
 #include <sstream>
 
+#include "tc/autotuner/utils/utils.h"
+
 namespace tc {
 namespace autotune {
 
@@ -74,39 +76,6 @@ void mutate(
   }
 }
 
-double mean(std::vector<double>& v) {
-  if (v.empty()) {
-    throw std::invalid_argument("Cannot compute the mean of an empty vector.");
-  }
-  auto sum = std::accumulate(v.begin(), v.end(), 0.0);
-  return sum / v.size();
-}
-
-double stdv(std::vector<double>& v, double mean) {
-  std::vector<double> diffs(v.size());
-  std::transform(v.begin(), v.end(), diffs.begin(), [mean](double val) {
-    return val - mean;
-  });
-
-  auto squareSum =
-      std::inner_product(diffs.begin(), diffs.end(), diffs.begin(), 0.0);
-  return std::sqrt(squareSum / v.size());
-}
-
-void sigmaScale(std::vector<double>& v) {
-  auto m = mean(v);
-  auto s = stdv(v, m);
-  std::transform(v.begin(), v.end(), v.begin(), [m, s](double val) {
-    return std::max(val - (m - 2 * s), 0.0);
-  });
-}
-
-void normalizeVector(std::vector<double>& v) {
-  auto sum = std::accumulate(v.begin(), v.end(), 0.0);
-  std::transform(
-      v.begin(), v.end(), v.begin(), [sum](double v) { return v / sum; });
-}
-
 std::vector<double> computeNormalizedFitness(
     const GeneticSearch::Population& population) {
   std::vector<double> fitness;

src/autotuner/utils/utils.cc

@@ -15,6 +15,7 @@
  */
 #include <algorithm>
 #include <cmath>
+#include <numeric>
 
 #include "tc/aten/aten_compiler.h"
 #include "tc/autotuner/utils/utils.h"
@@ -128,5 +129,38 @@ llvm::Optional<CudaMappingOptions> getBestOptions(
   return llvm::Optional<CudaMappingOptions>{};
 }
 
+double mean(std::vector<double>& v) {
+  if (v.empty()) {
+    throw std::invalid_argument("Cannot compute the mean of an empty vector.");
+  }
+  auto sum = std::accumulate(v.begin(), v.end(), 0.0);
+  return sum / v.size();
+}
+
+double stdv(std::vector<double>& v, double mean) {
+  std::vector<double> diffs(v.size());
+  std::transform(v.begin(), v.end(), diffs.begin(), [mean](double val) {
+    return val - mean;
+  });
+
+  auto squareSum =
+      std::inner_product(diffs.begin(), diffs.end(), diffs.begin(), 0.0);
+  return std::sqrt(squareSum / v.size());
+}
+
+void sigmaScale(std::vector<double>& v) {
+  auto m = mean(v);
+  auto s = stdv(v, m);
+  std::transform(v.begin(), v.end(), v.begin(), [m, s](double val) {
+    return std::max(val - (m - 2 * s), 0.0);
+  });
+}
+
+void normalizeVector(std::vector<double>& v) {
+  auto sum = std::accumulate(v.begin(), v.end(), 0.0);
+  std::transform(
+      v.begin(), v.end(), v.begin(), [sum](double v) { return v / sum; });
+}
+
 } // namespace autotune
 } // namespace tc