Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
3c3714fc
Commit
3c3714fc
authored
Nov 07, 2022
by
binmakeswell
Browse files
[NFC] polish strategies_constructor.py code style (#1806)
parent
3ce4463f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
13 additions
and
10 deletions
+13
-10
colossalai/auto_parallel/tensor_shard/deprecated/strategies_constructor.py
...arallel/tensor_shard/deprecated/strategies_constructor.py
+13
-10
No files found.
colossalai/auto_parallel/tensor_shard/deprecated/strategies_constructor.py
View file @
3c3714fc
import
builtins
import
math
import
operator
from
copy
import
deepcopy
from
typing
import
Dict
,
List
import
torch
from
torch.fx
import
Graph
,
Node
from
colossalai.tensor.sharding_spec
import
ShardingSpec
from
colossalai.device.device_mesh
import
DeviceMesh
from
colossalai.tensor.shape_consistency
import
ShapeConsistencyManager
from
colossalai.tensor.sharding_spec
import
ShardingSpec
from
._utils
import
generate_resharding_costs
,
generate_sharding_spec
from
.constants
import
*
from
.op_handler
import
*
from
.options
import
SolverOptions
from
.sharding_strategy
import
ShardingStrategy
,
StrategiesVector
from
.op_handler
import
*
from
.constants
import
*
from
copy
import
deepcopy
import
math
import
torch
import
operator
from
typing
import
Dict
,
List
from
._utils
import
generate_sharding_spec
,
generate_resharding_costs
import
builtins
__all__
=
[
'StrategiesConstructor'
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment