code: def main() -> None:
      s0 = T.int32()
      s1 = T.int32()
      # body
      with T.block("root"):
          T.block_attr({"warp_execution": True})
          for bx in T.thread_binding(8, thread="blockIdx.x"):
              for by in T.thread_binding(8, thread="blockIdx.y"):
                  for ty in T.thread_binding(8, thread="threadIdx.y"):
                      with T.block():
                          A_shared_dyn = T.alloc_buffer(
                              [128, 128],
                              dtype="float16",
                              strides=[136, 1],
                              scope="shared.dyn",
                          )
                          A_wmma = T.alloc_buffer(
                              [128, 128],
                              dtype="float16",
                              scope="wmma.matrix_a",
                          )
                          with T.block("C_shared"):
                              T.reads(A_shared_dyn[0:128, 0:128])
                              T.writes(A_wmma[0:128, 0:128])
                              T.block_attr({"auto_copy": 1})
                              for ax00, ax10 in T.grid(8, 8):
                                  with T.block("wmma_load"):
                                      T.reads(
                                          A_shared_dyn[
                                              ax00 * 16 : ax00 * 16
                                              + 16,
                                              ax10 * 16 : ax10 * 16
                                              + 16,
                                          ]
                                      )
                                      T.writes(
                                          A_wmma[
                                              ax00 * 16 : ax00 * 16
                                              + 16,
                                              ax10 * 16 : ax10 * 16
                                              + 16,
                                          ]
                                      )
                                      src = T.match_buffer(
                                          A_shared_dyn[
                                              ax00 * 16 : ax00 * 16
                                              + 16,
                                              ax10 * 16 : ax10 * 16
                                              + 16,
                                          ],
                                          [16, 16],
                                          dtype="float16",
                                          strides=[s1, s0],
                                          scope="shared.dyn",
                                          offset_factor=16,
                                      )
                                      tgt = T.match_buffer(
                                          A_wmma[
                                              ax00 * 16 : ax00 * 16
                                              + 16,
                                              ax10 * 16 : ax10 * 16
                                              + 16,
                                          ],
                                          [16, 16],
                                          dtype="float16",
                                          scope="wmma.matrix_a",
                                          offset_factor=16,
                                      )
                                      T.evaluate(
                                          T.tvm_load_matrix_sync(
                                              tgt.data,
                                              16,
                                              16,
                                              16,
                                              tgt.elem_offset // 256
                                              + tgt.elem_offset
                                              % 256
                                              // 16,
                                              T.tvm_access_ptr(
                                                  T.type_annotation(
                                                      dtype="float16"
                                                  ),
                                                  src.data,
                                                  src.elem_offset,
                                                  s1 * 16,
                                                  1,
                                                  dtype="handle",
                                              ),
                                              s1,
                                              "row_major",
                                              dtype="handle",
                                          )
                                      )
  
imports: import tvm
  from tvm import te
  from tvm.script import tir
  import sys
  import pytest
label_desc: Use image node_id 1 for calling the TransformedSharedToWmma obj's underlying member method code with example usage: obj.main() without return types