Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
e15735375eaa7042c7184fddba592706b59f92d9
[simgrid.git] / src / smpi / include / smpi_request.hpp
1 /* Copyright (c) 2010-2019. The SimGrid Team. All rights reserved.          */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #ifndef SMPI_REQUEST_HPP_INCLUDED
7 #define SMPI_REQUEST_HPP_INCLUDED
8
9 #include "smpi/smpi.h"
10 #include "smpi_f2c.hpp"
11
12 namespace simgrid{
13 namespace smpi{
14
15 typedef struct s_smpi_mpi_generalized_request_funcs {
16   MPI_Grequest_query_function *query_fn;
17   MPI_Grequest_free_function *free_fn;
18   MPI_Grequest_cancel_function *cancel_fn;
19   void* extra_state;
20   s4u::ConditionVariablePtr cond;
21   s4u::MutexPtr mutex;
22 } s_smpi_mpi_generalized_request_funcs_t; 
23 typedef struct s_smpi_mpi_generalized_request_funcs *smpi_mpi_generalized_request_funcs;
24
25 class Request : public F2C {
26   void* buf_;
27   /* in the case of non-contiguous memory the user address should be keep
28    * to unserialize the data inside the user memory*/
29   void* old_buf_;
30   /* this is especially for derived datatypes that we need to serialize/unserialize.
31    * It let us know how to unserialize at the end of the communication */
32   MPI_Datatype old_type_;
33   size_t size_;
34   int src_;
35   int dst_;
36   int tag_;
37   // to handle cases where we have an unknown sender
38   // We can't override src, tag, and size, because the request may be reused later
39   int real_src_;
40   int real_tag_;
41   int truncated_;
42   size_t real_size_;
43   MPI_Comm comm_;
44   smx_activity_t action_;
45   unsigned flags_;
46   int detached_;
47   MPI_Request detached_sender_;
48   int refcount_;
49   MPI_Op op_;
50   int cancelled_;
51   smpi_mpi_generalized_request_funcs generalized_funcs;
52
53 public:
54   Request() = default;
55   Request(void* buf, int count, MPI_Datatype datatype, int src, int dst, int tag, MPI_Comm comm, unsigned flags);
56   MPI_Comm comm() { return comm_; }
57   size_t size() { return size_; }
58   size_t real_size() { return real_size_; }
59   int src() { return src_; }
60   int dst() { return dst_; }
61   int tag() { return tag_; }
62   int flags() { return flags_; }
63   int detached() { return detached_; }
64   MPI_Datatype type() { return old_type_; }
65   void print_request(const char* message);
66   void start();
67   void cancel();
68   void ref();
69   static void finish_wait(MPI_Request* request, MPI_Status* status);
70   static void unref(MPI_Request* request);
71   static int wait(MPI_Request* req, MPI_Status* status);
72   static MPI_Request send_init(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
73   static MPI_Request isend_init(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
74   static MPI_Request ssend_init(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
75   static MPI_Request rma_send_init(void* buf, int count, MPI_Datatype datatype, int src, int dst, int tag,
76                                    MPI_Comm comm, MPI_Op op);
77   static MPI_Request recv_init(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm);
78   static MPI_Request rma_recv_init(void* buf, int count, MPI_Datatype datatype, int src, int dst, int tag,
79                                    MPI_Comm comm, MPI_Op op);
80   static MPI_Request irecv_init(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm);
81   static MPI_Request isend(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
82   static MPI_Request issend(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
83   static MPI_Request irecv(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm);
84
85   static void recv(void* buf, int count, MPI_Datatype datatype, int src, int tag, MPI_Comm comm, MPI_Status* status);
86   static void send(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
87   static void ssend(void* buf, int count, MPI_Datatype datatype, int dst, int tag, MPI_Comm comm);
88
89   static void sendrecv(void* sendbuf, int sendcount, MPI_Datatype sendtype, int dst, int sendtag, void* recvbuf,
90                        int recvcount, MPI_Datatype recvtype, int src, int recvtag, MPI_Comm comm, MPI_Status* status);
91
92   static void startall(int count, MPI_Request* requests);
93
94   static int test(MPI_Request* request, MPI_Status* status, int* flag);
95   static int testsome(int incount, MPI_Request requests[], int* outcounts, int* indices, MPI_Status status[]);
96   static int testany(int count, MPI_Request requests[], int* index, int* flag, MPI_Status* status);
97   static int testall(int count, MPI_Request requests[], int* flag, MPI_Status status[]);
98
99   static void probe(int source, int tag, MPI_Comm comm, MPI_Status* status);
100   static void iprobe(int source, int tag, MPI_Comm comm, int* flag, MPI_Status* status);
101
102   static int waitany(int count, MPI_Request requests[], MPI_Status* status);
103   static int waitall(int count, MPI_Request requests[], MPI_Status status[]);
104   static int waitsome(int incount, MPI_Request requests[], int* indices, MPI_Status status[]);
105
106   static int match_send(void* a, void* b, kernel::activity::CommImpl* ignored);
107   static int match_recv(void* a, void* b, kernel::activity::CommImpl* ignored);
108
109   static int grequest_start( MPI_Grequest_query_function *query_fn, MPI_Grequest_free_function *free_fn, MPI_Grequest_cancel_function *cancel_fn, void *extra_state, MPI_Request *request);
110   static int grequest_complete( MPI_Request request);
111   static int get_status(MPI_Request req, int* flag, MPI_Status * status);
112
113   int add_f() override;
114   static void free_f(int id);
115   static Request* f2c(int);
116 };
117
118
119 }
120 }
121
122 #endif