code: def expected_layout_attr_int64(
      A: T.Buffer((T.int64(128), T.int64(128)), "float32"),
      B: T.Buffer((T.int64(128), T.int64(128)), "float32"),
      D: T.Buffer((T.int64(128), T.int64(128)), "float32"),
  ):
      T.func_attr(
          {
              "global_symbol": "main",
              "tir.noalias": True,
              "layout_free_buffers": [1],
          }
      )
      C = T.alloc_buffer([T.int64(128), T.int64(128)], dtype="float32")
      for x, y, k in T.grid(T.int64(128), T.int64(128), T.int64(128)):
          with T.block("C"):
              v_x, v_y, v_k = T.axis.remap("SSR", [x, y, k])
              T.reads(A[v_x, v_k], B[v_y, v_k])
              T.writes(C[v_x, v_y])
              with T.init():
                  C[v_x, v_y] = T.float32(0)
              C[v_x, v_y] = C[v_x, v_y] + A[v_x, v_k] * B[v_y, v_k]
      for x, y in T.grid(T.int64(128), T.int64(128)):
          with T.block("D"):
              T.block_attr({"layout_free_placeholders": [C]})
              v_x, v_y = T.axis.remap("SS", [x, y])
              T.reads(C[v_x, v_y])
              T.writes(D[v_x, v_y])
              D[v_x, v_y] = C[v_x, v_y] + T.float32(1)
  
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 31 for calling a global function with example usage: expected_layout_attr_int64(A, B, D) without return types