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.

Sign up for GitHub

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

Jump to bottom

Allow GPU inference in BasicGNN #7548

Merged
merged 5 commits into from
Jun 12, 2023
Merged

Allow GPU inference in BasicGNN #7548

merged 5 commits into from
Jun 12, 2023

Conversation

Kh4L
Copy link
Contributor

@Kh4L Kh4L commented Jun 8, 2023

No description provided.

@Kh4L
Allow GPU inference in basic GNNs
028a9ca 
Signed-off-by: Serge Panev <spanev@nvidia.com>
@Kh4L Kh4L requested a review from EdisonLeeeee as a code owner June 8, 2023 16:19
@Kh4L Kh4L mentioned this pull request Jun 8, 2023
Closed
@codecov
Copy link

codecov bot commented Jun 8, 2023 •
edited

Codecov Report

❗ No coverage uploaded for pull request base (master@17e81c4). Click here to learn what that means.
The diff coverage is 100.00%.

❗ Current head bd45995 differs from pull request most recent head 7562db4. Consider uploading reports for the commit 7562db4 to get more accurate results

@@            Coverage Diff            @@
##             master    #7548   +/-   ##
=========================================
  Coverage          ?   91.41%           
=========================================
  Files             ?      449           
  Lines             ?    25091           
  Branches          ?        0           
=========================================
  Hits              ?    22936           
  Misses            ?     2155           
  Partials          ?        0
Impacted Files Coverage Δ
torch_geometric/nn/models/basic_gnn.py 89.72% <100.00%> (ø)

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

@Kh4L
Add embedding device
0604fe3 
Signed-off-by: Serge Panev <spanev@nvidia.com>
@rusty1s rusty1s changed the title Allow GPU inference in basic GNNs Allow GPU inference in BasicGNN Jun 10, 2023
@rusty1s rusty1s enabled auto-merge (squash) June 12, 2023 14:44
@rusty1s rusty1s merged commit 09cb440 into pyg-team:master Jun 12, 2023
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@rusty1s rusty1s rusty1s approved these changes

@EdisonLeeeee EdisonLeeeee Awaiting requested review from EdisonLeeeee EdisonLeeeee is a code owner

Assignees

@Kh4L Kh4L

Labels
0 - Priority P0 feature loader
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
2 participants
@Kh4L @rusty1s