minor change

This commit is contained in:
tqchen 2015-01-13 20:01:15 -08:00
parent 532575b752
commit 478d250818
3 changed files with 1 additions and 3 deletions

View File

@ -116,7 +116,7 @@ int main(int argc, char *argv[]) {
// at this point, the model object should allow us to recover the program state
...
// each iteration can contain multiple calls of allreduce/broadcast
rabit::Allreduce(&data[0], n);
rabit::Allreduce<rabit::op::Max>(&data[0], n);
...
// checkpoint model after one iteration finishes
rabit::CheckPoint(&model);

View File

@ -21,4 +21,3 @@ print '@node[%d] before-allreduce: a=%s' % (rank, str(a))
a = rabit.allreduce(a, rabit.MAX)
print '@node[%d] after-allreduce: a=%s' % (rank, str(a))
rabit.finalize()

View File

@ -20,4 +20,3 @@ s = rabit.broadcast(s, 0)
print '@node[%d] after-broadcast: s=\"%s\"' % (rank, str(s))
rabit.finalize()