code: def mixed_dtype_reindex_write(
      p0: T.Buffer((T.int64(2), 1280), "float16"),
      p1: T.Buffer((1280, 1280), "float16"),
      T_matmul_NT: T.Buffer((T.int64(2), 1280), "float16"),
  ) -> None:
      T_matmul_NT_reindex = T.alloc_buffer(
          [T.int64(2), 1280], dtype="float16"
      )
      for i0, i1, i2 in T.grid(T.int64(2), 1280, 1280):
          with T.block("T_matmul_NT"):
              i = T.axis.spatial(T.int64(2), i0)
              j, k = T.axis.remap("SR", [i1, i2])
              T.reads(p0[i, k], p1[j, k])
              T.writes(T_matmul_NT_reindex[i, j])
              with T.init():
                  T_matmul_NT_reindex[i, j] = T.float16(0)
              T_matmul_NT_reindex[i, j] = (
                  T_matmul_NT_reindex[i, j] + p0[i, k] * p1[j, k]
              )
      for ax0, ax1 in T.grid(T.int64(2), 1280):
          with T.block("T_matmul_NT_reindex"):
              v0 = T.axis.spatial(T.int64(2), ax0)
              v1 = T.axis.remap("S", [ax1])
              T.reads(T_matmul_NT_reindex[v0, v1])
              T.writes(T_matmul_NT[v0, v1])
              T_matmul_NT[v0, v1] = T_matmul_NT_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 10 for calling a global function with example usage: mixed_dtype_reindex_write(p0, p1, T_matmul_NT) without return types