Merge pull request #173 from mcian/PSO_params

Add PSO parameters support and search strategy selection from command…

1 files changed, 14 insertions, 0 deletions

diff --git a/src/utilities/utilities.hpp b/src/utilities/utilities.hpp
index 72997d7f..a9c492f3 100644
--- a/src/utilities/utilities.hpp
+++ b/src/utilities/utilities.hpp
@@ -79,6 +79,14 @@ constexpr auto kArgBatchCount = "batch_num";
 
 // The tuner-specific arguments in string form
 constexpr auto kArgFraction = "fraction";
+constexpr auto kArgHeuristicSelection = "heuristic";
+// PSO tuner-specific arguments in string form
+constexpr auto kArgPsoSwarmSize = "pso_swarm_size";
+constexpr auto kArgPsoInfGlobal = "pso_inf_global";
+constexpr auto kArgPsoInfLocal = "pso_inf_local";
+constexpr auto kArgPsoInfRandom = "pso_inf_random";
+// Annealing tuner-specific arguments in string form
+constexpr auto kArgAnnMaxTemp = "ann_max_temperature";
 
 // The common arguments in string form
 constexpr auto kArgPlatform = "platform";
@@ -172,7 +180,13 @@ struct Arguments {
   size_t ap_size = 1;
   size_t scalar_size = 1;
   // Tuner-specific arguments
+  size_t heuristic_selection = 0;
   double fraction = 1.0;
+  size_t pso_swarm_size = 8; 
+  double pso_inf_global = 0.3;
+  double pso_inf_local = 0.6;
+  double pso_inf_random = 0.1;
+  double ann_max_temperature = 1.0; // Is it a valid default value? 
   // Client-specific arguments
   int compare_clblas = 1;
   int compare_cblas = 1;