code: def transpose_elementwise_reindex_read(
      A: T.Buffer((128, 128), "float32"),
      B: T.Buffer((128, 128), "float32"),
  ) -> None:
      A_reindex = T.alloc_buffer((128, 128), "float32")
      for i, j in T.grid(128, 128):
          with T.block("A_reindex"):
              vi, vj = T.axis.remap("SS", [i, j])
              A_reindex[vi, vj] = A[vj, vi]
      for i, j in T.grid(128, 128):
          with T.block("B"):
              vi, vj = T.axis.remap("SS", [i, j])
              B[vi, vj] = A_reindex[vi, vj] * 2.0
  
imports: import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.schedule import ScheduleError
  from tvm.tir.schedule.testing import assert_structural_equal_ignore_global_symbol, verify_trace_roundtrip
label_desc: Use image node_id 2 for calling a global function with example usage: transpose_elementwise_reindex_read(A, B) without return types