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

change imports

pull/2364/head
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