code: def element_wise_storage_align(a: T.handle, c: T.handle) -> None:
      C = T.match_buffer(
          c, [128, 128], elem_offset=0, align=64, offset_factor=1
      )
      A = T.match_buffer(
          a, [128, 128], elem_offset=0, align=64, offset_factor=1
      )
      # body
      with T.block("root"):
          T.reads([])
          T.writes([])
          B = T.alloc_buffer(
              [128, 128], elem_offset=0, align=64, offset_factor=1
          )
          for i0 in T.serial(0, 128):
              for ax1 in T.serial(0, 128):
                  with T.block("B"):
                      vi = T.axis.S(128, i0)
                      vj = T.axis.S(128, ax1)
                      T.reads([A[vi, vj]])
                      T.writes([B[vi, vj]])
                      T.block_attr(
                          {"buffer_dim_align": [[0, 0, 128, 127]]}
                      )
                      B[vi, vj] = A[vi, vj] * T.float32(2)
              for i1 in T.serial(0, 128):
                  with T.block("C"):
                      vi_1, vj_1 = T.axis.remap("SS", [i0, i1])
                      T.reads([B[vi_1, vj_1]])
                      T.writes([C[vi_1, vj_1]])
                      C[vi_1, vj_1] = B[vi_1, vj_1] + T.float32(1)
  
imports: from tvm.script import tir
label_desc: Use image node_id 1 for calling a global function with example usage: element_wise_storage_align(a, c) without return types