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

[Tensor] remove useless import in tensor dir (#997)

pull/999/head
Jiarui Fang 2022-05-18 14:54:51 +08:00 committed by GitHub
parent 571f12eff3
commit 802ac297cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 3 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
colossalai/tensor/_ops/addmm.py
View File

@ -1,11 +1,8 @@
import torch
from typing import Union
from colossalai.tensor.op_wrapper import colo_op_impl
from colossalai.nn.layer.parallel_1d._utils import split_forward_gather_backward, reduce_input, reduce_grad
from colossalai.nn.layer.utils import divide
from colossalai.core import global_context as gpc
from colossalai.nn.layer.parallel_1d._utils import reduce_input, reduce_grad
from colossalai.tensor import ComputePattern, TensorSpec, ComputePattern, ParallelAction, ColoTensor
from colossalai.tensor.graph import GraphOpNode, GraphGlobalEnv
from colossalai.tensor import dist_spec

6
colossalai/tensor/_ops/embedding.py
View File

@ -1,11 +1,7 @@
import torch
from colossalai.tensor.op_wrapper import colo_op_impl
from colossalai.context import ParallelMode
from colossalai.nn.layer.parallel_1d._utils import split_forward_gather_backward, reduce_input, \
gather_forward_split_backward, reduce_grad
from colossalai.nn.layer.utils import divide
from colossalai.nn.layer.parallel_1d._utils import reduce_input
from colossalai.core import global_context as gpc
from packaging import version
from colossalai.tensor import ComputePattern, TensorSpec, ComputePattern, ParallelAction, ColoTensor, dist_spec

5
colossalai/tensor/colo_tensor.py
View File

@ -3,10 +3,7 @@ from copy import copy
import torch
from typing import Tuple, Optional, Callable, Union
from numpy import product
from colossalai.core import global_context as gpc
from colossalai.nn.layer.utils import divide
from colossalai.tensor import TensorSpec, ComputePattern
from colossalai.nn.layer.parallel_1d._utils import split_forward_gather_backward, gather_forward_split_backward
from colossalai.tensor import TensorSpec
from .const import TensorType
from colossalai.tensor import dist_spec
from colossalai.tensor.dist_spec_mgr import DistSpecManager

1
colossalai/tensor/dist_spec_mgr.py
View File

@ -1,4 +1,3 @@
from math import dist
from colossalai.tensor.dist_spec import _DistSpec
from colossalai.nn.layer.utils import divide
from numpy import prod