Skip to content

GitLab

Commit 3883db45 authored by YeAnbang's avatar YeAnbang Committed by binmakeswell
Browse files

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


Co-authored-by: default avataraye42 <aye42@gatech.edu>
parent 798cb729
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/auto_parallel/tensor_shard/node_handler/strategy/unary_elementwise_generator.py
View file @ 3883db45
import copy import copy
from typing import List 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 from .strategy_generator import FollowingStrategyGenerator
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment