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

change imports

Browse Source
This commit is contained in:
oahzxl 2023-01-10 11:10:45 +08:00
parent a591d45b29
commit fd818cf144
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/autochunk/autochunk_codegen.py
View File

@ -18,6 +18,7 @@ if CODEGEN_AVAILABLE:
inplace_methods,
magic_methods,
)
from torch.fx.node import Argument, Node, _get_qualified_name, _type_repr, map_arg
from .search_chunk import SearchChunk