From f1d2822929e90dc9023d06abc35065a0695a6a20 Mon Sep 17 00:00:00 2001
From: Sebastian Eibl <sebastian.eibl@fau.de>
Date: Fri, 8 Dec 2017 17:08:49 +0100
Subject: [PATCH] added ParMetis integration test
---
tests/core/CMakeLists.txt | 5 +-
.../core/load_balancing/PlainParMetisTest.cpp | 199 ++++++++++++++++++
2 files changed, 203 insertions(+), 1 deletion(-)
create mode 100644 tests/core/load_balancing/PlainParMetisTest.cpp
diff --git a/tests/core/CMakeLists.txt b/tests/core/CMakeLists.txt
index 214ec7d5f..80ddaa4ef 100644
--- a/tests/core/CMakeLists.txt
+++ b/tests/core/CMakeLists.txt
@@ -215,4 +215,7 @@ if( WALBERLA_BUILD_WITH_PARMETIS )
waLBerla_execute_test( NAME ParMetisTest1 COMMAND $<TARGET_FILE:ParMetisTest> 64 64 4 --no-vtk )
waLBerla_execute_test( NAME ParMetisTest2 COMMAND $<TARGET_FILE:ParMetisTest> 64 64 8 --no-vtk PROCESSES 2 )
waLBerla_execute_test( NAME ParMetisTest4 COMMAND $<TARGET_FILE:ParMetisTest> 64 64 16 --no-vtk PROCESSES 4 )
-endif()
\ No newline at end of file
+
+ waLBerla_compile_test( NAME PlainParMetisTest FILES load_balancing/PlainParMetisTest.cpp )
+ waLBerla_execute_test( NAME PlainParMetisTest PROCESSES 3 )
+endif()
diff --git a/tests/core/load_balancing/PlainParMetisTest.cpp b/tests/core/load_balancing/PlainParMetisTest.cpp
new file mode 100644
index 000000000..4419b3c0d
--- /dev/null
+++ b/tests/core/load_balancing/PlainParMetisTest.cpp
@@ -0,0 +1,199 @@
+//======================================================================================================================
+//
+// This file is part of waLBerla. waLBerla is free software: you can
+// redistribute it and/or modify it under the terms of the GNU General Public
+// License as published by the Free Software Foundation, either version 3 of
+// the License, or (at your option) any later version.
+//
+// waLBerla is distributed in the hope that it will be useful, but WITHOUT
+// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+// for more details.
+//
+// You should have received a copy of the GNU General Public License along
+// with waLBerla (see COPYING.txt). If not, see <http://www.gnu.org/licenses/>.
+//
+//! \file PlainParMetisTest.cpp
+//! \author Sebastian Eibl <sebastian.eibl@fau.de>
+//! \brief Integration test for parmetis library
+//
+//======================================================================================================================
+
+#include "core/mpi/MPIWrapper.h"
+#include "core/load_balancing/ParMetisWrapper.h"
+#include "core/debug/CheckFunctions.h"
+
+#include <cstdlib>
+#include <iostream>
+#include <vector>
+
+int main( int argc, char * argv[] )
+{
+ #ifdef WALBERLA_BUILD_WITH_MPI
+ #ifdef WALBERLA_BUILD_WITH_PARMETIS
+
+ using namespace walberla::core;
+
+ MPI_Init(&argc, &argv);
+ int world_rank;
+ MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
+
+ const int64_t numVertices = 25;
+
+ std::vector< int64_t > vtxdist, xadj, adjncy;
+ std::vector<double> xyz;
+ vtxdist.push_back( 0 );
+ vtxdist.push_back( 5 );
+ vtxdist.push_back( 10 );
+ vtxdist.push_back( 15 );
+
+ if (world_rank == 0)
+ {
+ xadj.push_back( 0 );
+ xadj.push_back( 2 );
+ xadj.push_back( 5 );
+ xadj.push_back( 8 );
+ xadj.push_back( 11 );
+ xadj.push_back( 13 );
+
+ xyz.push_back( 0 );
+ xyz.push_back( 2 );
+ xyz.push_back( 1 );
+ xyz.push_back( 2 );
+ xyz.push_back( 2 );
+ xyz.push_back( 2 );
+ xyz.push_back( 3 );
+ xyz.push_back( 2 );
+ xyz.push_back( 4 );
+ xyz.push_back( 2 );
+
+ adjncy.push_back( 1 );
+ adjncy.push_back( 5 );
+ adjncy.push_back( 0 );
+ adjncy.push_back( 2 );
+ adjncy.push_back( 6 );
+ adjncy.push_back( 1 );
+ adjncy.push_back( 3 );
+ adjncy.push_back( 7 );
+ adjncy.push_back( 2 );
+ adjncy.push_back( 4 );
+ adjncy.push_back( 8 );
+ adjncy.push_back( 3 );
+ adjncy.push_back( 9 );
+ }
+
+ if (world_rank == 1)
+ {
+ xadj.push_back( 0 );
+ xadj.push_back( 3 );
+ xadj.push_back( 7 );
+ xadj.push_back( 11 );
+ xadj.push_back( 15 );
+ xadj.push_back( 18 );
+
+ xyz.push_back( 0 );
+ xyz.push_back( 1 );
+ xyz.push_back( 1 );
+ xyz.push_back( 1 );
+ xyz.push_back( 2 );
+ xyz.push_back( 1 );
+ xyz.push_back( 3 );
+ xyz.push_back( 1 );
+ xyz.push_back( 4 );
+ xyz.push_back( 1 );
+
+ adjncy.push_back( 0 );
+ adjncy.push_back( 6 );
+ adjncy.push_back( 10 );
+ adjncy.push_back( 1 );
+ adjncy.push_back( 5 );
+ adjncy.push_back( 7 );
+ adjncy.push_back( 11 );
+ adjncy.push_back( 2 );
+ adjncy.push_back( 6 );
+ adjncy.push_back( 8 );
+ adjncy.push_back( 12 );
+ adjncy.push_back( 3 );
+ adjncy.push_back( 7 );
+ adjncy.push_back( 9 );
+ adjncy.push_back( 13 );
+ adjncy.push_back( 4 );
+ adjncy.push_back( 8 );
+ adjncy.push_back( 14 );
+ }
+
+ if (world_rank == 2)
+ {
+ xadj.push_back( 0 );
+ xadj.push_back( 2 );
+ xadj.push_back( 5 );
+ xadj.push_back( 8 );
+ xadj.push_back( 11 );
+ xadj.push_back( 13 );
+
+ xyz.push_back( 0 );
+ xyz.push_back( 0 );
+ xyz.push_back( 1 );
+ xyz.push_back( 0 );
+ xyz.push_back( 2 );
+ xyz.push_back( 0 );
+ xyz.push_back( 3 );
+ xyz.push_back( 0 );
+ xyz.push_back( 4 );
+ xyz.push_back( 0 );
+
+ adjncy.push_back( 5 );
+ adjncy.push_back( 11 );
+ adjncy.push_back( 6 );
+ adjncy.push_back( 10 );
+ adjncy.push_back( 12 );
+ adjncy.push_back( 7 );
+ adjncy.push_back( 11 );
+ adjncy.push_back( 13 );
+ adjncy.push_back( 8 );
+ adjncy.push_back( 12 );
+ adjncy.push_back( 14 );
+ adjncy.push_back( 9 );
+ adjncy.push_back( 13 );
+ }
+
+ int64_t wgtflag = 0;
+ int64_t numflag = 0;
+ int64_t ndims = 2;
+ int64_t ncon = int64_t( 1 );
+ int64_t nparts = int64_t( 5 );
+ std::vector< double > tpwgts( static_cast<size_t>(nparts), 1.0 / static_cast<double>( nparts ) );
+ std::vector< double > ubvec( static_cast<size_t>(ncon), 1.05 );
+ double ipc2redist = 1.0;
+ int64_t options[] = {0,0,0};
+ int64_t edgecut;
+ std::vector< int64_t > part( numVertices );
+ MPI_Comm comm = MPI_COMM_WORLD;
+
+ std::cout << "ParMETIS_V3_PartKway" << std::endl;
+ WALBERLA_CHECK_EQUAL( ParMETIS_V3_PartKway( &(vtxdist.front()), &(xadj.front()), &(adjncy.front()), NULL, NULL, &wgtflag, &numflag, &ncon, &nparts,
+ &(tpwgts.front()), &(ubvec.front()), options, &edgecut, &(part.front()), &comm ),
+ METIS_OK );
+ std::cout << "ParMETIS_V3_PartGeomKway" << std::endl;
+ WALBERLA_CHECK_EQUAL( ParMETIS_V3_PartGeomKway( &(vtxdist.front()), &(xadj.front()), &(adjncy.front()), NULL, NULL, &wgtflag, &numflag, &ndims, &(xyz.front()), &ncon, &nparts,
+ &(tpwgts.front()), &(ubvec.front()), options, &edgecut, &(part.front()), &comm ),
+ METIS_OK );
+ std::cout << "ParMETIS_V3_PartGeom" << std::endl;
+ WALBERLA_CHECK_EQUAL( ParMETIS_V3_PartGeom( &(vtxdist.front()), &ndims, &(xyz.front()), &(part.front()), &comm ),
+ METIS_OK );
+ std::cout << "ParMETIS_V3_AdaptiveRepart" << std::endl;
+ WALBERLA_CHECK_EQUAL( ParMETIS_V3_AdaptiveRepart( &(vtxdist.front()), &(xadj.front()), &(adjncy.front()), NULL, NULL, NULL, &wgtflag, &numflag, &ncon, &nparts,
+ &(tpwgts.front()), &(ubvec.front()), &ipc2redist, options, &edgecut, &(part.front()), &comm ),
+ METIS_OK );
+ std::cout << "ParMETIS_V3_RefineKway" << std::endl;
+ WALBERLA_CHECK_EQUAL( ParMETIS_V3_RefineKway( &(vtxdist.front()), &(xadj.front()), &(adjncy.front()), NULL, NULL, &wgtflag, &numflag, &ncon, &nparts,
+ &(tpwgts.front()), &(ubvec.front()), options, &edgecut, &(part.front()), &comm ),
+ METIS_OK );
+
+ MPI_Finalize();
+
+ #endif
+ #endif
+
+ return 0;
+}
--
GitLab