code: def test_add_unit_loop_above_block():
      @T.prim_func
      def zero_dim(
          A: T.Buffer((), "int32"),
          B: T.Buffer((), "int32"),
          C: T.Buffer((), "int32"),
      ) -> None:
          with T.block("C"):
              vi = T.axis.spatial(1, 0)
              C[()] = A[()] + B[()]
  
      @T.prim_func
      def zero_dim_added(
          A: T.Buffer((), "int32"),
          B: T.Buffer((), "int32"),
          C: T.Buffer((), "int32"),
      ) -> None:
          for u in range(1):
              with T.block("C"):
                  vi = T.axis.spatial(1, 0)
                  C[()] = A[()] + B[()]
  
      sch = tir.Schedule(zero_dim, debug_mask="all")
      block = sch.get_block("C")
      sch.add_unit_loop(block)
      assert_structural_equal_ignore_global_symbol(
          zero_dim_added, sch.mod["main"]
      )
  
imports: import pytest
  import tvm
  import tvm.testing
  from tvm import te, tir
  from tvm.script import tir
  from tvm.tir.expr import IntImm
  from tvm.tir.schedule.testing import assert_structural_equal_ignore_global_symbol, verify_trace_roundtrip
label_desc: Use image node_id 39 for calling a global function with example usage: test_add_unit_loop_above_block() without return types