code: def matmul(a: T.handle, b: T.handle, c: T.handle) -> None:
      A = T.match_buffer(a, [128, 128])
      B = T.match_buffer(b, [128, 128])
      C = T.match_buffer(c, [128, 128])
      for i, j in T.grid(128, 128):
          with T.block("init"):
              vi, vj = T.axis.remap("SS", [i, j])
              C[vi, vj] = 0.0
          for k in range(0, 128):
              with T.block("update"):
                  vi, vj, vk = T.axis.remap("SSR", [i, j, k])
                  C[vi, vj] = C[vi, vj] + A[vi, vk] * B[vj, vk]
  
imports: import sys
  import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.state import CachedFlags
  from tvm.tir.stmt_functor import post_order_visit
label_desc: Use image node_id 2 for calling a global function with example usage: matmul(a, b, c) without return types