redefine distributed means

This commit is contained in:
tqchen
2015-03-09 14:43:05 -07:00
parent c8efc01367
commit d558f6f550
6 changed files with 20 additions and 3 deletions

View File

@@ -63,6 +63,10 @@ class AllreduceBase : public IEngine {
if (world_size == -1) return 1;
return world_size;
}
/*! \brief whether is distributed or not */
virtual bool IsDistributed(void) const {
return tracker_uri == "NULL";
}
/*! \brief get rank */
virtual std::string GetHost(void) const {
return host_uri;

View File

@@ -56,6 +56,10 @@ class EmptyEngine : public IEngine {
virtual int GetWorldSize(void) const {
return 1;
}
/*! \brief whether it is distributed */
virtual bool IsDistributed(void) const {
return false;
}
/*! \brief get the host name of current node */
virtual std::string GetHost(void) const {
return std::string("");

View File

@@ -59,6 +59,10 @@ class MPIEngine : public IEngine {
virtual int GetWorldSize(void) const {
return MPI::COMM_WORLD.Get_size();
}
/*! \brief whether it is distributed */
virtual bool IsDistributed(void) const {
return true;
}
/*! \brief get the host name of current node */
virtual std::string GetHost(void) const {
int len;