7#include <memory_resource>
15#include "ddc/chunk_traits.hpp"
17#include "chunk_span.hpp"
22static constexpr PDI_inout_t default_access_v
23 = (std::is_lvalue_reference_v<T> && !std::is_const_v<std::remove_reference_t<T>>)
28static constexpr PDI_inout_t chunk_default_access_v = is_writable_chunk_v<T> ? PDI_INOUT : PDI_OUT;
32 std::string m_event_name;
34 std::vector<std::string> m_names;
37 std::pmr::monotonic_buffer_resource m_metadata;
40 explicit PdiEvent(std::string
const& event_name) : m_event_name(event_name) {}
48 PDI_event(m_event_name.c_str());
49 for (std::string
const& one_name : m_names) {
50 PDI_reclaim(one_name.c_str());
64 std::enable_if_t<is_borrowed_chunk_v<BorrowedChunk>,
int> = 0>
65 PdiEvent&
with(std::string
const& name, BorrowedChunk&& data)
68 !(access & PDI_IN) || (chunk_default_access_v<BorrowedChunk> & PDI_IN),
69 "Invalid access for constant data");
70 auto extents = detail::array(data.domain().extents());
71 std::size_t& rank = *std::pmr::polymorphic_allocator<std::size_t>(&m_metadata).allocate(1);
72 rank = extents.size();
73 PDI_share((name +
"_rank").c_str(), &rank, PDI_OUT);
74 m_names.push_back(name +
"_rank");
76 (name +
"_extents").c_str(),
77 std::pmr::vector<std::size_t>(extents.begin(), extents.end(), &m_metadata).data(),
79 m_names.push_back(name +
"_extents");
82 const_cast<chunk_value_t<BorrowedChunk>*>(data.data_handle()),
84 m_names.push_back(name);
91 std::enable_if_t<std::is_arithmetic_v<std::remove_reference_t<Arithmetic>>,
int> = 0>
92 PdiEvent&
with(std::string
const& name, Arithmetic&& data)
95 !(access & PDI_IN) || (default_access_v<Arithmetic> & PDI_IN),
96 "Invalid access for constant data");
97 using value_type = std::remove_cv_t<std::remove_reference_t<Arithmetic>>;
98 value_type* data_ptr =
const_cast<value_type*>(&data);
100 if constexpr (!(access & PDI_IN)) {
101 data_ptr = std::pmr::polymorphic_allocator<value_type>(&m_metadata).allocate(1);
104 PDI_share(name.c_str(), data_ptr, access);
105 m_names.push_back(name);
109 template <PDI_inout_t access,
class T>
112 return with<access>(name, std::forward<T>(t));
120 template <
class BorrowedChunk, std::enable_if_t<is_borrowed_chunk_v<BorrowedChunk>,
int> = 0>
121 PdiEvent&
with(std::string
const& name, BorrowedChunk&& data)
123 return with<chunk_default_access_v<BorrowedChunk>>(name, std::forward<BorrowedChunk>(data));
129 std::enable_if_t<std::is_arithmetic_v<std::remove_reference_t<Arithmetic>>,
int> = 0>
130 PdiEvent&
with(std::string
const& name, Arithmetic&& data)
132 return with<default_access_v<Arithmetic>>(name, std::forward<Arithmetic>(data));
139 return with(name, std::forward<T>(t));
145template <PDI_inout_t access,
class DataType>
148 PdiEvent(name
).with<access>(name, std::forward<DataType>(data));
151template <
class DataType>
154 PdiEvent(name
).with(name, std::forward<DataType>(data));
PdiEvent(PdiEvent const &rhs)=delete
PdiEvent & and_with(std::string const &name, T &&t)
With synonym.
Definition pdi.hpp:137
PdiEvent & and_with(std::string const &name, T &&t)
Definition pdi.hpp:110
PdiEvent & operator=(PdiEvent &&rhs) noexcept=delete
~PdiEvent() noexcept
Definition pdi.hpp:46
PdiEvent & operator=(PdiEvent const &rhs)=delete
PdiEvent(std::string const &event_name)
Definition pdi.hpp:40
PdiEvent & with(std::string const &name, Arithmetic &&data)
Arithmetic overload.
Definition pdi.hpp:130
PdiEvent & with(std::string const &name, BorrowedChunk &&data)
API with access deduction.
Definition pdi.hpp:121
PdiEvent & with(std::string const &name, Arithmetic &&data)
Definition pdi.hpp:92
PdiEvent & with(std::string const &name, BorrowedChunk &&data)
Definition pdi.hpp:65
PdiEvent(PdiEvent &&rhs) noexcept=delete
The top-level namespace of DDC.
Definition aligned_allocator.hpp:11
void expose_to_pdi(std::string const &name, DataType &&data)
Definition pdi.hpp:146
void expose_to_pdi(std::string const &name, DataType &&data)
Definition pdi.hpp:152