github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-05 19:13:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Jiarui Fang
2022-05-18 14:54:51 +08:00
committed by GitHub
parent 571f12eff3
commit 802ac297cc
4 changed files with 3 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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