Branch data Line data Source code
1 : : !{\src2tex{textfont=tt}}
2 : : !!****f* m_abi_linalg/abi_xaxpy
3 : : !! NAME
4 : : !! abi_xaxpy
5 : : !!
6 : : !! FUNCTION
7 : : !! abi_xaxpy is the generic function for BLAS-1 AXPY (y = alpha * x + y)
8 : : !! with optional GPU dispatch.
9 : : !!
10 : : !! COPYRIGHT
11 : : !! Copyright (C) 2001-2026 ABINIT group
12 : : !! This file is distributed under the terms of the
13 : : !! GNU General Public License, see ~abinit/COPYING
14 : : !! or http://www.gnu.org/copyleft/gpl.txt .
15 : : !!
16 : : !! SOURCE
17 : :
18 : : !!***
19 : :
20 : : !----------------------------------------------------------------------
21 : :
22 : : !!****f* m_abi_linalg/abi_daxpy
23 : : !! NAME
24 : : !! abi_daxpy
25 : : !!
26 : : !! FUNCTION
27 : : !! Real AXPY: y = alpha * x + y, with real(dp) alpha and real(dp) arrays.
28 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=1.
29 : : !!
30 : : !! INPUTS
31 : : !!
32 : : !! SOURCE
33 : :
34 : 0 : subroutine abi_daxpy(n, alpha, x, incx, y, incy, gpu_option)
35 : :
36 : : !Arguments ------------------------------------
37 : : integer, intent(in) :: n
38 : : real(dp), intent(in) :: alpha
39 : : real(dp), target, intent(in) :: x(*)
40 : : integer, intent(in) :: incx
41 : : real(dp), target, intent(inout) :: y(*)
42 : : integer, intent(in) :: incy
43 : : !Optionals -----------------------------------
44 : : integer, intent(in), optional :: gpu_option
45 : :
46 : : !Local variables-------------------------------
47 : : integer :: gpu_option_
48 : :
49 : : ! *********************************************************************
50 : :
51 [ # # ]: 0 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
52 : :
53 : 0 : call abi_d2zaxpy(n, dcmplx(alpha, 0.0_dp), x, incx, y, incy, x_cplx=1, gpu_option=gpu_option_)
54 : :
55 : 0 : end subroutine abi_daxpy
56 : : !!***
57 : :
58 : : !----------------------------------------------------------------------
59 : :
60 : : !!****f* m_abi_linalg/abi_daxpy_2d
61 : : !! NAME
62 : : !! abi_daxpy_2d
63 : : !!
64 : : !! FUNCTION
65 : : !! Real AXPY: y = alpha * x + y, with real(dp) alpha and real(dp) arrays.
66 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=1.
67 : : !!
68 : : !! INPUTS
69 : : !!
70 : : !! SOURCE
71 : :
72 [ # # # # ]: 0 : subroutine abi_daxpy_2d(n, alpha, x, incx, y, incy, gpu_option)
73 : :
74 : : !Arguments ------------------------------------
75 : : integer, intent(in) :: n
76 : : real(dp), intent(in) :: alpha
77 : : real(dp), target, intent(in) :: x(:,:)
78 : : integer, intent(in) :: incx
79 : : real(dp), target, intent(inout) :: y(:,:)
80 : : integer, intent(in) :: incy
81 : : !Optionals -----------------------------------
82 : : integer, intent(in), optional :: gpu_option
83 : :
84 : : !Local variables-------------------------------
85 : : integer :: gpu_option_
86 : :
87 : : ! *********************************************************************
88 : :
89 [ # # ]: 0 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
90 : :
91 [ # # # # : 0 : call abi_d2zaxpy(n, dcmplx(alpha, 0.0_dp), x, incx, y, incy, x_cplx=1, gpu_option=gpu_option_)
# # # # #
# # # # #
# # # # #
# # # #
# ]
92 : :
93 : 0 : end subroutine abi_daxpy_2d
94 : : !!***
95 : :
96 : : !----------------------------------------------------------------------
97 : :
98 : : !!****f* m_abi_linalg/abi_d2zaxpy_2d
99 : : !! NAME
100 : : !! abi_d2zaxpy_2d
101 : : !!
102 : : !! FUNCTION
103 : : !! Real AXPY: y = alpha * x + y, with real(dp) alpha and real(dp) arrays.
104 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=1.
105 : : !!
106 : : !! INPUTS
107 : : !!
108 : : !! SOURCE
109 : :
110 [ - + - + ]: 412274 : subroutine abi_d2zaxpy_2d(n, alpha, x, incx, y, incy, x_cplx, gpu_option)
111 : :
112 : : !Arguments ------------------------------------
113 : : integer, intent(in) :: n
114 : : complex(dp), intent(in) :: alpha
115 : : real(dp), target, intent(in) :: x(:,:)
116 : : integer, intent(in) :: incx
117 : : real(dp), target, intent(inout) :: y(:,:)
118 : : integer, intent(in) :: incy
119 : : !Optionals -----------------------------------
120 : : integer, intent(in), optional :: x_cplx, gpu_option
121 : :
122 : : !Local variables-------------------------------
123 : : integer :: cplx_,gpu_option_
124 : :
125 : : ! *********************************************************************
126 : :
127 [ + - ]: 412274 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
128 [ + - ]: 412274 : cplx_=1 ; if(PRESENT(x_cplx)) cplx_ = x_cplx
129 : :
130 [ - + - - : 412274 : call abi_d2zaxpy(n, alpha, x, incx, y, incy, x_cplx=cplx_, gpu_option=gpu_option_)
- - - - -
+ - - - -
- - - + -
+ - - -
- ]
131 : :
132 : 412274 : end subroutine abi_d2zaxpy_2d
133 : : !!***
134 : :
135 : :
136 : : !----------------------------------------------------------------------
137 : :
138 : : !!****f* m_abi_linalg/abi_d2zaxpy_5d
139 : : !! NAME
140 : : !! abi_d2zaxpy_5d
141 : : !!
142 : : !! FUNCTION
143 : : !! Real AXPY: y = alpha * x + y, with real(dp) alpha and real(dp) arrays.
144 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=1.
145 : : !!
146 : : !! INPUTS
147 : : !!
148 : : !! SOURCE
149 : :
150 [ # # # # ]: 0 : subroutine abi_d2zaxpy_5d(n, alpha, x, incx, y, incy, x_cplx, gpu_option)
151 : :
152 : : !Arguments ------------------------------------
153 : : integer, intent(in) :: n
154 : : complex(dp), intent(in) :: alpha
155 : : real(dp), target, intent(in) :: x(:,:,:,:,:)
156 : : integer, intent(in) :: incx
157 : : real(dp), target, intent(inout) :: y(:,:,:,:,:)
158 : : integer, intent(in) :: incy
159 : : !Optionals -----------------------------------
160 : : integer, intent(in), optional :: x_cplx, gpu_option
161 : :
162 : : !Local variables-------------------------------
163 : : integer :: cplx_,gpu_option_
164 : :
165 : : ! *********************************************************************
166 : :
167 [ # # ]: 0 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
168 [ # # ]: 0 : cplx_=1 ; if(PRESENT(x_cplx)) cplx_ = x_cplx
169 : :
170 [ # # # # : 0 : call abi_d2zaxpy(n, alpha, x, incx, y, incy, x_cplx=cplx_, gpu_option=gpu_option_)
# # # # #
# # # # #
# # # # #
# # # # #
# # # # #
# # # # #
# # # # #
# # # ]
171 : :
172 : 0 : end subroutine abi_d2zaxpy_5d
173 : : !!***
174 : :
175 : :
176 : : !----------------------------------------------------------------------
177 : :
178 : : !!****f* m_abi_linalg/abi_d2zaxpy
179 : : !! NAME
180 : : !! abi_d2zaxpy
181 : : !!
182 : : !! FUNCTION
183 : : !! Real AXPY: y = alpha * x + y, with real(dp) alpha and real(dp) arrays.
184 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=1.
185 : : !!
186 : : !! INPUTS
187 : : !!
188 : : !! SOURCE
189 : :
190 : 412274 : subroutine abi_d2zaxpy(n, alpha, x, incx, y, incy, x_cplx, gpu_option)
191 : :
192 : : !Arguments ------------------------------------
193 : : integer, intent(in) :: n
194 : : complex(dp), intent(in) :: alpha
195 : : real(dp), target, intent(in) :: x(*)
196 : : integer, intent(in) :: incx
197 : : real(dp), target, intent(inout) :: y(*)
198 : : integer, intent(in) :: incy
199 : : !Optionals -----------------------------------
200 : : integer, intent(in), optional :: x_cplx, gpu_option
201 : :
202 : : !Local variables-------------------------------
203 : : integer :: cplx_,gpu_option_
204 : :
205 : : ! *********************************************************************
206 : :
207 [ + - ]: 412274 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
208 [ + - ]: 412274 : cplx_=1 ; if(PRESENT(x_cplx)) cplx_ = x_cplx
209 : :
210 : : #if defined(DEBUG_VERBOSE) && defined(HAVE_OPENMP_OFFLOAD)
211 : : if ( gpu_option_ == ABI_GPU_OPENMP ) then
212 : : ABI_CHECK(xomp_target_is_present(c_loc(x)), "Array isn't mapped on GPU")
213 : : ABI_CHECK(xomp_target_is_present(c_loc(y)), "Array isn't mapped on GPU")
214 : : end if
215 : : #endif
216 : :
217 [ - + ]: 412274 : if(gpu_option_/=ABI_GPU_DISABLED) then
218 [ # # ]: 0 : if(gpu_option_==ABI_GPU_OPENMP) then
219 : : #ifdef HAVE_OPENMP_OFFLOAD
220 : : !$OMP TARGET DATA USE_DEVICE_ADDR(x,y)
221 : : call abi_gpu_xaxpy_cptr(cplx_, n, alpha, c_loc(x), incx, c_loc(y), incy)
222 : : !$OMP END TARGET DATA
223 : : #endif
224 : : else
225 : 0 : call abi_gpu_xaxpy_cptr(cplx_, n, alpha, c_loc(x), incx, c_loc(y), incy)
226 : : end if
227 : : else
228 [ + - ]: 412274 : if(cplx_==1) then
229 : 412274 : call daxpy(n, alpha, x, incx, y, incy)
230 : : else
231 : 0 : call zaxpy(n, alpha, x, incx, y, incy)
232 : : end if
233 : : end if
234 : :
235 : 412274 : end subroutine abi_d2zaxpy
236 : : !!***
237 : :
238 : : !----------------------------------------------------------------------
239 : :
240 : : !!****f* m_abi_linalg/abi_zaxpy
241 : : !! NAME
242 : : !! abi_zaxpy
243 : : !!
244 : : !! FUNCTION
245 : : !! Complex AXPY: y = alpha * x + y, with complex(dp) alpha and complex(dp) arrays.
246 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=2.
247 : : !!
248 : : !! INPUTS
249 : : !!
250 : : !! SOURCE
251 : :
252 : 673983 : subroutine abi_zaxpy(n, alpha, x, incx, y, incy, gpu_option)
253 : :
254 : : !Arguments ------------------------------------
255 : : integer, intent(in) :: n
256 : : complex(dp), intent(in) :: alpha
257 : : complex(dp), target, intent(in) :: x(*)
258 : : integer, intent(in) :: incx
259 : : complex(dp), target, intent(inout) :: y(*)
260 : : integer, intent(in) :: incy
261 : : !Optionals -----------------------------------
262 : : integer, intent(in), optional :: gpu_option
263 : :
264 : : !Local variables-------------------------------
265 : : integer :: gpu_option_
266 : :
267 : : ! *********************************************************************
268 : :
269 [ + - ]: 673983 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
270 : :
271 : : #if defined(DEBUG_VERBOSE) && defined(HAVE_OPENMP_OFFLOAD)
272 : : if ( gpu_option_ == ABI_GPU_OPENMP ) then
273 : : ABI_CHECK(xomp_target_is_present(c_loc(x)), "Array isn't mapped on GPU")
274 : : ABI_CHECK(xomp_target_is_present(c_loc(y)), "Array isn't mapped on GPU")
275 : : end if
276 : : #endif
277 : :
278 [ - + ]: 673983 : if(gpu_option_/=ABI_GPU_DISABLED) then
279 [ # # ]: 0 : if(gpu_option_==ABI_GPU_OPENMP) then
280 : : #ifdef HAVE_OPENMP_OFFLOAD
281 : : !$OMP TARGET DATA USE_DEVICE_ADDR(x,y)
282 : : call abi_gpu_xaxpy_cptr(2, n, alpha, c_loc(x), incx, c_loc(y), incy)
283 : : !$OMP END TARGET DATA
284 : : #endif
285 : : else
286 : 0 : call abi_gpu_xaxpy_cptr(2, n, alpha, c_loc(x), incx, c_loc(y), incy)
287 : : end if
288 : : else
289 : 673983 : call zaxpy(n, alpha, x, incx, y, incy)
290 : : end if
291 : :
292 : 673983 : end subroutine abi_zaxpy
293 : : !!***
294 : :
295 : : !----------------------------------------------------------------------
296 : :
297 : : !!****f* m_abi_linalg/abi_zaxpy_2d
298 : : !! NAME
299 : : !! abi_zaxpy_2d
300 : : !!
301 : : !! FUNCTION
302 : : !! Complex AXPY: y = alpha * x + y, with complex(dp) alpha and complex(dp) arrays.
303 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=2.
304 : : !!
305 : : !! INPUTS
306 : : !!
307 : : !! SOURCE
308 : :
309 [ - + - + ]: 673983 : subroutine abi_zaxpy_2d(n, alpha, x, incx, y, incy, gpu_option)
310 : :
311 : : !Arguments ------------------------------------
312 : : integer, intent(in) :: n
313 : : complex(dp), intent(in) :: alpha
314 : : complex(dp), target, intent(in) :: x(:,:)
315 : : integer, intent(in) :: incx
316 : : complex(dp), target, intent(inout) :: y(:,:)
317 : : integer, intent(in) :: incy
318 : : !Optionals -----------------------------------
319 : : integer, intent(in), optional :: gpu_option
320 : :
321 : : !Local variables-------------------------------
322 : : integer :: gpu_option_
323 : :
324 : : ! *********************************************************************
325 : :
326 [ + - ]: 673983 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
327 : :
328 [ - + - - : 673983 : call abi_zaxpy(n, alpha, x, incx, y, incy, gpu_option=gpu_option_)
- - - - -
+ - - - -
- - - + -
+ - - -
- ]
329 : :
330 : 673983 : end subroutine abi_zaxpy_2d
331 : : !!***
332 : :
333 : : !----------------------------------------------------------------------
334 : :
335 : : !!****f* m_abi_linalg/abi_zaxpy_3d
336 : : !! NAME
337 : : !! abi_zaxpy_3d
338 : : !!
339 : : !! FUNCTION
340 : : !! Complex AXPY: y = alpha * x + y, with complex(dp) alpha and complex(dp) arrays.
341 : : !! On GPU, calls abi_gpu_xaxpy_cptr with cplx=2.
342 : : !!
343 : : !! INPUTS
344 : : !!
345 : : !! SOURCE
346 : :
347 [ # # # # ]: 0 : subroutine abi_zaxpy_3d(n, alpha, x, incx, y, incy, gpu_option)
348 : :
349 : : !Arguments ------------------------------------
350 : : integer, intent(in) :: n
351 : : complex(dp), intent(in) :: alpha
352 : : complex(dp), target, intent(in) :: x(:,:,:)
353 : : integer, intent(in) :: incx
354 : : complex(dp), target, intent(inout) :: y(:,:,:)
355 : : integer, intent(in) :: incy
356 : : !Optionals -----------------------------------
357 : : integer, intent(in), optional :: gpu_option
358 : :
359 : : !Local variables-------------------------------
360 : : integer :: gpu_option_
361 : :
362 : : ! *********************************************************************
363 : :
364 [ # # ]: 0 : gpu_option_=ABI_GPU_DISABLED ; if(PRESENT(gpu_option)) gpu_option_ = gpu_option
365 : :
366 [ # # # # : 0 : call abi_zaxpy(n, alpha, x, incx, y, incy, gpu_option=gpu_option_)
# # # # #
# # # # #
# # # # #
# # # # #
# # # # #
# ]
367 : :
368 : 0 : end subroutine abi_zaxpy_3d
369 : : !!***
|
