code: def loop_split(a: T.handle, b: T.handle) -> None:
      A = T.match_buffer(a, [128, 128], dtype="float32")
      B = T.match_buffer(b, [128], dtype="float32")
      for i, ko in T.grid(128, 4):
          for ki in T.thread_binding(0, 32, thread="threadIdx.x"):
              with T.block("B"):
                  vi = T.axis.S(128, i)
                  vk = T.axis.R(128, ko * 32 + ki)
                  T.reads([B[vi], A[vi, vk]])
                  T.writes([B[vi]])
                  with T.init():
                      B[vi] = T.float32(0)
                  B[vi] = B[vi] + A[vi, vk]
  
imports: from tvm.script import tir
label_desc: Use image node_id 3 for calling a global function with example usage: loop_split(a, b) without return types