Unverified Commit 39e41138 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix #5322 (#5359)

parent 5fbafbb8
...@@ -849,7 +849,6 @@ def broadcast_pyobj( ...@@ -849,7 +849,6 @@ def broadcast_pyobj(
): ):
"""Broadcast inputs from rank=0 to all other ranks with torch.dist backend.""" """Broadcast inputs from rank=0 to all other ranks with torch.dist backend."""
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"[broadcast_pyobj] rank={rank}, device={device}")
if rank == 0: if rank == 0:
if len(data) == 0: if len(data) == 0:
......
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