rmontanana/BayesNet
rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 3 Wiki Activity

mpi_grid #14

Merged
rmontanana merged 12 commits from mpi_grid into main 2023-12-18 09:05:56 +00:00
Conversation 0 Commits 12 Files Changed 7 +3 -3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 49b26bd04b - Show all commits

6
src/Platform/GridSearch.cc
View File

@ -116,8 +116,7 @@ namespace platform {
} }
} }
// It's important to shuffle the array so heavy datasets are spread across the Workers // It's important to shuffle the array so heavy datasets are spread across the Workers
std::random_device rd; std::mt19937 g{ 271 }; // Use fixed seed to obtain the same shuffle
std::mt19937 g(rd());
std::shuffle(tasks.begin(), tasks.end(), g); std::shuffle(tasks.begin(), tasks.end(), g);
std::cout << "Tasks size: " << tasks.size() << std::endl; std::cout << "Tasks size: " << tasks.size() << std::endl;
std::cout << "|"; std::cout << "|";
@ -258,6 +257,7 @@ namespace platform {
char* msg; char* msg;
int tasks_size; int tasks_size;
if (config_mpi.rank == config_mpi.manager) { if (config_mpi.rank == config_mpi.manager) {
timer.start();
auto tasks = build_tasks_mpi(); auto tasks = build_tasks_mpi();
auto tasks_str = tasks.dump(); auto tasks_str = tasks.dump();
tasks_size = tasks_str.size(); tasks_size = tasks_str.size();
@ -305,7 +305,7 @@ namespace platform {
MPI_Gather(msg, max_size, MPI_CHAR, total, max_size, MPI_CHAR, config_mpi.manager, MPI_COMM_WORLD); MPI_Gather(msg, max_size, MPI_CHAR, total, max_size, MPI_CHAR, config_mpi.manager, MPI_COMM_WORLD);
delete[] msg; delete[] msg;
if (config_mpi.rank == config_mpi.manager) { if (config_mpi.rank == config_mpi.manager) {
std::cout << "|" << std::endl; std::cout << Colors::RESET() << "|" << std::endl;
json total_results; json total_results;
json best_results; json best_results;
// 3.3 Compile the results from all the workers // 3.3 Compile the results from all the workers