alpaka
Abstraction Library for Parallel Kernel Acceleration
DeviceGlobalCpu.hpp
Go to the documentation of this file.
1 /* Copyright 2024 Aurora Perego
2  * SPDX-License-Identifier: MPL-2.0
3  */
4 
5 #pragma once
6 
7 #include "alpaka/mem/buf/cpu/Copy.hpp"
8 #include "alpaka/mem/global/Traits.hpp"
9 #include "alpaka/mem/view/ViewPlainPtr.hpp"
10 
11 #include <type_traits>
12 
13 // memcpy specialization for device global variables
14 namespace alpaka
15 {
16 
17  namespace detail
18  {
19  template<typename T>
20  struct DevGlobalTrait<TagCpuOmp2Blocks, T>
21  {
22  using Type = detail::DevGlobalImplGeneric<TagCpuOmp2Blocks, T>;
23  };
24 
25  template<typename T>
26  struct DevGlobalTrait<TagCpuOmp2Threads, T>
27  {
28  using Type = detail::DevGlobalImplGeneric<TagCpuOmp2Threads, T>;
29  };
30 
31  template<typename T>
32  struct DevGlobalTrait<TagCpuSerial, T>
33  {
34  using Type = detail::DevGlobalImplGeneric<TagCpuSerial, T>;
35  };
36 
37  template<typename T>
38  struct DevGlobalTrait<TagCpuTbbBlocks, T>
39  {
40  using Type = detail::DevGlobalImplGeneric<TagCpuTbbBlocks, T>;
41  };
42 
43  template<typename T>
44  struct DevGlobalTrait<TagCpuThreads, T>
45  {
46  using Type = detail::DevGlobalImplGeneric<TagCpuThreads, T>;
47  };
48  } // namespace detail
49 
50  template<
51  concepts::Tag TTag,
52  typename TViewSrc,
53  typename TTypeDst,
54  typename TQueue,
55  typename std::enable_if_t<
56  std::is_same_v<TTag, TagCpuOmp2Blocks> || std::is_same_v<TTag, TagCpuOmp2Threads>
57  || std::is_same_v<TTag, TagCpuSerial> || std::is_same_v<TTag, TagCpuTbbBlocks>
58  || std::is_same_v<TTag, TagCpuThreads>,
59  int>
60  = 0>
61  ALPAKA_FN_HOST auto memcpy(
62  TQueue& queue,
63  alpaka::detail::DevGlobalImplGeneric<TTag, TTypeDst>& viewDst,
64  TViewSrc const& viewSrc) -> void
65  {
66  using Type = std::remove_const_t<std::remove_all_extents_t<TTypeDst>>;
67  auto extent = getExtents(viewSrc);
68  auto view = alpaka::ViewPlainPtr<DevCpu, Type, alpaka::Dim<decltype(extent)>, alpaka::Idx<decltype(extent)>>(
69  reinterpret_cast<Type*>(const_cast<std::remove_const_t<TTypeDst>*>(&viewDst)),
70  alpaka::getDev(queue),
71  extent);
72  enqueue(queue, createTaskMemcpy(std::forward<decltype(view)>(view), viewSrc, extent));
73  }
74 
75  template<
76  concepts::Tag TTag,
77  typename TTypeSrc,
78  typename TViewDstFwd,
79  typename TQueue,
80  typename std::enable_if_t<
81  std::is_same_v<TTag, TagCpuOmp2Blocks> || std::is_same_v<TTag, TagCpuOmp2Threads>
82  || std::is_same_v<TTag, TagCpuSerial> || std::is_same_v<TTag, TagCpuTbbBlocks>
83  || std::is_same_v<TTag, TagCpuThreads>,
84  int>
85  = 0>
86  ALPAKA_FN_HOST auto memcpy(
87  TQueue& queue,
88  TViewDstFwd&& viewDst,
89  alpaka::detail::DevGlobalImplGeneric<TTag, TTypeSrc>& viewSrc) -> void
90  {
91  using Type = std::remove_all_extents_t<TTypeSrc>;
92  auto extent = getExtents(viewDst);
93  auto view = alpaka::ViewPlainPtr<DevCpu, Type, alpaka::Dim<decltype(extent)>, alpaka::Idx<decltype(extent)>>(
94  reinterpret_cast<Type*>(&viewSrc),
95  alpaka::getDev(queue),
96  extent);
97  enqueue(queue, createTaskMemcpy(std::forward<TViewDstFwd>(viewDst), view, extent));
98  }
99 
100  template<
101  concepts::Tag TTag,
102  typename TExtent,
103  typename TViewSrc,
104  typename TTypeDst,
105  typename TQueue,
106  typename std::enable_if_t<
107  std::is_same_v<TTag, TagCpuOmp2Blocks> || std::is_same_v<TTag, TagCpuOmp2Threads>
108  || std::is_same_v<TTag, TagCpuSerial> || std::is_same_v<TTag, TagCpuTbbBlocks>
109  || std::is_same_v<TTag, TagCpuThreads>,
110  int>
111  = 0>
112  ALPAKA_FN_HOST auto memcpy(
113  TQueue& queue,
114  alpaka::detail::DevGlobalImplGeneric<TTag, TTypeDst>& viewDst,
115  TViewSrc const& viewSrc,
116  TExtent const& extent) -> void
117  {
118  using Type = std::remove_const_t<std::remove_all_extents_t<TTypeDst>>;
119  auto view = alpaka::ViewPlainPtr<DevCpu, Type, alpaka::Dim<TExtent>, alpaka::Idx<TExtent>>(
120  reinterpret_cast<Type*>(const_cast<std::remove_const_t<TTypeDst>*>(&viewDst)),
121  alpaka::getDev(queue),
122  extent);
123  enqueue(queue, createTaskMemcpy(std::forward<decltype(view)>(view), viewSrc, extent));
124  }
125 
126  template<
127  concepts::Tag TTag,
128  typename TExtent,
129  typename TTypeSrc,
130  typename TViewDstFwd,
131  typename TQueue,
132  typename std::enable_if_t<
133  std::is_same_v<TTag, TagCpuOmp2Blocks> || std::is_same_v<TTag, TagCpuOmp2Threads>
134  || std::is_same_v<TTag, TagCpuSerial> || std::is_same_v<TTag, TagCpuTbbBlocks>
135  || std::is_same_v<TTag, TagCpuThreads>,
136  int>
137  = 0>
138  ALPAKA_FN_HOST auto memcpy(
139  TQueue& queue,
140  TViewDstFwd&& viewDst,
141  alpaka::detail::DevGlobalImplGeneric<TTag, TTypeSrc>& viewSrc,
142  TExtent const& extent) -> void
143  {
144  using Type = std::remove_all_extents_t<TTypeSrc>;
145  auto view = alpaka::ViewPlainPtr<DevCpu, Type, alpaka::Dim<TExtent>, alpaka::Idx<TExtent>>(
146  reinterpret_cast<Type*>(&viewSrc),
147  alpaka::getDev(queue),
148  extent);
149  enqueue(queue, createTaskMemcpy(std::forward<TViewDstFwd>(viewDst), view, extent));
150  }
151 } // namespace alpaka
alpaka::DevCpu
The CPU device handle.
Definition: DevCpu.hpp:56
ALPAKA_FN_HOST
#define ALPAKA_FN_HOST
Definition: Common.hpp:40
alpaka::concepts::Tag
concept Tag
Definition: Tag.hpp:46
alpaka
The alpaka accelerator library.
Definition: AccCpuOmp2Blocks.hpp:49
alpaka::Idx
typename trait::IdxType< T >::type Idx
Definition: Traits.hpp:29
alpaka::memcpy
ALPAKA_FN_HOST auto memcpy(TQueue &queue, alpaka::detail::DevGlobalImplGeneric< TTag, TTypeDst > &viewDst, TViewSrc const &viewSrc) -> void
Definition: DeviceGlobalCpu.hpp:61
alpaka::createTaskMemcpy
ALPAKA_FN_HOST auto createTaskMemcpy(TViewDstFwd &&viewDst, TViewSrc const &viewSrc, TExtent const &extent)
Creates a memory copy task.
Definition: Traits.hpp:253
alpaka::getExtents
ALPAKA_NO_HOST_ACC_WARNING ALPAKA_FN_HOST_ACC auto getExtents(T const &object) -> Vec< Dim< T >, Idx< T >>
Definition: Traits.hpp:59
alpaka::getDev
ALPAKA_FN_HOST auto getDev(T const &t)
Definition: Traits.hpp:68
alpaka::enqueue
ALPAKA_FN_HOST auto enqueue(TQueue &queue, TTask &&task) -> void
Queues the given task in the given queue.
Definition: Traits.hpp:47
alpaka::Dim
typename trait::DimType< T >::type Dim
The dimension type trait alias template to remove the ::type.
Definition: Traits.hpp:19
alpaka::ViewPlainPtr
The memory view to wrap plain pointers.
Definition: ViewPlainPtr.hpp:24