Unverified Commit 26ba8727 authored by YuliangLiu0306's avatar YuliangLiu0306 Committed by GitHub
Browse files

[hotfix]fixed p2p process send stuck (#1181)

* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4500af6a9220ef7fe4d3c7b1daebd4c.

* [hotfix]fixed p2p process send stuck
parent 1b657f9c
...@@ -63,6 +63,8 @@ def process_object_to_send(object_send, scatter_gather_tensors): ...@@ -63,6 +63,8 @@ def process_object_to_send(object_send, scatter_gather_tensors):
send_split = _get_tensor_shape(tensor_send.shape, scatter_gather_tensors)[1] send_split = _get_tensor_shape(tensor_send.shape, scatter_gather_tensors)[1]
if send_split: if send_split:
object_send_list.append(split_tensor_into_1d_equal_chunks(tensor_send)) object_send_list.append(split_tensor_into_1d_equal_chunks(tensor_send))
else:
object_send_list.append(tensor_send)
object_send = tuple(object_send_list) object_send = tuple(object_send_list)
return object_send return object_send
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment