code: def tir_matmul(a: T.handle, b: T.handle, c: T.handle) -> None:
      T.func_attr({"global_symbol": "main", "tir.noalias": True})
      A = T.match_buffer(a, (128, 128))
      B = T.match_buffer(b, (128, 128))
      C = T.match_buffer(c, (128, 128))
  
      for i0, j0, k0 in T.grid(128, 128, 128):
          with T.block():
              i, j, k = T.axis.remap("SSR", [i0, j0, k0])
              with T.init():
                  C[i, j] = 0.0
              C[i, j] += A[i, k] * B[j, k]
  
imports: import numpy
  import tvm
  import tvm.testing
  from tvm import te, tir, topi, relay
  from tvm.script import tir
  import pytest
label_desc: Use image node_id 5 for calling a global function with example usage: tir_matmul(a, b, c) without return types