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

3
colossalai/autochunk/autochunk_codegen.py
View File

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