cugraph icon indicating copy to clipboard operation
cugraph copied to clipboard

Published 20 hours ago • verified publisher icon rapidsai

[FIX] Shuffle Inputs Globally in cuGraph-PyG

Open alexbarghi-nv opened this issue 6 months ago • 0 comments

Previously, inputs were only being shuffled locally. This PR uses NCCL to shuffle inputs across workers for better accuracy. It also exposes an option to disable global shuffling if desired.

Closes rapidsai/cugraph-gnn#19

alexbarghi-nv avatar Aug 08 '24 18:08 alexbarghi-nv