code: def test_gemm(device="vta", batch=128, in_feat=128, out_feat=128):
      def _run(env, remote):
          if device == "vta":
              target = env.target
              if env.TARGET not in ["sim", "tsim"]:
                  assert tvm.runtime.enabled("rpc")
                  program_fpga(remote, bitstream=None)
                  reconfig_runtime(remote)
          elif device == "arm_cpu":
              target = env.target_vta_cpu
          with autotvm.tophub.context(
              target
          ):  # load pre-tuned schedule parameters
              run_gemm(env, remote, target, batch, in_feat, out_feat)
  
      vta.testing.run(_run)
  
file_path: tvm/vta/tests/python/integration/test_benchmark_topi_dense.py
imports: import os
  import json
  from collections import namedtuple
  import numpy
  import tvm
  from tvm import te
  from tvm import autotvm
  from tvm.contrib import utils
  from tvm.contrib.pickle_memoize import memoize
  from tvm import topi
  import tvm.topi.testing
  import vta
  from vta import program_fpga, reconfig_runtime
  import vta.testing
  from vta.testing import simulator
label_desc: Use image node_id 3 for calling a global function with example usage: test_gemm(device, batch, in_feat, out_feat) without return types