Merging with branch master and fixing bugs

[satune.git] / src / Tuner / autotuner.cc

diff --git a/src/Tuner/autotuner.cc b/src/Tuner/autotuner.cc
index 37a9d077903e24f2393d3dfec3bf3fa6222adc82..f4509677eed754d27d63ed2bb8cea63d929f1a46 100644 (file)
--- a/src/Tuner/autotuner.cc
+++ b/src/Tuner/autotuner.cc
@@ -1,3 +1,99 @@
 #include "autotuner.h"
 #include "csolver.h"
+#include "searchtuner.h"
+#include <math.h>
+#include <stdlib.h>
+#include <float.h>
 
+AutoTuner::AutoTuner(uint _budget) :
+       budget(_budget) {
+}
+
+void AutoTuner::addProblem(CSolver *solver) {
+       solvers.push(solver);
+}
+
+long long AutoTuner::evaluate(CSolver * problem, SearchTuner *tuner) {
+       CSolver * copy=problem->clone();
+       copy->setTuner(tuner);
+       int result = copy->startEncoding();
+       long long elapsedTime=copy->getElapsedTime();
+       long long encodeTime=copy->getEncodeTime();
+       long long solveTime=copy->getSolveTime();
+       long long metric=elapsedTime;
+       model_print("Elapsed Time: %llu\n", elapsedTime);
+       model_print("Encode Time: %llu\n", encodeTime);
+       model_print("Solve Time: %llu\n", solveTime);
+       delete copy;
+       return metric;
+}
+
+double AutoTuner::evaluateAll(SearchTuner *tuner) {
+       double product=1;
+       for(uint i=0;i<solvers.getSize();i++) {
+               CSolver * problem=solvers.get(i);
+               double score=evaluate(problem, tuner);
+               product*=score;
+       }
+       return pow(product, 1/((double)solvers.getSize()));
+}
+
+SearchTuner * AutoTuner::mutateTuner(SearchTuner * oldTuner, uint k) {
+       SearchTuner *newTuner=oldTuner->copyUsed();
+       uint numSettings=oldTuner->getSize();
+       double factor=0.3;//Adjust this factor...
+       uint settingsToMutate=(uint)(factor*(((double)numSettings) * k)/(budget));
+       if (settingsToMutate < 1)
+               settingsToMutate=1;
+       model_print("Mutating %u settings\n", settingsToMutate);
+       while(settingsToMutate-- != 0) {
+               newTuner->randomMutate();
+       }
+       return newTuner;
+}
+
+
+void AutoTuner::tune() {
+       SearchTuner * bestTuner = NULL;
+       double bestScore=DBL_MAX;
+
+       SearchTuner * oldTuner=new SearchTuner();
+       double base_temperature=evaluateAll(oldTuner);
+       double oldScore=base_temperature;
+
+       for (uint i=0;i<budget;i++) {
+               SearchTuner *newTuner=mutateTuner(oldTuner, i);
+               double newScore=evaluateAll(newTuner);
+               newTuner->printUsed();
+               model_print("Received score %f\n", newScore);
+               double scoreDiff=newScore - oldScore; //smaller is better
+               if (newScore < bestScore) {
+                       if (bestTuner != NULL)
+                               delete bestTuner;
+                       bestScore = newScore;
+                       bestTuner = newTuner->copyUsed();
+               }
+
+               double acceptanceP;
+               if (scoreDiff < 0) {
+                       acceptanceP = 1;
+               } else {
+                       double currTemp=base_temperature * (((double)budget - i) / budget);
+                       acceptanceP = exp(-scoreDiff / currTemp);
+               }
+               double ran = ((double)random()) / RAND_MAX;
+               if (ran <= acceptanceP) {
+                       delete oldTuner;
+                       oldScore = newScore;
+                       oldTuner = newTuner;
+               } else {
+                       delete newTuner;
+               }
+       }
+       model_print("Best tuner:\n");
+       bestTuner->print();
+       model_print("Received score %f\n", bestScore);
+       if (bestTuner != NULL)
+               delete bestTuner;
+       delete oldTuner;
+}