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

[NFC] polish unary_elementwise_generator.py code style (#4267) 

Co-authored-by: aye42 <aye42@gatech.edu>
pull/4338/head
YeAnbang 2023-07-18 11:53:47 +08:00 committed by binmakeswell
parent 798cb72907
commit 3883db452c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/auto_parallel/tensor_shard/node_handler/strategy/unary_elementwise_generator.py
View File

@ -1,7 +1,7 @@
import copy
from typing import List
from colossalai.auto_parallel.tensor_shard.sharding_strategy import (MemoryCost, ShardingStrategy, TrainCycleItem)
from colossalai.auto_parallel.tensor_shard.sharding_strategy import MemoryCost, ShardingStrategy, TrainCycleItem
from .strategy_generator import FollowingStrategyGenerator