1 /* Copyright (c) 2010-2019. The SimGrid Team. All rights reserved. */
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. */
6 #ifndef SMPI_REQUEST_HPP_INCLUDED
7 #define SMPI_REQUEST_HPP_INCLUDED
10 #include "smpi_f2c.hpp"
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;
20 s4u::ConditionVariablePtr cond;
22 } s_smpi_mpi_generalized_request_funcs_t;
23 typedef struct s_smpi_mpi_generalized_request_funcs *smpi_mpi_generalized_request_funcs;
25 class Request : public F2C {
27 /* in the case of non-contiguous memory the user address should be keep
28 * to unserialize the data inside the user memory*/
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_;
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
44 smx_activity_t action_;
47 MPI_Request detached_sender_;
51 smpi_mpi_generalized_request_funcs generalized_funcs;
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);
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);
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);
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);
92 static void startall(int count, MPI_Request* requests);
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[]);
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);
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[]);
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);
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 int add_f() override;
112 static void free_f(int id);
113 static Request* f2c(int);