From d8579978b2383847a158bdc67bb7597c351a9723 Mon Sep 17 00:00:00 2001 From: tqchen Date: Mon, 21 Sep 2015 18:52:38 -0700 Subject: [PATCH] [KVStore] BUGFIX GPU merge --- src/kvstore/kvstore_local.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/kvstore/kvstore_local.h b/src/kvstore/kvstore_local.h index b08ea9997369..f8cbf53d27b3 100644 --- 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; } }