github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 02:26:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Tensor] remove ParallelAction, use ComputeSpec instread (#1166)

Browse Source
This commit is contained in:
Jiarui Fang
2022-06-23 17:34:59 +08:00
committed by GitHub
parent 177c374401
commit f4ef224358
20 changed files with 87 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/tensor/colo_tensor.py
View File

@@ -66,7 +66,7 @@ class ColoTensor(torch.Tensor):
self._tensor_spec = spec
def has_spec(self) -> bool:
return self._tensor_spec.parallel_action is not None
return self._tensor_spec.compute_spec is not None
def is_model_data(self) -> bool:
return self._type == TensorType.MODEL