You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
2.1 KiB
C++
62 lines
2.1 KiB
C++
/*=========================================================================
|
|
|
|
Program: Visualization Toolkit
|
|
Module: vtkDIYExplicitAssigner.h
|
|
|
|
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
|
|
All rights reserved.
|
|
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even
|
|
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
PURPOSE. See the above copyright notice for more information.
|
|
|
|
=========================================================================*/
|
|
/**
|
|
* @class vtkDIYExplicitAssigner
|
|
* @brief assigner for use with DIY
|
|
*
|
|
* vtkDIYExplicitAssigner is a diy::StaticAssigner specialization that be used
|
|
* where the block assignment is not strictly round-robin or contiguous which
|
|
* assumes blocks equally split among ranks. This supports the case where each
|
|
* rank has arbitrary number of blocks per rank. The constructor is provided the
|
|
* mpi communicator and the number of local blocks. It performs parallel
|
|
* communication to exchange information about blocks with all participating
|
|
* ranks.
|
|
*
|
|
* vtkDIYExplicitAssigner also supports ability to pad each rank such that the
|
|
* total number of blocks across all ranks is a power of two.
|
|
*/
|
|
|
|
#ifndef vtkDIYExplicitAssigner_h
|
|
#define vtkDIYExplicitAssigner_h
|
|
|
|
#include "vtkObject.h"
|
|
#include "vtkParallelDIYModule.h" // for export macros
|
|
// clang-format off
|
|
#include "vtk_diy2.h"
|
|
#include VTK_DIY2(diy/mpi.hpp)
|
|
#include VTK_DIY2(diy/assigner.hpp)
|
|
// clang-format on
|
|
|
|
#ifdef _WIN32
|
|
#pragma warning(disable : 4275) /* non dll-interface class `diy::StaticAssigner` used as base for \
|
|
dll-interface class */
|
|
#endif
|
|
|
|
class VTKPARALLELDIY_EXPORT vtkDIYExplicitAssigner : public diy::StaticAssigner
|
|
{
|
|
public:
|
|
vtkDIYExplicitAssigner(
|
|
diy::mpi::communicator comm, int local_blocks, bool force_power_of_two = false);
|
|
|
|
int rank(int gid) const override;
|
|
void local_gids(int rank, std::vector<int>& gids) const override;
|
|
|
|
private:
|
|
std::vector<int> IScanBlockCounts;
|
|
};
|
|
|
|
#endif
|
|
// VTK-HeaderTest-Exclude: vtkDIYExplicitAssigner.h
|