From fb770cbe13de11f24a37c6861412582ac211071f Mon Sep 17 00:00:00 2001 From: rusty1s Date: Thu, 4 Apr 2024 19:46:47 +0000 Subject: [PATCH] update --- pyg_lib/csrc/partition/cpu/metis_kernel.cpp | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/pyg_lib/csrc/partition/cpu/metis_kernel.cpp b/pyg_lib/csrc/partition/cpu/metis_kernel.cpp index df516224f..7430574f2 100644 --- a/pyg_lib/csrc/partition/cpu/metis_kernel.cpp +++ b/pyg_lib/csrc/partition/cpu/metis_kernel.cpp @@ -1,7 +1,7 @@ #include #include -#include +/* #include */ namespace pyg { namespace partition { @@ -31,14 +31,16 @@ at::Tensor metis_kernel(const at::Tensor& rowptr, auto part = at::empty({nvtxs}, rowptr.options()); auto part_data = part.data_ptr(); - if (recursive) { - METIS_PartGraphRecursive(&nvtxs, &ncon, xadj, adjncy, vwgt, NULL, adjwgt, - &num_partitions, NULL, NULL, NULL, &objval, - part_data); - } else { - METIS_PartGraphKway(&nvtxs, &ncon, xadj, adjncy, vwgt, NULL, adjwgt, - &num_partitions, NULL, NULL, NULL, &objval, part_data); - } + /* if (recursive) { */ + /* METIS_PartGraphRecursive(&nvtxs, &ncon, xadj, adjncy, vwgt, NULL, adjwgt, + */ + /* &num_partitions, NULL, NULL, NULL, &objval, */ + /* part_data); */ + /* } else { */ + /* METIS_PartGraphKway(&nvtxs, &ncon, xadj, adjncy, vwgt, NULL, adjwgt, */ + /* &num_partitions, NULL, NULL, NULL, &objval, + * part_data); */ + /* } */ return part; }