Commit 72c94489 authored by HELSON's avatar HELSON Committed by binmakeswell
Browse files

[NFC] polish...

[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py code style (#1845)
parent b25030cc
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']
...@@ -60,7 +62,7 @@ class OperatorHandler(ABC): ...@@ -60,7 +62,7 @@ class OperatorHandler(ABC):
@abstractmethod @abstractmethod
def register_strategy(self) -> StrategiesVector: def register_strategy(self) -> StrategiesVector:
""" """
Register Register
""" """
pass pass
......
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