Squashed 'subtree/rabit/' changes from 28ca7be..d558f6f

d558f6f redefine distributed means
c8efc01 more complicated yarn script

git-subtree-dir: subtree/rabit
git-subtree-split: d558f6f550
This commit is contained in:
tqchen
2015-03-09 14:44:42 -07:00
parent 57b5d7873f
commit 4c060df2f1
7 changed files with 29 additions and 5 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;