Merge pull request #121 from tqchen/master

[KVStore] BUGFIX GPU merge
apache · Sep 22, 2015 · 522aa64 · 522aa64
2 parents c0b0093 + d857997
commit 522aa64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/kvstore/kvstore_local.h b/src/kvstore/kvstore_local.h
@@ -119,7 +119,7 @@ class KVStoreLocal : public KVStore {
       } else {
         CHECK_EQ(ctx.dev_mask(), gpu::kDevMask);
         NDArray *copy_buf = buf.AllocCopyBuf(ctx.dev_id, val[0].shape());
-        CopyFromTo(val[0], copy_buf);
+        CopyFromTo(val[i], copy_buf);
         buf.merged += *copy_buf;
       }
     }