1 /* Copyright (c) 2009-2023. 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_DATATYPE_HPP
7 #define SMPI_DATATYPE_HPP
9 #include "smpi_f2c.hpp"
10 #include "smpi_keyvals.hpp"
14 constexpr unsigned DT_FLAG_DESTROYED = 0x00001; /**< user destroyed but some other layers still have a reference */
15 constexpr unsigned DT_FLAG_COMMITED = 0x00002; /**< ready to be used for a send/recv operation */
16 constexpr unsigned DT_FLAG_CONTIGUOUS = 0x00004; /**< contiguous datatype */
17 constexpr unsigned DT_FLAG_OVERLAP = 0x00008; /**< datatype is unproper for a recv operation */
18 constexpr unsigned DT_FLAG_USER_LB = 0x00010; /**< has a user defined LB */
19 constexpr unsigned DT_FLAG_USER_UB = 0x00020; /**< has a user defined UB */
20 constexpr unsigned DT_FLAG_PREDEFINED = 0x00040; /**< cannot be removed: initial and predefined datatypes */
21 constexpr unsigned DT_FLAG_NO_GAPS = 0x00080; /**< no gaps around the datatype */
22 constexpr unsigned DT_FLAG_DATA = 0x00100; /**< data or control structure */
23 constexpr unsigned DT_FLAG_ONE_SIDED = 0x00200; /**< datatype can be used for one sided operations */
24 constexpr unsigned DT_FLAG_UNAVAILABLE = 0x00400; /**< datatypes unavailable on the build (OS or compiler dependent) */
25 constexpr unsigned DT_FLAG_DERIVED = 0x00800; /**< is the datatype derived ? */
26 constexpr unsigned DT_FLAG_C_INTEGER = 0x01000; /**< Family: C ints */
27 constexpr unsigned DT_FLAG_F_INTEGER = 0x02000; /**< Family: F ints */
28 constexpr unsigned DT_FLAG_FP = 0x04000; /**< Family: Floating point */
29 constexpr unsigned DT_FLAG_LOGICAL = 0x08000; /**< Family: Logical */
30 constexpr unsigned DT_FLAG_COMPLEX = 0x10000; /**< Family: Complex */
31 constexpr unsigned DT_FLAG_BYTE = 0x20000; /**< Family: Poor lonesome byte */
32 constexpr unsigned DT_FLAG_MULTILANG = 0x40000; /**< Family: Multi-language */
33 constexpr unsigned DT_FLAG_REDUCTION = 0x80000; /**< Family: Dual types for maxloc/minloc */
35 * We should make the difference here between the predefined contiguous and non contiguous
36 * datatypes. The DT_FLAG_BASIC is held by all predefined contiguous datatypes.
38 constexpr unsigned DT_FLAG_BASIC =
39 (DT_FLAG_PREDEFINED | DT_FLAG_CONTIGUOUS | DT_FLAG_NO_GAPS | DT_FLAG_DATA | DT_FLAG_COMMITED);
41 extern MPI_Datatype MPI_PTR;
43 //The following are datatypes for the MPI functions MPI_MAXLOC and MPI_MINLOC.
56 struct double_double {
76 struct long_double_int {
85 namespace simgrid::smpi {
86 class Datatype_contents {
89 std::vector<int> integers_;
90 std::vector<MPI_Aint> addresses_;
91 std::vector<MPI_Datatype> datatypes_;
92 Datatype_contents(int combiner, int number_of_integers, const int* integers, int number_of_addresses,
93 const MPI_Aint* addresses, int number_of_datatypes, const MPI_Datatype* datatypes);
94 Datatype_contents(const Datatype_contents&) = delete;
95 Datatype_contents& operator=(const Datatype_contents&) = delete;
99 class Datatype : public F2C, public Keyval{
100 std::string name_ = "";
101 /* The id here is the (unique) datatype id used for this datastructure.
102 * It's default value is set to -1 since some code expects this return value
103 * when no other id has been assigned
105 std::string id = "-1";
111 std::unique_ptr<Datatype_contents> contents_ = nullptr;
112 MPI_Datatype duplicated_datatype_ = MPI_DATATYPE_NULL;
115 template <typename... Args> void set_contents(Args&&... args)
117 contents_ = std::make_unique<Datatype_contents>(std::forward<Args>(args)...);
121 static std::unordered_map<int, smpi_key_elem> keyvals_;
122 static int keyval_id_;
124 Datatype(int ident, int size, MPI_Aint lb, MPI_Aint ub, int flags);
125 Datatype(const char* name, int ident, int size, MPI_Aint lb, MPI_Aint ub, int flags);
126 Datatype(int size, MPI_Aint lb, MPI_Aint ub, int flags);
127 Datatype(Datatype* datatype, int* ret);
128 Datatype(const Datatype&) = delete;
129 Datatype& operator=(const Datatype&) = delete;
130 ~Datatype() override;
131 std::string name() const override { return name_.empty() ? "MPI_Datatype" : name_; }
132 size_t size() const { return size_; }
133 MPI_Aint lb() const { return lb_; }
134 MPI_Aint ub() const { return ub_; }
135 int flags() const { return flags_; }
136 int refcount() const { return refcount_; }
137 MPI_Datatype duplicated_datatype() const { return duplicated_datatype_; }
140 static void unref(MPI_Datatype datatype);
142 int copy_attrs(Datatype* datatype);
143 bool is_valid() const;
144 bool is_basic() const;
145 static const char* encode(const Datatype* dt) { return dt->id.c_str(); }
146 static MPI_Datatype decode(const std::string& datatype_id);
147 void addflag(int flag);
148 int extent(MPI_Aint* lb, MPI_Aint* extent) const;
149 MPI_Aint get_extent() const { return ub_ - lb_; };
150 void get_name(char* name, int* length) const;
151 void set_name(const char* name);
152 static int copy(const void* sendbuf, int sendcount, MPI_Datatype sendtype, void* recvbuf, int recvcount,
153 MPI_Datatype recvtype);
154 virtual int clone(MPI_Datatype* type);
155 virtual void serialize(const void* noncontiguous, void* contiguous, int count);
156 virtual void unserialize(const void* contiguous, void* noncontiguous, int count, MPI_Op op);
157 int pack(const void* inbuf, int incount, void* outbuf, int outcount, int* position, const Comm* comm);
158 int unpack(const void* inbuf, int insize, int* position, void* outbuf, int outcount, const Comm* comm);
159 int get_contents(int max_integers, int max_addresses, int max_datatypes, int* array_of_integers,
160 MPI_Aint* array_of_addresses, MPI_Datatype* array_of_datatypes) const;
161 int get_envelope(int* num_integers, int* num_addresses, int* num_datatypes, int* combiner) const;
162 static int create_contiguous(int count, MPI_Datatype old_type, MPI_Aint lb, MPI_Datatype* new_type);
163 static int create_vector(int count, int blocklen, int stride, MPI_Datatype old_type, MPI_Datatype* new_type);
164 static int create_hvector(int count, int blocklen, MPI_Aint stride, MPI_Datatype old_type, MPI_Datatype* new_type);
165 static int create_indexed(int count, const int* blocklens, const int* indices, MPI_Datatype old_type, MPI_Datatype* new_type);
166 static int create_hindexed(int count, const int* blocklens, const MPI_Aint* indices, MPI_Datatype old_type,
167 MPI_Datatype* new_type);
168 static int create_struct(int count, const int* blocklens, const MPI_Aint* indices, const MPI_Datatype* old_types,
169 MPI_Datatype* new_type);
170 static int create_subarray(int ndims, const int* array_of_sizes,
171 const int* array_of_subsizes, const int* array_of_starts,
172 int order, MPI_Datatype oldtype, MPI_Datatype *newtype);
173 static int create_resized(MPI_Datatype oldtype,MPI_Aint lb, MPI_Aint extent,
174 MPI_Datatype *newtype);
175 static Datatype* f2c(int id);
178 } // namespace simgrid::smpi