Skip to content

GitLab

"docs/source/vscode:/vscode.git/clone" did not exist on "6a03c933a0ef43090c8add9303898287b1482d74"
Commit fd818cf1 authored by oahzxl's avatar oahzxl
Browse files

change imports

parent a591d45b
Show 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:
inplace_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 .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