rapidsai / cugraph-gnn

Apache License 2.0
9 stars 15 forks source link

[BUG] Fix Import Order #49

Closed alexbarghi-nv closed 2 months ago

alexbarghi-nv commented 2 months ago

For the multi-GPU examples, we set RAPIDS_NO_INITIALIZE to prevent RAPIDS from creating a CUDA context. However, this had no effect because there were some RAPIDS imports before that. This PR resolves that problem by moving those imports elsewhere.