code: def test_unroll_single_count_loops():
      n = te.size_var("n")
      A = te.placeholder((n,), name="A")
      B = te.compute((n,), lambda *i: A(*i), name="B")
      s = te.create_schedule(B.op)
      s = s.normalize()
      dom_map = tvm.te.schedule.InferBound(s)
      stmt = tvm.te.schedule.ScheduleOps(s, dom_map)
      # all parameters to UnrolLoops are default values except for
      # auto_unroll_max_extent which has been set to 1 (default:0)
      mod = tvm.IRModule.from_expr(tvm.tir.PrimFunc([], stmt))
  
      with tvm.transform.PassContext(
          config={"tir.UnrollLoop": {"auto_max_step": 1}}
      ):
          ret = tvm.tir.transform.UnrollLoop()(mod)["main"].body
          assert ret == stmt
  
file_path: tvm/tests/python/tir-transform/test_tir_transform_unroll_loop.py
imports: import tvm
  from tvm import te
  from tvm.script import tir
  import os
label_desc: Use image node_id 3 for calling a global function with example usage: test_unroll_single_count_loops() without return types