github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-31 16:40:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Tensor] rename parallel_action (#1174)

Browse Source 
* rename parallel_action

* polish
This commit is contained in:
Ziyue Jiang
2022-06-27 10:04:45 +08:00
committed by GitHub
parent e27645376d
commit dd0420909f
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/tensor/tensor_spec.py
View File

@@ -9,7 +9,7 @@ class TensorSpec(object):
The specification of the ColoTensor.
Args:
dist_spec (_DistSpec): descriping the layout among processes.
parallel_action (Optional[ComputeSpec], optional): actions conducted on the tensor after initialization if it's a model data tensor.
compute_spec (Optional[ComputeSpec], optional): actions conducted on the tensor after initialization if it's a model data tensor.
Defaults to None.
"""