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

[NFC] policy colossalai/fx/proxy.py code style (#3269)

pull/3313/head
CZYCW 2023-03-28 10:31:02 +08:00 committed by binmakeswell
parent d58fa705b2
commit 4cadb25b96
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/proxy.py
View File

@ -1,7 +1,9 @@
import operator import operator
from typing import Any, List, Union
import torch import torch
from torch.fx.proxy import Proxy, Attribute from torch.fx.proxy import Attribute, Proxy
from typing import List, Union, Any
from colossalai.fx.tracer.meta_patch import meta_patched_function from colossalai.fx.tracer.meta_patch import meta_patched_function
__all__ = ['ColoProxy'] __all__ = ['ColoProxy']