github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-22 09:49:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
yuxuan-lou 2023-01-05 16:53:24 +08:00 committed by GitHub
parent bd12a49e2a
commit 28e2d16794
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
colossalai/auto_parallel/tensor_shard/deprecated/graph_analysis.py
View File

@ -1,9 +1,11 @@
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']