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

[tensor] remove gpc in tensor tests (#1186)

Browse Source
This commit is contained in:
Jiarui Fang
2022-06-29 14:08:40 +08:00
committed by GitHub
parent 372f791444
commit c463f8adf9
4 changed files with 26 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/tensor/__init__.py
View File

@@ -1,3 +1,4 @@
from .process_group import ProcessGroup
from .tensor_spec import TensorSpec
from .compute_spec import ComputeSpec, ComputePattern
from .colo_tensor import ColoTensor
@@ -6,7 +7,6 @@ from .utils import convert_parameter, named_params_with_colotensor
from .dist_spec_mgr import DistSpecManager
from .param_op_hook import ParamOpHook, ParamOpHookManager
from . import distspec
from .process_group import ProcessGroup
__all__ = [
'ColoTensor', 'convert_parameter', 'ComputePattern', 'TensorSpec', 'ComputeSpec', 'named_params_with_colotensor',