Loading...
Searching...
No Matches
ParMETIS.h
// Copyright (C) 2008-2009 Niclas Jansson, Ola Skavhaug, Anders Logg,
// Garth N. Wells and Chris Richardson
//
// This file is part of DOLFIN.
//
// DOLFIN is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// DOLFIN 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DOLFIN. If not, see <http://www.gnu.org/licenses/>.
#ifndef __PARMETIS_PARTITIONER_H
#define __PARMETIS_PARTITIONER_H
#include <cstdint>
#include <cstddef>
#include <map>
#include <string>
#include <vector>
#include <boost/multi_array.hpp>
#include <dolfin/common/MPI.h>
#include "CSRGraph.h"
namespace dolfin
{
class CellType;
class ParMETIS
{
public:
static void
compute_partition(const MPI_Comm mpi_comm,
std::vector<int>& cell_partition,
std::map<std::int64_t, std::vector<int>>& ghost_procs,
const boost::multi_array<std::int64_t, 2>& cell_vertices,
const std::size_t num_global_vertices,
const CellType& cell_type,
const std::string mode="partition");
private:
#ifdef HAS_PARMETIS
// Standard ParMETIS partition. CSRGraph should be const, but
// ParMETIS accesses it non-const, so has to be non-const here
template <typename T>
static void partition(MPI_Comm mpi_comm,
CSRGraph<T>& csr_graph,
std::vector<int>& cell_partition,
std::map<std::int64_t, std::vector<int>>& ghost_procs);
// ParMETIS adaptive repartition. CSRGraph should be const, but
// ParMETIS accesses it non-const, so has to be non-const here
template <typename T>
static void adaptive_repartition(MPI_Comm mpi_comm,
CSRGraph<T>& csr_graph,
std::vector<int>& cell_partition);
// ParMETIS refine repartition. CSRGraph should be const, but
// ParMETIS accesses it non-const, so has to be non-const here
template <typename T>
static void refine(MPI_Comm mpi_comm, CSRGraph<T>& csr_graph,
std::vector<int>& cell_partition);
#endif
};
}
#endif
dolfin::CSRGraph
Compressed Sparse Row graph.
Definition CSRGraph.h:45
dolfin::CellType
Definition CellType.h:47
dolfin::ParMETIS
This class provides an interface to ParMETIS.
Definition ParMETIS.h:39
dolfin::ParMETIS::compute_partition
static void compute_partition(const MPI_Comm mpi_comm, std::vector< int > &cell_partition, std::map< std::int64_t, std::vector< int > > &ghost_procs, const boost::multi_array< std::int64_t, 2 > &cell_vertices, const std::size_t num_global_vertices, const CellType &cell_type, const std::string mode="partition")
Definition ParMETIS.cpp:117
dolfin
Definition adapt.h:30