code: def test_rsp_push_pull_large_rowid():
      num_rows = 793470
      val = (
          mx.nd.ones((num_rows, 1))
          .tostype("row_sparse")
          .copyto(mx.gpu())
      )
      kv = mx.kv.create("device")
      kv.init("a", val)
      out = mx.nd.zeros((num_rows, 1), stype="row_sparse").copyto(
          mx.gpu()
      )
      kv.push("a", val)
      kv.row_sparse_pull(
          "a", out=out, row_ids=mx.nd.arange(0, num_rows, dtype="int64")
      )
      assert out.indices.shape[0] == num_rows
  
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 4 for calling a global function with example usage: test_rsp_push_pull_large_rowid() without return types