Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "dfc88b85ea3cecfe7e12e64c091fc94fe39e81a8"
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
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 5 deletions
colossalai/auto_parallel/tensor_shard/deprecated/op_handler/operator_handler.py
View file @ 72c94489
from abc import ABC, abstractmethod
from typing import Dict, List
from webbrowser import Opera
import torch
import torch.nn as nn
from abc import ABC, abstractmethod
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.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
__all__ = ['OperatorHandler']
......
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