github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/layer_norm_handler.py code style (#2305)

Browse Source
pull/2317/head
Ofey Chan 2 years ago committed by Frank Lee
parent 116e3d0b8f
commit 87d2defda6
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File

10
colossalai/auto_parallel/tensor_shard/deprecated/op_handler/layer_norm_handler.py
Unescape View File

@ -2,10 +2,14 @@ import operator
from functools import reduce from functools import reduce
import torch import torch
from colossalai.auto_parallel.tensor_shard.deprecated._utils import (enumerate_all_possible_1d_sharding,
from colossalai.auto_parallel.tensor_shard.deprecated._utils import (
enumerate_all_possible_1d_sharding,
enumerate_all_possible_2d_sharding, enumerate_all_possible_2d_sharding,
generate_sharding_size, ignore_sharding_exception) generate_sharding_size,
from colossalai.auto_parallel.tensor_shard.deprecated.sharding_strategy import (ShardingStrategy, StrategiesVector) ignore_sharding_exception,
)
from colossalai.auto_parallel.tensor_shard.deprecated.sharding_strategy import ShardingStrategy, StrategiesVector
from .operator_handler import OperatorHandler from .operator_handler import OperatorHandler

Write Preview
Loading…
Cancel
Save