github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-28 04:55:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
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
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']