code: def different_access_indices(a: T.handle, b: T.handle) -> None:
      A = T.match_buffer(a, [128, 128, 128], dtype="float32")
      B = T.match_buffer(b, [128, 128], dtype="float32")
      for i, j in T.grid(128, 128):
          for k in T.thread_binding(0, 128, thread="threadIdx.x"):
              with T.block("B"):
                  vi, vj, vk = T.axis.remap("SSR", [i, j, k])
                  T.reads([B[vi, vj], A[vi, vj, vk]])
                  T.writes(
                      [
                          B[
                              T.min(vj, vi) : T.min(vj, vi)  # type: ignore[misc]
                              + (T.max(vj, vi) + 1 - T.min(vj, vi)),
                              T.min(vi, vj) : T.min(vi, vj)  # type: ignore[misc]
                              + (T.max(vi, vj) + 1 - T.min(vi, vj)),
                          ]
                      ]
                  )
                  with T.init():
                      B[vj, vi] = T.exp(B[vj, vi], dtype="float32")
                  B[vi, vj] = B[vi, vj] + A[vi, vj, vk]
  
imports: from tvm.script import tir
label_desc: Use image node_id 5 for calling a global function with example usage: different_access_indices(a, b) without return types