code: def check_diff(A, x, rank=None):
      """assert A == x
      x can be scalar as well as numpy array
      """
      assert np.sum(np.abs((A - x).asnumpy())) == 0, (
          rank,
          A.asnumpy(),
          x,
      )
  
file_path: mxnet/tests/nightly/dist_sync_kvstore.py
imports: import sys
  import argparse
  import mxnet
  import numpy
  import numpy.random
  from mxnet.test_utils import assert_almost_equal, assert_exception
  from test_kvstore import compute_expected_quantization, compute_1bit, compute_2bit
label_desc: Use image node_id 1 for calling a global function with example usage: check_diff(A, x, rank) without return types