def METHOD_NAME():
    func = element_wise
    s = tir.Schedule(func, debug_mask='all')
    B = s.get_block("B")
    s.storage_align(B, 0, axis=0, factor=128, offset=0)
    s.storage_align(B, 0, axis=0, factor=128, offset=127)
    assert_structural_equal_ignore_global_symbol(element_wise_storage_align, s.mod["main"])
    verify_trace_roundtrip(sch=s, mod=func)