[RESEARCH] Investigate Renumbering as Part of Uniform Neighbor Sample Call #3668
Closed
3 tasks done
Labels
feature request
New feature or request
It should be possible in our C++ sampling code to renumber the final edgelist to conform with PyG numbering requirements (vertices are numbered in order of appearance). This would be done in parallel over all GPUs, yielding a significant performance boost over creating the MFGs during the loading process.
The only catch is that
uniform_neighbor_sample
would need to also output the original vertex ids so that the features can be retrieved during the loading process.This would remove the need for a separate
create_mfg
C++ function to be called at loading time.Tasks
The text was updated successfully, but these errors were encountered: