update METIS

31397ce3 · Wenjie Huang · 563c0199 · 31397ce3
Commit 31397ce3 authored Jan 01, 2024 by Wenjie Huang
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletions

csrc/partition/mtmetis.cpp
+7 -1

No files found.
--- a/csrc/partition/mtmetis.cpp
+++ b/csrc/partition/mtmetis.cpp
 #include <torch/all.h>
-#include <mtmetis.h>
 #include "utils.h"

+#ifdef WITH_MTMETIS
+
+#include <mtmetis.h>
+
 at::Tensor mt_metis_partition(
    at::Tensor rowptr,
    at::Tensor col,
@@ -76,3 +79,5 @@ at::Tensor mt_metis_partition(
    AT_ASSERT(ret == MTMETIS_SUCCESS);
    return part;
 }
+
+#endif
\ No newline at end of file