@@ -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:
|
||||||
|
|||||||
Reference in New Issue
Block a user