code: def matmul_after_reorder_block_iter_var(
      A: T.Buffer((128, 128), "float32"),
      B: T.Buffer((128, 128), "float32"),
      C: T.Buffer((128, 128), "float32"),
  ):
      for i, j, k in T.grid(128, 128, 128):
          with T.block("C"):
              vk, vj, vi = T.axis.remap("RSS", [k, j, i])
              T.reads(A[vi, vk], B[vj, vk])
              T.writes(C[vi, vj])
              with T.init():
                  C[vi, vj] = T.float32(0)
              C[vi, vj] = C[vi, vj] + A[vi, vk] * B[vj, vk]
  
file_path: tvm/tests/python/tir-schedule/test_tir_schedule_reorder_block_iter_var.py
imports: import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.testing import verify_trace_roundtrip
label_desc: Use image node_id 2 for calling a global function with example usage: matmul_after_reorder_block_iter_var(A, B, C) without return types