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

[distGB] fix local variable 'sorted_idx' referenced before assignment in convert_partition.py #7830

Open
wants to merge 15 commits into
base: master
Choose a base branch
from
Open

[distGB] fix local variable 'sorted_idx' referenced before assignment in convert_partition.py #7830

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
8f9e639
change a variable
Aug 14, 2024
91792fb
Merge branch 'dmlc:master' into master
CfromBU Aug 20, 2024
45ddaa6
resolve conflic in partition
Sep 10, 2024
97d000b
change partition
Sep 23, 2024
7f339b9
fix convert_partition.py
Oct 17, 2024
530e5ef
change partition.py
Oct 17, 2024
0a10717
fix partition
Oct 21, 2024
46d9fe8
remove unused cde
Oct 21, 2024
ee67200
change utils.py
Oct 21, 2024
bdffbe6
change test_dist_partition_graphbolt
Oct 21, 2024
e03e689
change test_dist_part.py function name
Oct 21, 2024
a0b020c
change format
Oct 21, 2024
45c772c
change format
Oct 21, 2024
02e4b91
Merge branch 'master' of github.com:CfromBU/dgl into fix_sorted_idx
Oct 21, 2024
0668d5e
fix conflict
Oct 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
remove unused cde
  • Loading branch information
Ubuntu committed Oct 21, 2024
commit 46d9fe80b3da31a42d54b376587ab248a6a50690
18 changes: 9 additions & 9 deletions python/dgl/distributed/partition.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1417,9 +1417,9 @@ def get_homogeneous(g, balance_ntypes):
for name in g.edges[etype].data:
if name in [EID, "inner_edge"]:
continue
edge_feats[_etype_tuple_to_str(etype) + "/" + name] = (
F.gather_row(g.edges[etype].data[name], local_edges)
)
edge_feats[
_etype_tuple_to_str(etype) + "/" + name
] = F.gather_row(g.edges[etype].data[name], local_edges)
else:
for ntype in g.ntypes:
if len(g.ntypes) > 1:
Expand Down Expand Up @@ -1454,9 +1454,9 @@ def get_homogeneous(g, balance_ntypes):
for name in g.edges[etype].data:
if name in [EID, "inner_edge"]:
continue
edge_feats[_etype_tuple_to_str(etype) + "/" + name] = (
F.gather_row(g.edges[etype].data[name], local_edges)
)
edge_feats[
_etype_tuple_to_str(etype) + "/" + name
] = F.gather_row(g.edges[etype].data[name], local_edges)
# delete `orig_id` from ndata/edata
del part.ndata["orig_id"]
del part.edata["orig_id"]
Expand Down Expand Up @@ -1502,9 +1502,9 @@ def get_homogeneous(g, balance_ntypes):
for part_id, part in parts.items():
part_dir = os.path.join(out_path, "part" + str(part_id))
part_graph_file = os.path.join(part_dir, "graph.dgl")
part_metadata["part-{}".format(part_id)]["part_graph"] = (
os.path.relpath(part_graph_file, out_path)
)
part_metadata["part-{}".format(part_id)][
"part_graph"
] = os.path.relpath(part_graph_file, out_path)
# save DGLGraph
_save_dgl_graphs(
part_graph_file,
Expand Down