alpaka
Abstraction Library for Parallel Kernel Acceleration
Loading...
Searching...
No Matches
DeviceGlobalUniformCudaHipBuiltIn.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/dev/DevUniformCudaHipRt.hpp"
8#include "alpaka/mem/global/Traits.hpp"
9#include "alpaka/mem/view/ViewPlainPtr.hpp"
10#include "alpaka/queue/cuda-hip/QueueUniformCudaHipRt.hpp"
11
12#if defined(ALPAKA_ACC_GPU_CUDA_ENABLED) || defined(ALPAKA_ACC_GPU_HIP_ENABLED)
13
14# if defined(ALPAKA_ACC_GPU_CUDA_ENABLED)
15# include "alpaka/core/ApiCudaRt.hpp"
16# endif
17
18# if defined(ALPAKA_ACC_GPU_HIP_ENABLED)
19# include "alpaka/core/ApiHipRt.hpp"
20# endif
21
22namespace alpaka
23{
24
25 namespace detail
26 {
27 template<typename T>
28 struct DevGlobalTrait<TagGpuCudaRt, T>
29 {
30 // CUDA implementation
31 using Type = detail::DevGlobalImplGeneric<TagGpuCudaRt, T>;
32 };
33
34 template<typename T>
35 struct DevGlobalTrait<TagGpuHipRt, T>
36 {
37 // HIP/ROCm implementation
38 using Type = detail::DevGlobalImplGeneric<TagGpuHipRt, T>;
39 };
40 } // namespace detail
41
42 // from device to host
43 template<
44 concepts::Tag TTag,
45 typename TApi,
46 bool TBlocking,
47 typename TViewDst,
48 typename TTypeSrc,
49 typename std::enable_if_t<
50# if defined(ALPAKA_ACC_GPU_CUDA_ENABLED)
51 (std::is_same_v<TTag, TagGpuCudaRt> && std::is_same_v<TApi, ApiCudaRt>)
52# else
53 (std::is_same_v<TTag, TagGpuHipRt> && std::is_same_v<TApi, ApiHipRt>)
54# endif
55 ,
56 int>
57 = 0>
58 ALPAKA_FN_HOST auto memcpy(
59 uniform_cuda_hip::detail::QueueUniformCudaHipRt<TApi, TBlocking>& queue,
60 TViewDst& viewDst,
61 alpaka::detail::DevGlobalImplGeneric<TTag, TTypeSrc>& viewSrc)
62 {
63 using Type = std::remove_const_t<std::remove_all_extents_t<TTypeSrc>>;
64 using TypeExt = std::remove_const_t<TTypeSrc>;
65 auto extent = getExtents(viewDst);
66 TypeExt* pMemAcc(nullptr);
67 ALPAKA_UNIFORM_CUDA_HIP_RT_CHECK(
68 TApi::getSymbolAddress(reinterpret_cast<void**>(&pMemAcc), *(const_cast<TypeExt*>(&viewSrc))));
69
70 auto view = alpaka::ViewPlainPtr<
71 DevUniformCudaHipRt<TApi>,
72 Type,
73 alpaka::Dim<decltype(extent)>,
74 alpaka::Idx<decltype(extent)>>(reinterpret_cast<Type*>(pMemAcc), alpaka::getDev(queue), extent);
75 enqueue(queue, createTaskMemcpy(std::forward<TViewDst>(viewDst), view, extent));
76 }
77
78 // from host to device
79 template<
80 concepts::Tag TTag,
81 typename TApi,
82 bool TBlocking,
83 typename TTypeDst,
84 typename TViewSrc,
85 typename std::enable_if_t<
86# if defined(ALPAKA_ACC_GPU_CUDA_ENABLED)
87 (std::is_same_v<TTag, TagGpuCudaRt> && std::is_same_v<TApi, ApiCudaRt>)
88# else
89 (std::is_same_v<TTag, TagGpuHipRt> && std::is_same_v<TApi, ApiHipRt>)
90# endif
91 ,
92 int>
93 = 0>
94 ALPAKA_FN_HOST auto memcpy(
95 uniform_cuda_hip::detail::QueueUniformCudaHipRt<TApi, TBlocking>& queue,
96 alpaka::detail::DevGlobalImplGeneric<TTag, TTypeDst>& viewDst,
97 TViewSrc const& viewSrc)
98 {
99 using Type = std::remove_const_t<std::remove_all_extents_t<TTypeDst>>;
100 using TypeExt = std::remove_const_t<TTypeDst>;
101 auto extent = getExtents(viewSrc);
102 Type* pMemAcc(nullptr);
103 ALPAKA_UNIFORM_CUDA_HIP_RT_CHECK(
104 TApi::getSymbolAddress(reinterpret_cast<void**>(&pMemAcc), *(const_cast<TypeExt*>(&viewDst))));
105
106 auto view = alpaka::ViewPlainPtr<
107 DevUniformCudaHipRt<TApi>,
108 Type,
109 alpaka::Dim<decltype(extent)>,
110 alpaka::Idx<decltype(extent)>>(reinterpret_cast<Type*>(pMemAcc), alpaka::getDev(queue), extent);
111 enqueue(queue, createTaskMemcpy(std::forward<decltype(view)>(view), viewSrc, extent));
112 }
113
114 // from device to host
115 template<
116 concepts::Tag TTag,
117 typename TApi,
118 bool TBlocking,
119 typename TViewDst,
120 typename TTypeSrc,
121 typename TExtent,
122 typename std::enable_if_t<
123# if defined(ALPAKA_ACC_GPU_CUDA_ENABLED)
124 (std::is_same_v<TTag, TagGpuCudaRt> && std::is_same_v<TApi, ApiCudaRt>)
125# else
126 (std::is_same_v<TTag, TagGpuHipRt> && std::is_same_v<TApi, ApiHipRt>)
127# endif
128 ,
129 int>
130 = 0>
131 ALPAKA_FN_HOST auto memcpy(
132 uniform_cuda_hip::detail::QueueUniformCudaHipRt<TApi, TBlocking>& queue,
133 TViewDst& viewDst,
134 alpaka::detail::DevGlobalImplGeneric<TTag, TTypeSrc>& viewSrc,
135 TExtent extent)
136 {
137 using Type = std::remove_const_t<std::remove_all_extents_t<TTypeSrc>>;
138 using TypeExt = std::remove_const_t<TTypeSrc>;
139 Type* pMemAcc(nullptr);
140 ALPAKA_UNIFORM_CUDA_HIP_RT_CHECK(
141 TApi::getSymbolAddress(reinterpret_cast<void**>(&pMemAcc), *(const_cast<TypeExt*>(&viewSrc))));
142
143 auto view = alpaka::ViewPlainPtr<DevUniformCudaHipRt<TApi>, Type, alpaka::Dim<TExtent>, alpaka::Idx<TExtent>>(
144 reinterpret_cast<Type*>(pMemAcc),
145 alpaka::getDev(queue),
146 extent);
147 enqueue(queue, createTaskMemcpy(std::forward<TViewDst>(viewDst), view, extent));
148 }
149
150 // from host to device
151 template<
152 concepts::Tag TTag,
153 typename TApi,
154 bool TBlocking,
155 typename TTypeDst,
156 typename TViewSrc,
157 typename TExtent,
158 typename std::enable_if_t<
159# if defined(ALPAKA_ACC_GPU_CUDA_ENABLED)
160 (std::is_same_v<TTag, TagGpuCudaRt> && std::is_same_v<TApi, ApiCudaRt>)
161# else
162 (std::is_same_v<TTag, TagGpuHipRt> && std::is_same_v<TApi, ApiHipRt>)
163# endif
164 ,
165 int>
166 = 0>
167 ALPAKA_FN_HOST auto memcpy(
168 uniform_cuda_hip::detail::QueueUniformCudaHipRt<TApi, TBlocking>& queue,
169 alpaka::detail::DevGlobalImplGeneric<TTag, TTypeDst>& viewDst,
170 TViewSrc const& viewSrc,
171 TExtent extent)
172 {
173 using Type = std::remove_const_t<std::remove_all_extents_t<TTypeDst>>;
174 using TypeExt = std::remove_const_t<TTypeDst>;
175 Type* pMemAcc(nullptr);
176 ALPAKA_UNIFORM_CUDA_HIP_RT_CHECK(
177 TApi::getSymbolAddress(reinterpret_cast<void**>(&pMemAcc), *(const_cast<TypeExt*>(&viewDst))));
178
179 auto view = alpaka::ViewPlainPtr<DevUniformCudaHipRt<TApi>, Type, alpaka::Dim<TExtent>, alpaka::Idx<TExtent>>(
180 reinterpret_cast<Type*>(pMemAcc),
181 alpaka::getDev(queue),
182 extent);
183 enqueue(queue, createTaskMemcpy(std::forward<decltype(view)>(view), viewSrc, extent));
184 }
185} // namespace alpaka
186
187#endif
ALPAKA_UNIFORM_CUDA_HIP_RT_CHECK
#define ALPAKA_UNIFORM_CUDA_HIP_RT_CHECK(cmd)
CUDA/HIP runtime error checking with log and exception.
Definition UniformCudaHip.hpp:105
alpaka::DevUniformCudaHipRt
The CUDA/HIP RT device handle.
Definition DevUniformCudaHipRt.hpp:62
alpaka::uniform_cuda_hip::detail::QueueUniformCudaHipRt
The CUDA/HIP RT queue.
Definition QueueUniformCudaHipRt.hpp:115
ALPAKA_FN_HOST
#define ALPAKA_FN_HOST
Definition Common.hpp:40
alpaka
The alpaka accelerator library.
Definition AccCpuOmp2Blocks.hpp:52
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:312
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