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/operator_handler.py code style (#1845)

pull/1849/head
HELSON 2022-11-09 10:46:10 +08:00 committed by binmakeswell
parent b25030cc07
commit 72c9448920
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py
View File

@ -1,14 +1,16 @@
from abc import ABC, abstractmethod
from typing import Dict, List
from webbrowser import Opera from webbrowser import Opera
import torch import torch
import torch.nn as nn import torch.nn as nn
from abc import ABC, abstractmethod
from torch.fx.node import Node from torch.fx.node import Node
from typing import Dict, List
from colossalai.auto_parallel.tensor_shard.deprecated.constants import *
from colossalai.device.device_mesh import DeviceMesh from colossalai.device.device_mesh import DeviceMesh
from colossalai.tensor.sharding_spec import ShardingSpec from colossalai.tensor.sharding_spec import ShardingSpec
from .._utils import generate_resharding_costs, generate_sharding_spec
from colossalai.auto_parallel.tensor_shard.deprecated.constants import *
from .._utils import generate_resharding_costs, generate_sharding_spec
from ..sharding_strategy import StrategiesVector from ..sharding_strategy import StrategiesVector
__all__ = ['OperatorHandler'] __all__ = ['OperatorHandler']