code: def main(a: T.handle, b: T.handle) -> None:
      # function attr dict
      T.func_attr({"global_symbol": "main"})
      A = T.match_buffer(a, [2048, 2048, 2048], dtype="float32")
      B = T.match_buffer(b, [2048, 2048, 2048], dtype="float32")
      A_cached = T.alloc_buffer([2048, 2048, 2048], dtype="float32")
      # body
      for i, j, k in T.grid(2048, 2048, 2048):
          with T.block("move"):
              vi, vj, vk = T.axis.remap("SSS", [i, j, k])
              T.reads([A[vi, vj, vk]])
              T.writes([A_cached[vi, vj, vk]])
              A_cached[vi, vj, vk] = A[vi, vj, vk]
      for i0, j0, i1, j1, k0, i2, j2, k1 in T.grid(
          128, 64, 4, 4, 64, 4, 8, 32
      ):
          with T.block("add"):
              vi = T.axis.spatial(2048, i0 * 16 + i1 * 4 + i2)
              vj = T.axis.spatial(2048, j0 * 32 + j1 * 8 + j2)
              vk = T.axis.spatial(2048, k0 * 32 + k1)
              T.reads([A_cached[vi, vj, vk]])
              T.writes([B[vi, vj, vk]])
              B[vi, vj, vk] = A_cached[vi, vj, vk] + T.float32(1)
  
file_path: tvm/tests/python/meta_schedule/test_meta_schedule_schedule_rule_random_compute_location.py
imports: import tvm
  from tvm import meta_schedule
  from tvm.meta_schedule.testing.space_generation import check_sketches, generate_design_space
  from tvm.script import tir
  from tvm.target import Target
label_desc: Use image node_id 1 for calling the Add obj's underlying member method code with example usage: obj.main(a, b) without return types