Skip to content

Commit

Permalink
[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_hand…
Browse files Browse the repository at this point in the history
…ler/where_handler.py code style (hpcaitech#2723)
  • Loading branch information
yuxuan-lou authored Feb 15, 2023
1 parent 4603538 commit 51c45c2
Showing 1 changed file with 6 additions and 4 deletions.
Original file line number Diff line number Diff line change
@@ -6,10 +6,12 @@

import torch

from colossalai.auto_parallel.tensor_shard.deprecated._utils import (enumerate_all_possible_1d_sharding,
enumerate_all_possible_2d_sharding,
ignore_sharding_exception)
from colossalai.auto_parallel.tensor_shard.deprecated.sharding_strategy import (ShardingStrategy, StrategiesVector)
from colossalai.auto_parallel.tensor_shard.deprecated._utils import (
enumerate_all_possible_1d_sharding,
enumerate_all_possible_2d_sharding,
ignore_sharding_exception,
)
from colossalai.auto_parallel.tensor_shard.deprecated.sharding_strategy import ShardingStrategy, StrategiesVector
from colossalai.tensor.shape_consistency import ShapeConsistencyManager
from colossalai.tensor.sharding_spec import ShardingSpec

0 comments on commit 51c45c2

Please sign in to comment.