code: def test_decompose_hw_padding_direct():
      """Case 0. direct decompose"""
  
      @T.prim_func
      def pooling_decompose_0(
          x: T.Buffer((1, 16, 225, 225), "int8"),
          tensor: T.Buffer((1, 16, 225, 225), "int8"),
      ):
          pad_temp = T.alloc_buffer([1, 16, 231, 231], dtype="int8")
          for i0, i1, i2, i3 in T.grid(1, 16, 231, 231):
              with T.block("pad_temp_pad_const"):
                  ax0, ax1, ax2, ax3 = T.axis.remap(
                      "SSSS", [i0, i1, i2, i3]
                  )
                  pad_temp[ax0, ax1, ax2, ax3] = T.int8(0)
          for i0, i1, i2, i3 in T.grid(1, 16, 225, 225):
              with T.block("pad_temp"):
                  ax0, ax1, ax2, ax3 = T.axis.remap(
                      "SSSS", [i0, i1, i2, i3]
                  )
                  pad_temp[ax0, ax1, ax2 + 3, ax3 + 3] = x[
                      ax0, ax1, ax2, ax3
                  ]
          for i0, i1, i2, i3, i4, i5 in T.grid(1, 16, 225, 225, 7, 7):
              with T.block("tensor"):
                  ax0, ax1, ax2, ax3, rv0, rv1 = T.axis.remap(
                      "SSSSRR", [i0, i1, i2, i3, i4, i5]
                  )
                  with T.init():
                      tensor[ax0, ax1, ax2, ax3] = T.int8(0)
                  tensor[ax0, ax1, ax2, ax3] = (
                      tensor[ax0, ax1, ax2, ax3]
                      + pad_temp[ax0, ax1, ax2 + rv0, ax3 + rv1]
                  )
  
      sch = tir.Schedule(sum_pool_2d, debug_mask="all")
      pad = sch.get_block("pad_temp")
      sch.decompose_padding(pad, sch.get_loops(pad)[0])
      check_decompose_padding(
          sum_pool_2d,
          sch.mod["main"],
          pooling_decompose_0,
          check_run=True,
      )
  
file_path: tvm/tests/python/tir-schedule/test_tir_schedule_decompose_padding.py
imports: import numpy
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.tir.schedule.testing import assert_structural_equal_ignore_global_symbol
  from tvm.script import tir
label_desc: Use image node_id 5 for calling a global function with example usage: test_decompose_hw_padding_direct() without return types