code: def test_blockize_inner():
      @T.prim_func
      def after_blockize_inner(
          A: T.Buffer((128, 128), "float32"),
          B: T.Buffer((128, 128), "float32"),
      ) -> None:
          for i in T.serial(128):
              with T.block("blockized_B"):
                  vi = T.axis.spatial(128, i)
                  vjo = T.axis.spatial(1, 0)
                  for j in T.serial(128):
                      with T.block("B"):
                          vj = T.axis.remap("S", [j])
                          B[vi, vj] = A[vi, vj] * 2.0
  
      func = single_elementwise
      s = tir.Schedule(func, debug_mask="all")
      _, y = s.get_loops(s.get_block("B"))
      s.blockize(y)
      tvm.ir.assert_structural_equal(
          s.mod["main"],
          after_blockize_inner.with_attr(
              "global_symbol", "single_elementwise"
          ),
      )
      verify_trace_roundtrip(sch=s, mod=func)
  
file_path: tvm/tests/python/tir-schedule/test_tir_schedule_blockize.py
imports: import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.testing import verify_trace_roundtrip
  import pytest
label_desc: Use image node_id 3 for calling a global function with example usage: test_blockize_inner() without return types