typo

b6dfb9ee · rusty1s · f8697f48 · b6dfb9ee
Commit b6dfb9ee authored Feb 23, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/metis_wrapper.cpp csrc/metis_wrapper.cpp +1 -1

No files found.
--- a/csrc/metis_wrapper.cpp
+++ b/csrc/metis_wrapper.cpp
@@ -18,7 +18,7 @@ torch::Tensor partition(torch::Tensor rowptr, torch::Tensor col,
    AT_ERROR("Not compiled with CUDA support");
 #endif
  } else {
-    return partition_kway_cpu(rowptr, col, num_parts, recursive);
+    return partition_cpu(rowptr, col, num_parts, recursive);
  }
 }