code: def test_row_sparse_pull_single_device():
      kvstore = mx.kv.create("device")
      copy = mx.nd.random_normal(shape=(4, 4), ctx=mx.gpu(0))
      grad = copy.tostype("row_sparse")
  
      key = 0
      kvstore.init(key, grad)
      idx = grad.indices
      kvstore.push(key, grad)
      kvstore.row_sparse_pull(key, out=grad, row_ids=idx)
  
      assert_almost_equal(grad.asnumpy(), copy.asnumpy())
  
file_path: mxnet/tests/python/gpu/test_kvstore_gpu.py
imports: import sys
  import os
  import mxnet
  import numpy
  import pytest
  from mxnet.test_utils import assert_almost_equal, default_device, environment
label_desc: Use image node_id 3 for calling a global function with example usage: test_row_sparse_pull_single_device() without return types