rapidsai / cugraph-gnn

Apache License 2.0
9 stars 15 forks source link

[IMP] Respect the Device Argument in the cuGraph-DGL Data Loaders #11

Open alexbarghi-nv opened 4 months ago

alexbarghi-nv commented 4 months ago

Currently, the device argument is not respected, with the default device being used instead. This should be changed to move data to the desired device instead, regardless of which device is used to process the graph/samples.