code: def matmul_reindex_write(
      A: T.Buffer((512, 512), "float32"),
      B: T.Buffer((512, 512), "float32"),
      C: T.Buffer((512, 512), "float32"),
  ) -> None:
      C_reindex = T.alloc_buffer([512, 512], dtype="float32")
      for i0, i1, i2 in T.grid(512, 512, 512):
          with T.block("matmul"):
              i, j, k = T.axis.remap("SSR", [i0, i1, i2])
              T.reads(C_reindex[i, j], A[i, k], B[k, j])
              T.writes(C_reindex[i, j])
              with T.init():
                  C_reindex[i, j] = T.float32(0)
              C_reindex[i, j] = C_reindex[i, j] + A[i, k] * B[k, j]
      for i0, i1 in T.grid(512, 512):
          with T.block("C_reindex"):
              v0, v1 = T.axis.remap("SS", [i0, i1])
              T.reads(C_reindex[v0, v1])
              T.writes(C[v0, v1])
              C[v0, v1] = C_reindex[v0, v1]
  
imports: import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.schedule import ScheduleError
  from tvm.tir.schedule.testing import assert_structural_equal_ignore_global_symbol, verify_trace_roundtrip
label_desc: Use image node_id 7 for calling a global function with example usage: matmul_reindex_write(A, B, C) without return types