Update dist_dataloader.py (PaddlePaddle#9451)

DesmonDay · Nov 19, 2024 · b6652b2 · b6652b2
1 parent f403c19
commit b6652b2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/paddlenlp/data/dist_dataloader.py b/paddlenlp/data/dist_dataloader.py
@@ -205,6 +205,7 @@ def init_dataloader_comm_group():
     hcg = fleet.get_hybrid_communicate_group()
     topo = hcg._topo
     parallel_groups = topo.get_comm_list("pipe")
+    parallel_comm_group = None
 
     for group in parallel_groups:
         ranks = [group[0], group[-1]]