def METHOD_NAME(use_block_name):
    func = element_wise
    sch = tir.Schedule(func, debug_mask="all")
    sch.unsafe_set_dtype("B" if use_block_name else sch.get_block("B"), 0, "float16")
    assert_structural_equal_ignore_global_symbol(element_wise_set_dtype, sch.mod["main"])
    verify_trace_roundtrip(sch=sch, mod=func)