code: def test_broadcast():
      ctx = (
          mx.gpu(kv.local_rank)
          if mx.device.num_gpus() > 0
          else mx.cpu(kv.local_rank)
      )
      val = mx.nd.zeros(shape, ctx)
      kv.broadcast("0", mx.nd.ones(shape), out=val)
      expected = 1
      check_diff_to_scalar(val, expected, kv.rank)
      print("worker " + str(kv.local_rank) + " passed test_broadcast")
  
file_path: mxnet/tests/nightly/dist_device_sync_kvstore_horovod.py
imports: import sys
  import mxnet
  import numpy
  import numpy.random
  import time
  import argparse
label_desc: Use image node_id 3 for calling a global function with example usage: test_broadcast() without return types