Browse Source

[hotfix] fix coloproxy typos. (#1519)

pull/1520/head^2
Super Daniel 2 years ago committed by GitHub
parent
commit
ea1a95b8b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 4
      colossalai/fx/proxy.py

4
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)

Loading…
Cancel
Save