gallo/atrip
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add rankround robin in bench

Browse Source
This commit is contained in:
Alejandro Gallo 2021-11-08 11:49:37 +01:00
parent cc4029a3f9
commit 9ae055cb40
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bench/test_main.cxx
View File

@ -21,7 +21,7 @@ int main(int argc, char** argv) {
MPI_Init(&argc, &argv); MPI_Init(&argc, &argv);
int no(10), nv(10), itMod(100); int no(10), nv(10), itMod(100);
bool nochrono(false), barrier(false); bool nochrono(false), barrier(false), rankRoundRobin(false);
std::string tuplesDistributionString = "naive"; std::string tuplesDistributionString = "naive";
CLI::App app{"Main bench for atrip"}; CLI::App app{"Main bench for atrip"};
@ -29,6 +29,7 @@ int main(int argc, char** argv) {
app.add_option("--nv", nv, "Virtual orbitals"); app.add_option("--nv", nv, "Virtual orbitals");
app.add_option("--mod", itMod, "Iteration modifier"); app.add_option("--mod", itMod, "Iteration modifier");
app.add_flag("--nochrono", nochrono, "Do not print chrono"); app.add_flag("--nochrono", nochrono, "Do not print chrono");
app.add_flag("--rank-round-robin", rankRoundRobin, "Do rank round robin");
app.add_flag("--barrier", barrier, "Use the first barrier"); app.add_flag("--barrier", barrier, "Use the first barrier");
app.add_option("--dist", tuplesDistributionString, "Which distribution"); app.add_option("--dist", tuplesDistributionString, "Which distribution");
@ -94,6 +95,7 @@ int main(int argc, char** argv) {
// some options // some options
.with_barrier(barrier) .with_barrier(barrier)
.with_chrono(!nochrono) .with_chrono(!nochrono)
.with_rankRoundRobin(rankRoundRobin)
.with_iterationMod(itMod) .with_iterationMod(itMod)
.with_tuplesDistribution(tuplesDistribution) .with_tuplesDistribution(tuplesDistribution)
; ;