Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Bugfix] Fix UVA sampling with lazy feature slicing #3862

Merged
merged 10 commits into from
Mar 24, 2022

Conversation

BarclayII
Copy link
Collaborator

@BarclayII BarclayII commented Mar 18, 2022

Fixes #3847 .

@jermainewang I might need your opinion on the behavior addition on lazy feature slicing. Now if a graph is previously in CPU pinned memory and a subgraph (sample_neighbors) is taken, the new graph's features will be on GPU. Previously it was on CPU. This further complicates the code in Column.data.

Need to run a regression test on this.

Also fixes #3861 .

python/dgl/storages/pytorch_tensor.py Outdated Show resolved Hide resolved
@BarclayII BarclayII merged commit a0e8cf0 into dmlc:master Mar 24, 2022
@BarclayII BarclayII deleted the fix-pinned-again branch March 24, 2022 12:13
BarclayII added a commit that referenced this pull request Mar 31, 2022
* fix uva sampling with features

* fix

* add is_listlike function to distinguish strings from sequence

* fix

Co-authored-by: nv-dlasalle <63612878+nv-dlasalle@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Python 3.10 Support GraphSAGE UVA sampling code crashes when not specifying feature prefetching
2 participants