github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish colossalai/fx/tracer/_tracer_utils.py (#3323) 

* [NFC] polish colossalai/engine/schedule/_pipeline_schedule.py code style

* [NFC] polish colossalai/fx/tracer/_tracer_utils.py  code style

---------

Co-authored-by: Qianran Ma <qianranm@luchentech.com>
pull/3327/head
Michelle 2023-03-29 17:53:32 +08:00 committed by GitHub
parent 64350029fe
commit ad285e1656
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
colossalai/fx/tracer/_tracer_utils.py
View File

@ -1,6 +1,8 @@
from typing import List, Union, Any
from ..proxy import ColoProxy, ColoAttribute
from typing import Any, List, Union
import torch
from ..proxy import ColoAttribute, ColoProxy
from .meta_patch import meta_patched_function, meta_patched_module
__all__ = ['is_element_in_list', 'extract_meta']