Skip to content

GitLab

Commit fd818cf1 authored by oahzxl's avatar oahzxl
Browse files

change imports

parent a591d45b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
colossalai/autochunk/autochunk_codegen.py
View file @ fd818cf1
...@@ -18,7 +18,8 @@ if CODEGEN_AVAILABLE: ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment