Skip to content

B
BTS-MTGNN
Commit 31397ce3 by Wenjie Huang
Options

update METIS

parent 563c0199
Showing with 8 additions and 3 deletions
csrc/partition/mtmetis.cpp
#include <torch/all.h> #include <torch/all.h>
#include <mtmetis.h>
#include "utils.h" #include "utils.h"
#ifdef WITH_MTMETIS
#include <mtmetis.h>
at::Tensor mt_metis_partition( at::Tensor mt_metis_partition(
at::Tensor rowptr, at::Tensor rowptr,
at::Tensor col, at::Tensor col,
...@@ -75,4 +78,6 @@ at::Tensor mt_metis_partition( ...@@ -75,4 +78,6 @@ at::Tensor mt_metis_partition(
AT_ASSERT(ret == MTMETIS_SUCCESS); AT_ASSERT(ret == MTMETIS_SUCCESS);
return part; return part;
} }
\ No newline at end of file
#endif
\ No newline at end of file
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