code: def _check_module_with_numpy(mod, shape=(128, 128, 128)):
      m, n, k = shape
      a = tvm.nd.array(np.random.rand(m, k).astype("float32"))
      b = tvm.nd.array(np.random.rand(n, k).astype("float32"))
      c = tvm.nd.array(np.zeros((m, n), dtype="float32"))
      c_np = np.dot(a.numpy(), b.numpy().transpose())
      mod(a, b, c)
      tvm.testing.assert_allclose(c.numpy(), c_np, rtol=1e-5)
  
file_path: tvm/tests/python/tir-base/test_lower_build.py
imports: import numpy
  import tvm
  from tvm import te
  from tvm.ir.module import IRModule
  from tvm.script import tir
  import tvm.testing
label_desc: Use image node_id 1 for calling a global function with example usage: _check_module_with_numpy(mod, shape) without return types