tvm.register_global_func not exist

hi, when i test test_tir_transform_storage_rewrite.py. the following register_mem will cause the bug:

def register_mem(scope_tb, max_bits):
    # Register mem
    @tvm.register_global_func("tvm.info.mem.%s" % scope_tb)
    def mem_info_inp_buffer():
        return tvm.ir.make_node(
            "target.MemoryInfo",
            unit_bits=16,
            max_simd_bits=32,
            max_num_bits=max_bits,
            head_address=None,
        )
AttributeError: module 'tvm' has no attribute 'register_global_func'. Did you mean: 'get_global_func'?

tvm 0.21, is there any api to replace @tvm.register_global_func()?