VTK  9.0.1
vtkMPICommunicator.h
Go to the documentation of this file.
1 /*=========================================================================
2 
3  Program: Visualization Toolkit
4  Module: vtkMPICommunicator.h
5 
6  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
7  All rights reserved.
8  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
9 
10  This software is distributed WITHOUT ANY WARRANTY; without even
11  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
12  PURPOSE. See the above copyright notice for more information.
13 
14 =========================================================================*/
36 #ifndef vtkMPICommunicator_h
37 #define vtkMPICommunicator_h
38 
39 #include "vtkCommunicator.h"
40 #include "vtkParallelMPIModule.h" // For export macro
41 
42 class vtkMPIController;
43 class vtkProcessGroup;
44 
45 class vtkMPICommunicatorOpaqueComm;
46 class vtkMPICommunicatorOpaqueRequest;
47 class vtkMPICommunicatorReceiveDataInfo;
48 
49 class VTKPARALLELMPI_EXPORT vtkMPICommunicator : public vtkCommunicator
50 {
51 public:
52  class VTKPARALLELMPI_EXPORT Request
53  {
54  public:
55  Request();
56  Request(const Request&);
57  ~Request();
58  Request& operator=(const Request&);
59  int Test();
60  void Cancel();
61  void Wait();
62  vtkMPICommunicatorOpaqueRequest* Req;
63  };
64 
66  void PrintSelf(ostream& os, vtkIndent indent) override;
67 
71  static vtkMPICommunicator* New();
72 
77  static vtkMPICommunicator* GetWorldCommunicator();
78 
83  int Initialize(vtkProcessGroup* group);
84 
90  int SplitInitialize(vtkCommunicator* oldcomm, int color, int key);
91 
93 
98  virtual int SendVoidArray(
99  const void* data, vtkIdType length, int type, int remoteProcessId, int tag) override;
100  virtual int ReceiveVoidArray(
101  void* data, vtkIdType length, int type, int remoteProcessId, int tag) override;
103 
105 
113  int NoBlockSend(const int* data, int length, int remoteProcessId, int tag, Request& req);
114  int NoBlockSend(
115  const unsigned long* data, int length, int remoteProcessId, int tag, Request& req);
116  int NoBlockSend(const char* data, int length, int remoteProcessId, int tag, Request& req);
117  int NoBlockSend(
118  const unsigned char* data, int length, int remoteProcessId, int tag, Request& req);
119  int NoBlockSend(const float* data, int length, int remoteProcessId, int tag, Request& req);
120  int NoBlockSend(const double* data, int length, int remoteProcessId, int tag, Request& req);
121 #ifdef VTK_USE_64BIT_IDS
122  int NoBlockSend(const vtkIdType* data, int length, int remoteProcessId, int tag, Request& req);
123 #endif
124 
125 
127 
134  int NoBlockReceive(int* data, int length, int remoteProcessId, int tag, Request& req);
135  int NoBlockReceive(unsigned long* data, int length, int remoteProcessId, int tag, Request& req);
136  int NoBlockReceive(char* data, int length, int remoteProcessId, int tag, Request& req);
137  int NoBlockReceive(unsigned char* data, int length, int remoteProcessId, int tag, Request& req);
138  int NoBlockReceive(float* data, int length, int remoteProcessId, int tag, Request& req);
139  int NoBlockReceive(double* data, int length, int remoteProcessId, int tag, Request& req);
140 #ifdef VTK_USE_64BIT_IDS
141  int NoBlockReceive(vtkIdType* data, int length, int remoteProcessId, int tag, Request& req);
142 #endif
143 
144 
146 
151  virtual void Barrier() override;
152  virtual int BroadcastVoidArray(void* data, vtkIdType length, int type, int srcProcessId) override;
153  virtual int GatherVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType length, int type,
154  int destProcessId) override;
155  virtual int GatherVVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType sendLength,
156  vtkIdType* recvLengths, vtkIdType* offsets, int type, int destProcessId) override;
157  virtual int ScatterVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType length, int type,
158  int srcProcessId) override;
159  virtual int ScatterVVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType* sendLengths,
160  vtkIdType* offsets, vtkIdType recvLength, int type, int srcProcessId) override;
161  virtual int AllGatherVoidArray(
162  const void* sendBuffer, void* recvBuffer, vtkIdType length, int type) override;
163  virtual int AllGatherVVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType sendLength,
164  vtkIdType* recvLengths, vtkIdType* offsets, int type) override;
165  virtual int ReduceVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType length, int type,
166  int operation, int destProcessId) override;
167  virtual int ReduceVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType length, int type,
168  Operation* operation, int destProcessId) override;
169  virtual int AllReduceVoidArray(
170  const void* sendBuffer, void* recvBuffer, vtkIdType length, int type, int operation) override;
171  virtual int AllReduceVoidArray(const void* sendBuffer, void* recvBuffer, vtkIdType length,
172  int type, Operation* operation) override;
174 
176 
185  int Iprobe(int source, int tag, int* flag, int* actualSource);
186  int Iprobe(int source, int tag, int* flag, int* actualSource, int* type, int* size);
187  int Iprobe(int source, int tag, int* flag, int* actualSource, unsigned long* type, int* size);
188  int Iprobe(int source, int tag, int* flag, int* actualSource, const char* type, int* size);
189  int Iprobe(int source, int tag, int* flag, int* actualSource, float* type, int* size);
190  int Iprobe(int source, int tag, int* flag, int* actualSource, double* type, int* size);
192 
197  int WaitAll(const int count, Request requests[]);
198 
204  int WaitAny(const int count, Request requests[], int& idx) VTK_SIZEHINT(requests, count);
205 
211  int WaitSome(const int count, Request requests[], int& NCompleted, int* completed)
212  VTK_SIZEHINT(requests, count);
213 
219  int TestAll(const int count, Request requests[], int& flag) VTK_SIZEHINT(requests, count);
220 
224  int TestAny(const int count, Request requests[], int& idx, int& flag)
225  VTK_SIZEHINT(requests, count);
226 
233  int TestSome(const int count, Request requests[], int& NCompleted, int* completed)
234  VTK_SIZEHINT(requests, count);
235 
236  friend class vtkMPIController;
237 
238  vtkMPICommunicatorOpaqueComm* GetMPIComm() { return this->MPIComm; }
239 
240  int InitializeExternal(vtkMPICommunicatorOpaqueComm* comm);
241 
242  static char* Allocate(size_t size);
243  static void Free(char* ptr);
244 
246 
250  vtkSetClampMacro(UseSsend, int, 0, 1);
251  vtkGetMacro(UseSsend, int);
252  vtkBooleanMacro(UseSsend, int);
254 
262  void CopyFrom(vtkMPICommunicator* source);
263 
264 protected:
266  ~vtkMPICommunicator() override;
267 
268  // Obtain size and rank setting NumberOfProcesses and LocalProcessId Should
269  // not be called if the current communicator does not include this process
270  int InitializeNumberOfProcesses();
271 
273 
283  vtkSetMacro(KeepHandle, int);
284  vtkBooleanMacro(KeepHandle, int);
286 
288 
289  void InitializeCopy(vtkMPICommunicator* source);
290 
300  void Duplicate(vtkMPICommunicator* source);
301 
305  virtual int ReceiveDataInternal(char* data, int length, int sizeoftype, int remoteProcessId,
306  int tag, vtkMPICommunicatorReceiveDataInfo* info, int useCopy, int& senderId);
307 
308  vtkMPICommunicatorOpaqueComm* MPIComm;
309 
312 
314  int UseSsend;
315  static int CheckForMPIError(int err);
316 
317 private:
318  vtkMPICommunicator(const vtkMPICommunicator&) = delete;
319  void operator=(const vtkMPICommunicator&) = delete;
320 };
321 
322 #endif
vtkMPICommunicator::Request
Definition: vtkMPICommunicator.h:52
vtkMPICommunicator::Initialized
int Initialized
Definition: vtkMPICommunicator.h:310
vtkX3D::type
@ type
Definition: vtkX3D.h:522
vtkCommunicator
Used to send/receive messages in a multiprocess environment.
Definition: vtkCommunicator.h:50
vtkIdType
int vtkIdType
Definition: vtkType.h:338
vtkMPICommunicator::UseSsend
int UseSsend
Definition: vtkMPICommunicator.h:314
vtkX3D::data
@ data
Definition: vtkX3D.h:321
vtkCommunicator::Operation
A custom operation to use in a reduce command.
Definition: vtkCommunicator.h:103
vtkX3D::key
@ key
Definition: vtkX3D.h:263
vtkX3D::length
@ length
Definition: vtkX3D.h:399
vtkMPICommunicator::LastSenderId
int LastSenderId
Definition: vtkMPICommunicator.h:313
vtkMPICommunicator
Class for creating user defined MPI communicators.
Definition: vtkMPICommunicator.h:49
vtkX3D::color
@ color
Definition: vtkX3D.h:227
VTK_SIZEHINT
#define VTK_SIZEHINT(...)
Definition: vtkWrappingHints.h:45
impl::Test
int Test(int argc, char *argv[], const char *dfile, const InitializationCallback &initCallback)
Definition: TestMotionFXCFGReaderCommon.h:109
vtkMPICommunicator::WorldCommunicator
static vtkMPICommunicator * WorldCommunicator
Definition: vtkMPICommunicator.h:284
vtkIndent
a simple class to control print indentation
Definition: vtkIndent.h:33
vtkX3D::size
@ size
Definition: vtkX3D.h:259
vtkX3D::info
@ info
Definition: vtkX3D.h:382
vtkProcessGroup
A subgroup of processes from a communicator.
Definition: vtkProcessGroup.h:54
vtkCommunicator.h
vtkMPICommunicator::KeepHandle
int KeepHandle
Definition: vtkMPICommunicator.h:311
vtkMPICommunicator::MPIComm
vtkMPICommunicatorOpaqueComm * MPIComm
Definition: vtkMPICommunicator.h:308
vtkMPICommunicator::GetMPIComm
vtkMPICommunicatorOpaqueComm * GetMPIComm()
Definition: vtkMPICommunicator.h:238
source
boost::graph_traits< vtkGraph * >::vertex_descriptor source(boost::graph_traits< vtkGraph * >::edge_descriptor e, vtkGraph *)
Definition: vtkBoostGraphAdapter.h:959
vtkMPIController
Process communication using MPI.
Definition: vtkMPIController.h:55
vtkMPICommunicator::Request::Req
vtkMPICommunicatorOpaqueRequest * Req
Definition: vtkMPICommunicator.h:62