Repository: incubator-singa Updated Branches: refs/heads/master 8ac511c70 -> b0033533d
SINGA-100 Implement layers using CUDNN for GPU training fix a bug in stub.cc Project: http://git-wip-us.apache.org/repos/asf/incubator-singa/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-singa/commit/b0033533 Tree: http://git-wip-us.apache.org/repos/asf/incubator-singa/tree/b0033533 Diff: http://git-wip-us.apache.org/repos/asf/incubator-singa/diff/b0033533 Branch: refs/heads/master Commit: b0033533d32c3eebf08f6cb4817cdc7389b9ebd0 Parents: 8ac511c Author: WANG Sheng <[email protected]> Authored: Thu Dec 17 19:28:39 2015 +0800 Committer: WANG Sheng <[email protected]> Committed: Thu Dec 17 19:28:39 2015 +0800 ---------------------------------------------------------------------- src/stub.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-singa/blob/b0033533/src/stub.cc ---------------------------------------------------------------------- diff --git a/src/stub.cc b/src/stub.cc index 82526de..7c0ec90 100644 --- a/src/stub.cc +++ b/src/stub.cc @@ -207,12 +207,12 @@ void Stub::GenMsgs(int type, int version, ParamEntry* entry, Msg* msg, Msg* new_msg = nullptr; if (type == kPut) { CHECK_GT(entry->num_total, 0); - new_msg = param->GenPutMsg(dst_procs == procs_id, idx); + new_msg = param->GenPutMsg(dst_procs != procs_id, idx); new_msg->AddFormatFrame("i", entry->num_total); } else if (type == kGet) { - new_msg = param->GenGetMsg(dst_procs == procs_id, idx); + new_msg = param->GenGetMsg(dst_procs != procs_id, idx); } else if (type == kUpdate) { - new_msg = param->GenUpdateMsg(dst_procs == procs_id, idx); + new_msg = param->GenUpdateMsg(dst_procs != procs_id, idx); new_msg->AddFormatFrame("i", entry->num_local); } else { LOG(FATAL) << "Wrong type";
