85 lines
2.4 KiB
C++
85 lines
2.4 KiB
C++
/*!
|
|
* Copyright (c) 2014 by Contributors
|
|
* \file engine.cc
|
|
* \brief this file governs which implementation of engine we are actually using
|
|
* provides an singleton of engine interface
|
|
*
|
|
* \author Tianqi Chen, Ignacio Cano, Tianyi Zhou
|
|
*/
|
|
#define _CRT_SECURE_NO_WARNINGS
|
|
#define _CRT_SECURE_NO_DEPRECATE
|
|
#define NOMINMAX
|
|
|
|
#include "../include/rabit/engine.h"
|
|
#include "./allreduce_base.h"
|
|
#include "./allreduce_robust.h"
|
|
|
|
namespace rabit {
|
|
namespace engine {
|
|
// singleton sync manager
|
|
#ifndef RABIT_USE_BASE
|
|
#ifndef RABIT_USE_MOCK
|
|
AllreduceRobust manager;
|
|
#else
|
|
AllreduceMock manager;
|
|
#endif
|
|
#else
|
|
AllreduceBase manager;
|
|
#endif
|
|
|
|
/*! \brief intiialize the synchronization module */
|
|
void Init(int argc, char *argv[]) {
|
|
for (int i = 1; i < argc; ++i) {
|
|
char name[256], val[256];
|
|
if (sscanf(argv[i], "%[^=]=%s", name, val) == 2) {
|
|
manager.SetParam(name, val);
|
|
}
|
|
}
|
|
manager.Init();
|
|
}
|
|
|
|
/*! \brief finalize syncrhonization module */
|
|
void Finalize(void) {
|
|
manager.Shutdown();
|
|
}
|
|
/*! \brief singleton method to get engine */
|
|
IEngine *GetEngine(void) {
|
|
return &manager;
|
|
}
|
|
// perform in-place allreduce, on sendrecvbuf
|
|
void Allreduce_(void *sendrecvbuf,
|
|
size_t type_nbytes,
|
|
size_t count,
|
|
IEngine::ReduceFunction red,
|
|
mpi::DataType dtype,
|
|
mpi::OpType op,
|
|
IEngine::PreprocFunction prepare_fun,
|
|
void *prepare_arg) {
|
|
GetEngine()->Allreduce(sendrecvbuf, type_nbytes, count,
|
|
red, prepare_fun, prepare_arg);
|
|
}
|
|
|
|
// code for reduce handle
|
|
ReduceHandle::ReduceHandle(void)
|
|
: handle_(NULL), redfunc_(NULL), htype_(NULL) {
|
|
}
|
|
ReduceHandle::~ReduceHandle(void) {}
|
|
|
|
int ReduceHandle::TypeSize(const MPI::Datatype &dtype) {
|
|
return static_cast<int>(dtype.type_size);
|
|
}
|
|
void ReduceHandle::Init(IEngine::ReduceFunction redfunc, size_t type_nbytes) {
|
|
utils::Assert(redfunc_ == NULL, "cannot initialize reduce handle twice");
|
|
redfunc_ = redfunc;
|
|
}
|
|
void ReduceHandle::Allreduce(void *sendrecvbuf,
|
|
size_t type_nbytes, size_t count,
|
|
IEngine::PreprocFunction prepare_fun,
|
|
void *prepare_arg) {
|
|
utils::Assert(redfunc_ != NULL, "must intialize handle to call AllReduce");
|
|
GetEngine()->Allreduce(sendrecvbuf, type_nbytes, count,
|
|
redfunc_, prepare_fun, prepare_arg);
|
|
}
|
|
} // namespace engine
|
|
} // namespace rabit
|