Skip to content

GitLab

Unverified Commit 2fd528b9 authored by xyupeng's avatar xyupeng Committed by GitHub
Browse files

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

parent c9e3ee38
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 @ 2fd528b9
from collections import OrderedDict as ODict
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_module import GraphModule
from collections import OrderedDict as ODict
from typing import List, OrderedDict, Union, Any
from torch.fx.node import Node
from colossalai.fx.passes.utils import get_node_module
__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