code: def test_ptx_cp_async_barrier():
      f = ptx_cp_async_barrier
  
      mod = tvm.build(f, target="cuda")
      A_np = np.random.rand(32, 128).astype("float16")
      B_np = np.zeros((32, 128)).astype("float16")
      dev = tvm.cuda(0)
      A_nd = tvm.nd.array(A_np, device=dev)
      B_nd = tvm.nd.array(B_np, device=dev)
      mod(A_nd, B_nd)
      tvm.testing.assert_allclose(B_nd.numpy(), A_np)
  
file_path: tvm/tests/python/tir-base/test_tir_ptx_cp_async.py
imports: import tvm
  from tvm.script import tir
  import numpy
  import tvm.testing
label_desc: Use image node_id 4 for calling a global function with example usage: test_ptx_cp_async_barrier() without return types