update. (dmlc#1857)

Co-authored-by: Ubuntu <[email protected]>
standbyme · Jul 24, 2020 · 0e896a9 · 0e896a9
1 parent 53fc639
commit 0e896a9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/dgl/partition.py b/python/dgl/partition.py
@@ -209,7 +209,7 @@ def metis_partition_assignment(g, k, balance_ntypes=None, balance_edges=False):
                 nids = F.asnumpy(F.nonzero_1d(balance_ntypes == ntype))
                 degs = np.zeros((g.number_of_nodes(),), np.int64)
                 degs[nids] = F.asnumpy(g.in_degrees(nids))
-                vwgt.append(F.tensor(degs))
+                vwgt.append(F.zerocopy_from_numpy(degs))
 
     # The vertex weights have to be stored in a vector.
     if len(vwgt) > 0:

diff --git a/python/dgl/transform.py b/python/dgl/transform.py
@@ -1078,7 +1078,7 @@ def metis_partition_assignment(g, k, balance_ntypes=None, balance_edges=False):
                 nids = F.asnumpy(F.nonzero_1d(balance_ntypes == ntype))
                 degs = np.zeros((g.number_of_nodes(),), np.int64)
                 degs[nids] = F.asnumpy(g.in_degrees(nids))
-                vwgt.append(F.tensor(degs))
+                vwgt.append(F.zerocopy_from_numpy(degs))
 
     # The vertex weights have to be stored in a vector.
     if len(vwgt) > 0:

diff --git a/third_party/METIS b/third_party/METIS