code: def _get_profile_runtime(mod):
      func = mod["main"]
      func = _quantize.CreateStatsCollector(func)
  
      if tvm.target.Target.current():
          target = tvm.target.Target.current()
          dev = tvm.device(target.kind.name)
      else:
          target = "llvm"
          dev = tvm.device(target)
  
      with tvm.transform.PassContext(opt_level=3):
          lib = _build_module.build(func, target=target)
      runtime = graph_executor.GraphModule(lib["default"](dev))
  
      return runtime
  
imports: from __future__ import absolute_import
  import logging
  import multiprocessing
  import numpy
  import tvm
  import tvm.driver
  from tvm.ir import IRModule
  from .None import _quantize
  from .None import quantize
  from ..None import op
  from ..None import expr
  from ..None import analysis
  from ..None import build_module
  from ...contrib import graph_executor
  from .kl_divergence import _find_scale_by_kl
label_desc: Use image node_id 1 for calling a global function with example usage: _get_profile_runtime(mod) and returns: runtime