github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-26 07:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] fix coloproxy typos. (#1519)

Browse Source
This commit is contained in:
Super Daniel 2022-08-30 11:39:03 +08:00 committed by GitHub
parent af5438caa2
commit ea1a95b8b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
colossalai/fx/proxy.py
View File

@ -117,8 +117,8 @@ class ColoAttribute(ColoProxy):
method = getattr(meta_args[0].__class__, self.attr) method = getattr(meta_args[0].__class__, self.attr)
if meta_patched_function.has(method): if meta_patched_function.has(method):
meta_target = meta_patched_function.get(method) meta_target = meta_patched_function.get(method)
elif meta_patched_function.has(target.__name__): elif meta_patched_function.has(method.__name__):
meta_target = meta_patched_function.get(target.__name__) meta_target = meta_patched_function.get(method.__name__)
else: else:
meta_target = method meta_target = method
meta_out = meta_target(*meta_args, **meta_kwargs) meta_out = meta_target(*meta_args, **meta_kwargs)