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

[NFC] polish colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py code style (#2289)

pull/2317/head
Zirui Zhu 2 years ago committed by Frank Lee
parent
3a02b46447
commit
1c29b173c9
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py

2
colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py
Unescape View File

@ -6,7 +6,7 @@ import torch
from ..sharding_strategy import OperationData, OperationDataType from ..sharding_strategy import OperationData, OperationDataType
from .node_handler import NodeHandler from .node_handler import NodeHandler
from .registry import operator_registry from .registry import operator_registry
from .strategy import (StrategyGenerator, TensorStrategyGenerator, TensorTupleStrategyGenerator) from .strategy import StrategyGenerator, TensorStrategyGenerator, TensorTupleStrategyGenerator
__all__ = ['GetItemHandler'] __all__ = ['GetItemHandler']

Write Preview
Loading…
Cancel
Save