code: def test_unroll_allocations():
      @tvm.script.ir_module
      class before:
          @T.prim_func
          def main():
              for i in T.unroll(2):
                  with T.decl_buffer([16], "float32") as buf:
                      buf[0] = 0.0
  
      @tvm.script.ir_module
      class expected:
          @T.prim_func
          def main():
              with T.decl_buffer([16], "float32") as buf1:
                  buf1[0] = 0.0
              with T.decl_buffer([16], "float32") as buf2:
                  buf2[0] = 0.0
  
      after = tvm.tir.transform.UnrollLoop()(before)
  
      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 4 for calling a global function with example usage: test_unroll_allocations() without return types