fix for unweighted graph partition

363c29ad · bowendeng · 1e959269 · 363c29ad
Commit 363c29ad authored Apr 12, 2020 by bowendeng
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/metis.cpp csrc/metis.cpp +1 -1

No files found.
--- a/csrc/metis.cpp
+++ b/csrc/metis.cpp
@@ -7,7 +7,7 @@ PyMODINIT_FUNC PyInit__metis(void) { return NULL; }
 #endif

 torch::Tensor partition(torch::Tensor rowptr, torch::Tensor col,int64_t num_parts,
-                        torch::optional<torch::Tensor> edge_wgt,bool recursive) {
+                        torch::Tensor edge_wgt,bool recursive) {
  if (rowptr.device().is_cuda()) {
 #ifdef WITH_CUDA
    AT_ERROR("No CUDA version supported");