From b9d35d0988f9f705e5f1d21da42ec2bee685a376 Mon Sep 17 00:00:00 2001 From: redbopo Date: Sun, 12 Dec 2021 05:10:29 +0800 Subject: [PATCH] [FIX] Remove the repeat setting in variable --- python/tvm/_ffi/registry.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/python/tvm/_ffi/registry.py b/python/tvm/_ffi/registry.py index 677ca5d8de8d..7b174cbb0307 100644 --- a/python/tvm/_ffi/registry.py +++ b/python/tvm/_ffi/registry.py @@ -296,15 +296,13 @@ def _init_api(namespace, target_module_name=None): def _init_api_prefix(module_name, prefix): - module = sys.modules[module_name] + target_module = sys.modules[module_name] for name in list_global_func_names(): if not name.startswith(prefix): continue fname = name[len(prefix) + 1 :] - target_module = module - if fname.find(".") != -1: continue f = get_global_func(name)