code: def test_unroll_local_access():
      @tvm.script.ir_module
      class Before:
          @T.prim_func
          def main(B: T.Buffer((64,), "float32")):
              for bx in T.thread_binding(4, thread="blockIdx.x"):
                  for tx in T.thread_binding(4, thread="threadIdx.x"):
                      A_local_data = T.allocate(
                          [4], dtype="float32", scope="local"
                      )
                      A_local = T.Buffer(
                          [4], dtype="float32", data=A_local_data
                      )
                      for i in T.serial(4):
                          A_local[i] = T.float32(i)
  
      @tvm.script.ir_module
      class Expected:
          @T.prim_func
          def main(B: T.Buffer((64,), "float32")):
              for bx in T.thread_binding(4, thread="blockIdx.x"):
                  for tx in T.thread_binding(4, thread="threadIdx.x"):
                      A_local_data = T.allocate(
                          [4], dtype="float32", scope="local"
                      )
                      A_local = T.Buffer(
                          [4], dtype="float32", data=A_local_data
                      )
                      A_local[0] = T.float32(0)
                      A_local[1] = T.float32(1)
                      A_local[2] = T.float32(2)
                      A_local[3] = T.float32(3)
  
      with tvm.transform.PassContext(
          config={
              "tir.UnrollLoop": {
                  "auto_max_depth": 0,
                  "auto_max_extent": 1,
                  "explicit_unroll": True,
                  "unroll_local_access": True,
              }
          }
      ):
          after = tvm.tir.transform.UnrollLoop()(Before)
          after = tvm.tir.transform.Simplify()(after)
  
      tvm.ir.assert_structural_equal(after, Expected)
  
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 5 for calling a global function with example usage: test_unroll_local_access() without return types