code: def elementwise_dependent_loops(a: T.handle, b: T.handle) -> None:
      A = T.match_buffer(a, (128, 128, 128))
      B = T.match_buffer(b, (128, 128, 128))
      for i in T.serial(0, 128):
          for j, k in T.grid(i, 128):
              with T.block("B"):
                  vi = T.axis.S(128, i)
                  vj = T.axis.S(i, j)
                  vk = T.axis.S(128, k)
                  B[vi, vj, vk] = A[vi, vj, vk] * 2.0
  
imports: import pytest
  import tvm
  import tvm.testing
  from tvm import te, tir
  from tvm.script import tir
  from tvm.tir.expr import IntImm
  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: elementwise_dependent_loops(a, b) without return types