haraldwolff/CLBlast
1
0
Fork 0
mirror of https://github.com/CNugteren/CLBlast.git synced 2024-07-02 12:26:57 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fixed some things in the tuner: bugs, style, and defaults to random search

Browse source
This commit is contained in:
Cedric Nugteren 2017-08-31 20:28:01 +02:00
parent 6e95752054
commit 54e160cd88
3 changed files with 30 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/tuning/kernels/xgemm.cpp
View file

@ -60,7 +60,7 @@ class TuneXgemm {
static double DefaultInfluenceGlobalPSO(){ return 0.1; }
static double DefaultInfluenceLocalPSO(){ return 0.3; }
static double DefaultInfluenceRandomPSO(){ return 0.6; }
static size_t DefaultHeuristic(){ return static_cast<size_t> (cltune::SearchMethod::PSO);}
static size_t DefaultHeuristic(){ return static_cast<size_t>(cltune::SearchMethod::RandomSearch); }
static double DefaultMaxTempAnn(){ return 1.0;}
// Describes how to obtain the sizes of the buffers
@ -180,15 +180,17 @@ class TuneXgemm {
// Returns which Heuristic to run
static size_t GetHeuristic(const Arguments<T> &args){
if (V==1) { return static_cast<size_t>(cltune::SearchMethod::FullSearch); }
else {
// Use full-search to explore all parameter combinations or another strategy to search only a
// part of the parameter values. The fraction is set as a command-line argument.
if (args.fraction == 1.0 || args.fraction == 0.0) {
return static_cast<size_t> (cltune::SearchMethod::FullSearch);
}
else {
return static_cast<size_t>(cltune::SearchMethod::FullSearch);
} else {
return args.heuristic_selection;
}
}
}
};
// =================================================================================================

10
src/tuning/kernels/xgemm_direct.cpp
View file

@ -60,7 +60,7 @@ class TuneXgemmDirect {
static double DefaultInfluenceGlobalPSO(){ return 0.1; }
static double DefaultInfluenceLocalPSO(){ return 0.3; }
static double DefaultInfluenceRandomPSO(){ return 0.6; }
static size_t DefaultHeuristic(){ return static_cast<size_t>(cltune::SearchMethod::PSO);}
static size_t DefaultHeuristic(){ return static_cast<size_t>(cltune::SearchMethod::RandomSearch);}
static double DefaultMaxTempAnn(){ return 1.0;}
// Describes how to obtain the sizes of the buffers
@ -177,15 +177,17 @@ class TuneXgemmDirect {
// Returns which Heuristic to run
static size_t GetHeuristic(const Arguments<T> &args){
if (V==1) { return static_cast<size_t>(cltune::SearchMethod::FullSearch); }
else {
// Use full-search to explore all parameter combinations or another strategy to search only a
// part of the parameter values. The fraction is set as a command-line argument.
if (args.fraction == 1.0 || args.fraction == 0.0) {
return static_cast<size_t> (cltune::SearchMethod::FullSearch);
}
else {
return static_cast<size_t>(cltune::SearchMethod::FullSearch);
} else {
return args.heuristic_selection;
}
}
}
};
// =================================================================================================

4
src/tuning/tuning.hpp
View file

@ -102,9 +102,9 @@ void Tuner(int argc, char* argv[]) {
auto method = C::GetHeuristic(args);
if (method == 1) { tuner.UseRandomSearch(1.0/args.fraction); }
else if (method == 2) { tuner.UseAnnealing(args.fraction, args.ann_max_temperature); }
else if (method == 2) { tuner.UseAnnealing(1.0/args.fraction, args.ann_max_temperature); }
else if (method == 3) {
tuner.UsePSO(args.fraction, args.pso_swarm_size, args.pso_inf_global, args.pso_inf_local, args.pso_inf_random);
tuner.UsePSO(1.0/args.fraction, args.pso_swarm_size, args.pso_inf_global, args.pso_inf_local, args.pso_inf_random);
}
else { tuner.UseFullSearch(); }