From ea940f72b5f1005d768707ecbc79354513deffae Mon Sep 17 00:00:00 2001
From: Muhammed Fatih BALIN <m.f.balin@gmail.com>
Date: Fri, 23 Aug 2024 15:43:12 -0400
Subject: [PATCH] [GraphBolt][CUDA] Destroy dist group at the end of script.
 (#7736)

---
 examples/multigpu/graphbolt/node_classification.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/multigpu/graphbolt/node_classification.py b/examples/multigpu/graphbolt/node_classification.py
index f4d1e40f0d..d074e1fedd 100644
--- a/examples/multigpu/graphbolt/node_classification.py
+++ b/examples/multigpu/graphbolt/node_classification.py
@@ -364,6 +364,7 @@ def run(rank, world_size, args, devices, dataset):
 
     if rank == 0:
         print(f"Test Accuracy {test_acc.item():.4f}")
+    dist.destroy_process_group()
 
 
 def parse_args():