From ea1a95b8b9240982325bfd4f44b5afc25294525b Mon Sep 17 00:00:00 2001 From: Super Daniel <78588128+super-dainiu@users.noreply.github.com> Date: Tue, 30 Aug 2022 11:39:03 +0800 Subject: [PATCH] [hotfix] fix coloproxy typos. (#1519) --- colossalai/fx/proxy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/colossalai/fx/proxy.py b/colossalai/fx/proxy.py index c016b4db7..06272c48f 100644 --- a/colossalai/fx/proxy.py +++ b/colossalai/fx/proxy.py @@ -117,8 +117,8 @@ class ColoAttribute(ColoProxy): method = getattr(meta_args[0].__class__, self.attr) if meta_patched_function.has(method): meta_target = meta_patched_function.get(method) - elif meta_patched_function.has(target.__name__): - meta_target = meta_patched_function.get(target.__name__) + elif meta_patched_function.has(method.__name__): + meta_target = meta_patched_function.get(method.__name__) else: meta_target = method meta_out = meta_target(*meta_args, **meta_kwargs)