code: def matmul_relu_ann2(a: T.handle, b: T.handle, d: T.handle) -> None:
      A = T.match_buffer(a, (1024, 1024))
      B = T.match_buffer(b, (1024, 1024))
      C = T.alloc_buffer((1024, 1024))
      D = T.match_buffer(d, (1024, 1024))
      for i, j, k in T.grid(1024, 1024, 1024):
          with T.block("matmul"):
              vi, vj, vk = T.axis.remap("SSR", [i, j, k])
              with T.init():
                  C[vi, vj] = 0.0
              T.block_attr(
                  {"test1": "aaa", "test4": {"arr": [0, 0], "key": 3}}
              )
              C[vi, vj] = C[vi, vj] + A[vi, vk] * B[vk, vj]
      for i, j in T.grid(1024, 1024):
          with T.block("relu"):
              vi, vj = T.axis.remap("SS", [i, j])
              T.block_attr({"test2": 0.22, "test3": ["aa", 1]})
              D[vi, vj] = T.max(C[vi, vj], 0.0)
  
imports: import sys
  import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.ir import IRModule
  from tvm.script import tir
  from tvm.tir.schedule.testing import assert_structural_equal_ignore_global_symbol, verify_trace_roundtrip
label_desc: Use image node_id 4 for calling a global function with example usage: matmul_relu_ann2(a, b, d) without return types