rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Refactor mpi classes

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-12-20 19:10:17 +01:00
parent f88944de36
commit 1cc19a7b19
8 changed files with 30 additions and 293 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/grid/GridBase.h
View File

@@ -32,12 +32,22 @@ namespace platform {
};
~GridBase() = default;
virtual void go(struct ConfigMPI& config_mpi) = 0;
virtual json build_tasks_mpi() = 0;
protected:
virtual json build_tasks() = 0;
struct ConfigGrid config;
Timer timer; // used to measure the time of the whole process
const std::string separator = "|";
bayesnet::Smoothing_t smooth_type{ bayesnet::Smoothing_t::NONE };
};
class MPI_Base {
public:
static std::string get_color_rank(int rank)
{
auto colors = { Colors::WHITE(), Colors::RED(), Colors::GREEN(), Colors::BLUE(), Colors::MAGENTA(), Colors::CYAN(), Colors::YELLOW(), Colors::BLACK() };
std::string id = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
auto idx = rank % id.size();
return *(colors.begin() + rank % colors.size()) + id[idx];
}
};
} /* namespace platform */
#endif