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 import torch
from typing import Union from typing import Union
from colossalai.tensor.op_wrapper import colo_op_impl 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.parallel_1d._utils import reduce_input, reduce_grad
from colossalai.nn.layer.utils import divide
from colossalai.core import global_context as gpc
from colossalai.tensor import ComputePattern, TensorSpec, ComputePattern, ParallelAction, ColoTensor from colossalai.tensor import ComputePattern, TensorSpec, ComputePattern, ParallelAction, ColoTensor
from colossalai.tensor.graph import GraphOpNode, GraphGlobalEnv
from colossalai.tensor import dist_spec from colossalai.tensor import dist_spec

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

@ -1,11 +1,7 @@
import torch import torch
from colossalai.tensor.op_wrapper import colo_op_impl from colossalai.tensor.op_wrapper import colo_op_impl
from colossalai.context import ParallelMode from colossalai.nn.layer.parallel_1d._utils import reduce_input
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.core import global_context as gpc from colossalai.core import global_context as gpc
from packaging import version
from colossalai.tensor import ComputePattern, TensorSpec, ComputePattern, ParallelAction, ColoTensor, dist_spec 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 import torch
from typing import Tuple, Optional, Callable, Union from typing import Tuple, Optional, Callable, Union
from numpy import product from numpy import product
from colossalai.core import global_context as gpc from colossalai.tensor import TensorSpec
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 .const import TensorType from .const import TensorType
from colossalai.tensor import dist_spec from colossalai.tensor import dist_spec
from colossalai.tensor.dist_spec_mgr import DistSpecManager 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.tensor.dist_spec import _DistSpec
from colossalai.nn.layer.utils import divide from colossalai.nn.layer.utils import divide
from numpy import prod from numpy import prod