search

rapidsai / cugraph-gnn

Apache License 2.0
9 stars 15 forks source link

[FEA] Support Bulk Feature Fetching in cuGraph-DGL/PyG #38

Closed alexbarghi-nv closed 3 weeks ago

alexbarghi-nv commented 3 months ago

We currently sample in bulk, but do not fetch features in bulk. This does not take full advantage of PCI-e bandwidth and, in some cases, the memory capacity of the GPU. This should be added as an option, and should possibly be the default means of feature fetching if performance is good enough.

Implementing this will require implementation of a new buffer that stores fetched features for a set of bulk samples.

alexbarghi-nv commented 3 weeks ago

This no longer seems to be as relevant after the recent performance improvements in WholeGraph.