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()?