This commit is contained in:
Yineng Zhang
2025-04-13 13:57:36 -07:00
committed by GitHub
parent 5fbafbb8f8
commit 39e411385c

View File

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