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

[tensor]add assert for colo_tensor 1Drow (#846)

Browse Source
This commit is contained in:
Ziyue Jiang
2022-04-24 14:12:45 +08:00
committed by GitHub
parent 05023ecfee
commit 2a0a427e04
5 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
tests/test_tensor/test_tensor_utils/__init__.py → tests/test_tensor/_utils/__init__.py
View File

0
tests/test_tensor/test_tensor_utils/_util.py → tests/test_tensor/_utils/_util.py
View File

2
tests/test_tensor/test_linear_tp.py
View File

@@ -14,7 +14,7 @@ from colossalai.utils import free_port
from colossalai.core import global_context as gpc
import torch.distributed as dist
from test_tensor_utils import check_equal, replace_parameter_add_grad, broadcast_tensor_chunk
from _utils import check_equal, replace_parameter_add_grad, broadcast_tensor_chunk
def run_linear_tp1d_row_test():
device = get_current_device()