kate/BadFish
1
0
Fork 0
mirror of https://github.com/sockspls/badfish synced 2025-04-30 16:53:09 +00:00
Code Issues Releases Wiki Activity

[cluster] avoid creating MPI data type.

Browse source 
there is no need to make an MPI data type for the sendbuffer, simpler and faster.

No functional change
This commit is contained in:
Joost VandeVondele 2018-12-07 18:29:36 +01:00 • committed by Stéphane Nicolet
parent 7af3f4da7a
commit 1cd2c7861a
1 changed files with 6 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
src/cluster.cpp
View file

@ -45,33 +45,12 @@ static MPI_Comm TTComm = MPI_COMM_NULL;
static MPI_Comm MoveComm = MPI_COMM_NULL;
static MPI_Comm StopComm = MPI_COMM_NULL;
static MPI_Datatype TTEntryDatatype = MPI_DATATYPE_NULL;
static std::vector<KeyedTTEntry> TTBuff;
static MPI_Datatype MIDatatype = MPI_DATATYPE_NULL;
void init() {
int thread_support;
constexpr std::array<int, 7> TTblocklens = {1, 1, 1, 1, 1, 1, 1};
const std::array<MPI_Aint, 7> TTdisps = {offsetof(KeyedTTEntry, first),
offsetof(KeyedTTEntry, second) + offsetof(TTEntry, key16),
offsetof(KeyedTTEntry, second) + offsetof(TTEntry, move16),
offsetof(KeyedTTEntry, second) + offsetof(TTEntry, value16),
offsetof(KeyedTTEntry, second) + offsetof(TTEntry, eval16),
offsetof(KeyedTTEntry, second) + offsetof(TTEntry, genBound8),
offsetof(KeyedTTEntry, second) + offsetof(TTEntry, depth8)};
const std::array<MPI_Datatype, 7> TTtypes = {MPI_UINT64_T,
MPI_UINT16_T,
MPI_UINT16_T,
MPI_INT16_T,
MPI_INT16_T,
MPI_UINT8_T,
MPI_INT8_T};
const std::array<MPI_Aint, 4> MIdisps = {offsetof(MoveInfo, move),
offsetof(MoveInfo, depth),
offsetof(MoveInfo, score),
offsetof(MoveInfo, rank)};
MPI_Init_thread(nullptr, nullptr, MPI_THREAD_MULTIPLE, &thread_support);
if (thread_support < MPI_THREAD_MULTIPLE)
{
@ -85,10 +64,10 @@ void init() {
TTBuff.resize(TTSendBufferSize * world_size);
MPI_Type_create_struct(7, TTblocklens.data(), TTdisps.data(), TTtypes.data(),
&TTEntryDatatype);
MPI_Type_commit(&TTEntryDatatype);
const std::array<MPI_Aint, 4> MIdisps = {offsetof(MoveInfo, move),
offsetof(MoveInfo, depth),
offsetof(MoveInfo, score),
offsetof(MoveInfo, rank)};
MPI_Type_create_hindexed_block(4, 1, MIdisps.data(), MPI_INT, &MIDatatype);
MPI_Type_commit(&MIDatatype);
@ -228,8 +207,8 @@ void save(Thread* thread, TTEntry* tte,
}
// Start next communication
MPI_Iallgather(send_buff.data(), send_buff.size(), TTEntryDatatype,
TTBuff.data(), TTSendBufferSize, TTEntryDatatype,
MPI_Iallgather(send_buff.data(), send_buff.size() * sizeof(KeyedTTEntry), MPI_BYTE,
TTBuff.data(), TTSendBufferSize * sizeof(KeyedTTEntry), MPI_BYTE,
TTComm, &req);
}
}