Skip to content

B
BTS-MTGNN
Commit 8768fe6f by Wenjie Huang
Options

try pyg metis

parent 09c175e2
Showing with 1 additions and 0 deletions
starrygl/utils/partition.py
...@@ -33,6 +33,7 @@ def metis_partition( ...@@ -33,6 +33,7 @@ def metis_partition(
rowptr, col, value = adj_t.coalesce().to_symmetric().csr() rowptr, col, value = adj_t.coalesce().to_symmetric().csr()
node_parts = starrygl.ops.metis_partition( node_parts = starrygl.ops.metis_partition(
rowptr, col, value, node_weight, node_sizes, num_parts, recursive, min_edge_cut) rowptr, col, value, node_weight, node_sizes, num_parts, recursive, min_edge_cut)
# node_parts = torch.ops.torch_sparse.partition(rowptr, col, None, num_parts, num_parts < 8)
return node_parts return node_parts
def mt_metis_partition( def mt_metis_partition(
......
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