1 /* Copyright (c) 2010-2019. The SimGrid Team.
2 * All rights reserved. */
4 /* This program is free software; you can redistribute it and/or modify it
5 * under the terms of the license (GNU LGPL) which comes with this package. */
7 #ifndef SMPI_WIN_HPP_INCLUDED
8 #define SMPI_WIN_HPP_INCLUDED
10 #include "simgrid/s4u/Barrier.hpp"
11 #include "smpi_errhandler.hpp"
12 #include "smpi_f2c.hpp"
13 #include "smpi_keyvals.hpp"
14 #include "xbt/synchro.h"
22 class Win : public F2C, public Keyval {
29 std::vector<MPI_Request> *requests_;
32 MPI_Win* connected_wins_;
36 int count_; //for ordering the accs
37 s4u::MutexPtr lock_mut_;
38 s4u::MutexPtr atomic_mut_;
39 std::list<int> lockers_;
40 int rank_; // to identify owner for barriers in MPI_COMM_WORLD
41 int mode_; // exclusive or shared lock
44 MPI_Errhandler errhandler_;
47 static std::unordered_map<int, smpi_key_elem> keyvals_;
48 static int keyval_id_;
50 Win(void *base, MPI_Aint size, int disp_unit, MPI_Info info, MPI_Comm comm, int allocated = 0, int dynamic = 0);
51 Win(MPI_Info info, MPI_Comm comm) : Win(MPI_BOTTOM, 0, 1, info, comm, 0, 1) {};
52 Win(const Win&) = delete;
53 Win& operator=(const Win&) = delete;
55 int attach (void *base, MPI_Aint size);
56 int detach (const void *base);
57 void get_name( char* name, int* length);
58 void get_group( MPI_Group* group);
59 void set_name(const char* name);
62 int start(MPI_Group group, int assert);
63 int post(MPI_Group group, int assert);
66 void set_info( MPI_Info info);
71 int fence(int assert);
72 int put(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, int target_rank,
73 MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Request* request=nullptr);
74 int get( void *origin_addr, int origin_count, MPI_Datatype origin_datatype, int target_rank,
75 MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Request* request=nullptr);
76 int accumulate(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, int target_rank,
77 MPI_Aint target_disp, int target_count, MPI_Datatype target_datatype, MPI_Op op, MPI_Request* request=nullptr);
78 int get_accumulate(const void *origin_addr, int origin_count, MPI_Datatype origin_datatype, void *result_addr,
79 int result_count, MPI_Datatype result_datatype, int target_rank, MPI_Aint target_disp, int target_count,
80 MPI_Datatype target_datatype, MPI_Op op, MPI_Request* request=nullptr);
81 int compare_and_swap(const void *origin_addr, void *compare_addr,
82 void *result_addr, MPI_Datatype datatype, int target_rank,
83 MPI_Aint target_disp);
84 static Win* f2c(int id);
86 int lock(int lock_type, int rank, int assert);
88 int lock_all(int assert);
91 int flush_local(int rank);
93 int flush_local_all();
95 int finish_comms(int rank);
96 int shared_query(int rank, MPI_Aint* size, int* disp_unit, void* baseptr);
97 MPI_Errhandler errhandler();
98 void set_errhandler( MPI_Errhandler errhandler);