Skip to content

GitLab

Unverified Commit 28e2d167 authored by yuxuan-lou's avatar yuxuan-lou Committed by GitHub
Browse files

[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/graph_analysis.py code style (#2340)

parent bd12a49e
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
colossalai/auto_parallel/tensor_shard/deprecated/graph_analysis.py
View file @ 28e2d167
from collections import OrderedDict as ODict
from dataclasses import dataclass from dataclasses import dataclass
from torch.fx.node import Node from typing import Any, List, OrderedDict, Union
from torch.fx.graph import Graph from torch.fx.graph import Graph
from torch.fx.graph_module import GraphModule from torch.fx.graph_module import GraphModule
from collections import OrderedDict as ODict from torch.fx.node import Node
from typing import List, OrderedDict, Union, Any
from colossalai.fx.passes.utils import get_node_module from colossalai.fx.passes.utils import get_node_module
__all__ = ['LiveVariable', 'LiveVariableVector', 'LiveStage', 'GraphAnalyser'] __all__ = ['LiveVariable', 'LiveVariableVector', 'LiveStage', 'GraphAnalyser']
......
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