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

[tensor] sharded global process group (#1219)

Browse Source
This commit is contained in:
Jiarui Fang
2022-07-07 13:38:48 +08:00
committed by GitHub
parent db1bef9032
commit 15d988f954
2 changed files with 42 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_tensor/test_tensor.py
View File

@@ -103,9 +103,9 @@ def run_dist_tests(rank, world_size, port):
_run_view(world_size)
_run_process_group(world_size)
_run_tensor_indexing()
_run_operand()
# TODO not passed
# _run_wrapped_tensor_func()
_run_operand()
@pytest.mark.dist