diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2017-08-21 20:06:29 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-08-21 20:06:29 +0200 |
commit | e5eb6b1d3a66358093cb40f0fad51ecdc4654771 (patch) | |
tree | e5eb03736240ec07534319fdd15661e1093f04ac /src/utilities/utilities.hpp | |
parent | d67fd6604b4a6584c4f9e856057fcc8076ce377d (diff) | |
parent | dfd332524ab0e66a04d803bb075787e35cd2db1a (diff) |
Merge pull request #173 from mcian/PSO_params
Add PSO parameters support and search strategy selection from command…
Diffstat (limited to 'src/utilities/utilities.hpp')
-rw-r--r-- | src/utilities/utilities.hpp | 14 |
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; |