Commit 563c0199 by Wenjie Huang

update METIS

parent 27d6712a
......@@ -61,7 +61,7 @@ if(WITH_METIS)
# target_link_libraries(metis_partition PRIVATE ${METIS_LIBRARIES})
add_definitions(-DWITH_METIS)
set(METIS_DIR "${CMAKE_SOURCE_DIR}/third_party/mts/METIS")
set(METIS_DIR "${CMAKE_SOURCE_DIR}/third_party/METIS")
set(METIS_GKLIB_DIR "${METIS_DIR}/GKlib")
file(GLOB METIS_SRCS "${METIS_DIR}/libmetis/*.c")
......
......@@ -14,6 +14,7 @@ at::Tensor metis_partition(
bool min_edge_cut
) {
static_assert(sizeof(idx_t) == sizeof(int64_t));
static_assert(sizeof(real_t) == sizeof(float));
CHECK_CPU(rowptr);
AT_ASSERT(rowptr.dim() == 1);
......
Subproject commit f5ae915a84d3bbf1508b529af90292dd7085b9ec
Subproject commit 22008804e8c9b78893ae10a94c0d8b4b592438b4
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