Add test

tests/python/gpu/test_kvstore_gpu.py

@@ -91,6 +91,21 @@ def check_rsp_pull(kv, count, ctxs, is_same_rowid=False, use_slice=False):
     check_rsp_push_pull('device')
     check_rsp_push_pull('device', is_push_cpu=False)
 
+
+def test_kvstore_row_sparse_pull_single_device():
+    kvstore = mx.kv.create('device')
+    grad = mx.nd.random_normal(shape=(10,30),ctx=mx.gpu(0))
+    grad = grad.tostype("row_sparse")
+    copy = grad.copy()
+
+    i = 0
+    kvstore.init(i, grad)
+    kvstore.push(i, grad)
+    kvstore.row_sparse_pull(i, out=grad, row_ids=grad.indices)
+
+    assert_almost_equal(grad.asnumpy(), copy.asnumpy())
+
+
 def test_rsp_push_pull_large_rowid():
     num_rows = 793470
     val = mx.nd.ones((num_rows, 1)).tostype('row_sparse').copyto(mx.gpu())