Skip to content

GitLab

"docs/git@developer.sourcefind.cn:norm/vllm.git" did not exist on "6549aef24501d5cd1a0d22a19e9e0f8e04f8bd77"
Unverified Commit 782527d4 authored by xiang song(charlie.song)'s avatar xiang song(charlie.song) Committed by GitHub
Browse files

Fix partition for 0.5.1 (#2187) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-24-210.ec2.internal>
parent 877101a9
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/rgcn/experimental/partition_graph.py
View file @ 782527d4
...@@ -50,7 +50,7 @@ def load_ogb(dataset, global_norm): ...@@ -50,7 +50,7 @@ def load_ogb(dataset, global_norm):
if ntype == category: if ntype == category:
category_id = i category_id = i
g = dgl.to_homo(hg) g = dgl.to_homogeneous(hg, edata=['norm'])
if global_norm: if global_norm:
u, v, eid = g.all_edges(form='all') u, v, eid = g.all_edges(form='all')
_, inverse_index, count = th.unique(v, return_inverse=True, return_counts=True) _, inverse_index, count = th.unique(v, return_inverse=True, return_counts=True)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment