pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-03 19:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[GraphBolt][CUDA] Destroy dist group at the end of script. (#7736)

Browse Source
This commit is contained in:
Muhammed Fatih BALIN
2024-08-23 15:43:12 -04:00
committed by GitHub
parent 9514e7b9cd
commit ea940f72b5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/multigpu/graphbolt/node_classification.py
View File

@@ -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():