Neko 1.99.2
A portable framework for high-order spectral element flow simulations
Loading...
Searching...
No Matches
scalar_stats.f90
Go to the documentation of this file.
1! Copyright (c) 2025, The Neko Authors
2! All rights reserved.
3!
4! Redistribution and use in source and binary forms, with or without
5! modification, are permitted provided that the following conditions
6! are met:
7!
8! * Redistributions of source code must retain the above copyright
9! notice, this list of conditions and the following disclaimer.
10!
11! * Redistributions in binary form must reproduce the above
12! copyright notice, this list of conditions and the following
13! disclaimer in the documentation and/or other materials provided
14! with the distribution.
15!
16! * Neither the name of the authors nor the names of its
17! contributors may be used to endorse or promote products derived
18! from this software without specific prior written permission.
19!
20! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21! "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22! LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23! FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24! COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
25! INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
26! BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
27! LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
28! CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29! LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
30! ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
31! POSSIBILITY OF SUCH DAMAGE.
32!
36module scalar_stats
37 use mean_field, only : mean_field_t
38 use device_math, only : device_col3, device_col2, device_cfill, &
39 device_invcol2, device_addcol3
40 use num_types, only : rp
41 use math, only : invers2, col2, addcol3, col3, copy, subcol3
42 use operators, only : opgrad
43 use coefs, only : coef_t
44 use field, only : field_t
45 use field_list, only : field_list_t
46 use stats_quant, only : stats_quant_t
47 use device, only : device_memcpy, host_to_device, device_to_host
48 use neko_config, only : neko_bcknd_device
49 use utils, only : neko_warning
50 implicit none
51 private
52
53 type, public, extends(stats_quant_t) :: scalar_stats_t
55 type(field_t) :: stats_ss
56 type(field_t) :: stats_uiuj
57 type(field_t) :: stats_work
58
60 type(field_t), pointer :: s
61 type(field_t), pointer :: u
62 type(field_t), pointer :: v
63 type(field_t), pointer :: w
64 type(field_t), pointer :: p
65
66 type(mean_field_t) :: s_mean
67
68 type(mean_field_t) :: us
69 type(mean_field_t) :: vs
70 type(mean_field_t) :: ws
71
72 type(mean_field_t) :: ss
73 type(mean_field_t) :: sss
74 type(mean_field_t) :: ssss
75
76 type(mean_field_t) :: uss
77 type(mean_field_t) :: vss
78 type(mean_field_t) :: wss
79
80 type(mean_field_t) :: uus
81 type(mean_field_t) :: vvs
82 type(mean_field_t) :: wws
83 type(mean_field_t) :: uvs
84 type(mean_field_t) :: uws
85 type(mean_field_t) :: vws
86
87 type(mean_field_t) :: ps
88
89 type(mean_field_t) :: pdsdx
90 type(mean_field_t) :: pdsdy
91 type(mean_field_t) :: pdsdz
92
93 type(mean_field_t) :: udsdx
94 type(mean_field_t) :: udsdy
95 type(mean_field_t) :: udsdz
96 type(mean_field_t) :: vdsdx
97 type(mean_field_t) :: vdsdy
98 type(mean_field_t) :: vdsdz
99 type(mean_field_t) :: wdsdx
100 type(mean_field_t) :: wdsdy
101 type(mean_field_t) :: wdsdz
102
103 type(mean_field_t) :: sdudx
104 type(mean_field_t) :: sdudy
105 type(mean_field_t) :: sdudz
106 type(mean_field_t) :: sdvdx
107 type(mean_field_t) :: sdvdy
108 type(mean_field_t) :: sdvdz
109 type(mean_field_t) :: sdwdx
110 type(mean_field_t) :: sdwdy
111 type(mean_field_t) :: sdwdz
112
113 type(mean_field_t) :: ess
114 type(mean_field_t) :: eus
115 type(mean_field_t) :: evs
116 type(mean_field_t) :: ews
117
119 type(field_t) :: dsdx
120 type(field_t) :: dsdy
121 type(field_t) :: dsdz
122 type(field_t) :: dudx
123 type(field_t) :: dudy
124 type(field_t) :: dudz
125 type(field_t) :: dvdx
126 type(field_t) :: dvdy
127 type(field_t) :: dvdz
128 type(field_t) :: dwdx
129 type(field_t) :: dwdy
130 type(field_t) :: dwdz
131
133 type(coef_t), pointer :: coef
135 integer :: n_stats = 42
138 character(5) :: stat_set
141 type(field_list_t) :: stat_fields
142 contains
144 procedure, pass(this) :: init => scalar_stats_init
146 procedure, pass(this) :: free => scalar_stats_free
148 procedure, pass(this) :: update => scalar_stats_update
150 procedure, pass(this) :: reset => scalar_stats_reset
151 ! Convert computed weak gradients to strong.
152 procedure, pass(this) :: make_strong_grad => scalar_stats_make_strong_grad
153 end type scalar_stats_t
154
155contains
156
166 subroutine scalar_stats_init(this, coef, s, u, v, w, p, set)
167 class(scalar_stats_t), intent(inout), target:: this
168 type(coef_t), target, optional :: coef
169 type(field_t), target, intent(in) :: s, u, v, w, p
170 character(*), intent(in), optional :: set
171
172 call this%free()
173 this%coef => coef
174
175 this%s => s
176 this%u => u
177 this%v => v
178 this%w => w
179 this%p => p
180
181 if (present(set)) then
182 this%stat_set = trim(set)
183 if (this%stat_set .eq. 'basic') then
184 this%n_stats = 5
185 end if
186 else
187 this%stat_set = 'full'
188 this%n_stats = 42
189 end if
190
191 ! Initialize work fields
192 call this%stats_work%init(this%u%dof, 'stats')
193 call this%stats_ss%init(this%u%dof, 'ss temp')
194
195 ! Initialize mean fields
196 call this%s_mean%init(this%s)
197
198 call this%us%init(this%stats_work, 'us')
199 call this%vs%init(this%stats_work, 'vs')
200 call this%ws%init(this%stats_work, 'ws')
201
202 call this%ss%init(this%stats_ss, 'ss')
203
204 if (this%n_stats .eq. 42) then
205 ! Initialize req. work fields
206 call this%stats_uiuj%init(this%u%dof, 'uiuj temp')
207
208 ! Initialize req. gradient fields
209 call this%dsdx%init(this%u%dof, 'dsdx')
210 call this%dsdy%init(this%u%dof, 'dsdy')
211 call this%dsdz%init(this%u%dof, 'dsdz')
212
213 call this%dudx%init(this%u%dof, 'dudx')
214 call this%dudy%init(this%u%dof, 'dudy')
215 call this%dudz%init(this%u%dof, 'dudz')
216 call this%dvdx%init(this%u%dof, 'dvdx')
217 call this%dvdy%init(this%u%dof, 'dvdy')
218 call this%dvdz%init(this%u%dof, 'dvdz')
219 call this%dwdx%init(this%u%dof, 'dwdx')
220 call this%dwdy%init(this%u%dof, 'dwdy')
221 call this%dwdz%init(this%u%dof, 'dwdz')
222
223 ! Initialize req. mean fields
224 call this%sss%init(this%stats_work, 'sss')
225 call this%ssss%init(this%stats_work, 'ssss')
226
227 call this%uss%init(this%stats_work, 'uss')
228 call this%vss%init(this%stats_work, 'vss')
229 call this%wss%init(this%stats_work, 'wss')
230
231 call this%uus%init(this%stats_work, 'uus')
232 call this%vvs%init(this%stats_work, 'vvs')
233 call this%wws%init(this%stats_work, 'wws')
234 call this%uvs%init(this%stats_work, 'uvs')
235 call this%uws%init(this%stats_work, 'uws')
236 call this%vws%init(this%stats_work, 'vws')
237
238 call this%ps%init(this%stats_work, 'ps')
239
240 call this%pdsdx%init(this%stats_work, 'pdsdx')
241 call this%pdsdy%init(this%stats_work, 'pdsdy')
242 call this%pdsdz%init(this%stats_work, 'pdsdz')
243
244 call this%udsdx%init(this%stats_work, 'udsdx')
245 call this%udsdy%init(this%stats_work, 'udsdy')
246 call this%udsdz%init(this%stats_work, 'udsdz')
247 call this%vdsdx%init(this%stats_work, 'vdsdx')
248 call this%vdsdy%init(this%stats_work, 'vdsdy')
249 call this%vdsdz%init(this%stats_work, 'vdsdz')
250 call this%wdsdx%init(this%stats_work, 'wdsdx')
251 call this%wdsdy%init(this%stats_work, 'wdsdy')
252 call this%wdsdz%init(this%stats_work, 'wdsdz')
253
254 call this%sdudx%init(this%stats_work, 'sdudx')
255 call this%sdudy%init(this%stats_work, 'sdudy')
256 call this%sdudz%init(this%stats_work, 'sdudz')
257 call this%sdvdx%init(this%stats_work, 'sdvdx')
258 call this%sdvdy%init(this%stats_work, 'sdvdy')
259 call this%sdvdz%init(this%stats_work, 'sdvdz')
260 call this%sdwdx%init(this%stats_work, 'sdwdx')
261 call this%sdwdy%init(this%stats_work, 'sdwdy')
262 call this%sdwdz%init(this%stats_work, 'sdwdz')
263
264 call this%ess%init(this%stats_work, 'ess')
265 call this%eus%init(this%stats_work, 'eus')
266 call this%evs%init(this%stats_work, 'evs')
267 call this%ews%init(this%stats_work, 'ews')
268 end if
269
270 allocate(this%stat_fields%items(this%n_stats))
271
272 call this%stat_fields%assign_to_field(1, this%s_mean%mf)
273
274 call this%stat_fields%assign_to_field(2, this%us%mf)
275 call this%stat_fields%assign_to_field(3, this%vs%mf)
276 call this%stat_fields%assign_to_field(4, this%ws%mf)
277
278 call this%stat_fields%assign_to_field(5, this%ss%mf)
279
280 if (this%n_stats .eq. 42) then
281 call this%stat_fields%assign_to_field(6, this%sss%mf)
282 call this%stat_fields%assign_to_field(7, this%ssss%mf)
283
284 call this%stat_fields%assign_to_field(8, this%uss%mf)
285 call this%stat_fields%assign_to_field(9, this%vss%mf)
286 call this%stat_fields%assign_to_field(10, this%wss%mf)
287 call this%stat_fields%assign_to_field(11, this%uus%mf)
288 call this%stat_fields%assign_to_field(12, this%vvs%mf)
289 call this%stat_fields%assign_to_field(13, this%wws%mf)
290 call this%stat_fields%assign_to_field(14, this%uvs%mf)
291 call this%stat_fields%assign_to_field(15, this%uws%mf)
292 call this%stat_fields%assign_to_field(16, this%vws%mf)
293
294 call this%stat_fields%assign_to_field(17, this%ps%mf)
295
296 call this%stat_fields%assign_to_field(18, this%pdsdx%mf)
297 call this%stat_fields%assign_to_field(19, this%pdsdy%mf)
298 call this%stat_fields%assign_to_field(20, this%pdsdz%mf)
299
300 call this%stat_fields%assign_to_field(21, this%udsdx%mf)
301 call this%stat_fields%assign_to_field(22, this%udsdy%mf)
302 call this%stat_fields%assign_to_field(23, this%udsdz%mf)
303 call this%stat_fields%assign_to_field(24, this%vdsdx%mf)
304 call this%stat_fields%assign_to_field(25, this%vdsdy%mf)
305 call this%stat_fields%assign_to_field(26, this%vdsdz%mf)
306 call this%stat_fields%assign_to_field(27, this%wdsdx%mf)
307 call this%stat_fields%assign_to_field(28, this%wdsdy%mf)
308 call this%stat_fields%assign_to_field(29, this%wdsdz%mf)
309
310 call this%stat_fields%assign_to_field(30, this%sdudx%mf)
311 call this%stat_fields%assign_to_field(31, this%sdudy%mf)
312 call this%stat_fields%assign_to_field(32, this%sdudz%mf)
313 call this%stat_fields%assign_to_field(33, this%sdvdx%mf)
314 call this%stat_fields%assign_to_field(34, this%sdvdy%mf)
315 call this%stat_fields%assign_to_field(35, this%sdvdz%mf)
316 call this%stat_fields%assign_to_field(36, this%sdwdx%mf)
317 call this%stat_fields%assign_to_field(37, this%sdwdy%mf)
318 call this%stat_fields%assign_to_field(38, this%sdwdz%mf)
319
320 call this%stat_fields%assign_to_field(39, this%ess%mf)
321 call this%stat_fields%assign_to_field(40, this%eus%mf)
322 call this%stat_fields%assign_to_field(41, this%evs%mf)
323 call this%stat_fields%assign_to_field(42, this%ews%mf)
324 end if
325
326 end subroutine scalar_stats_init
327
330 subroutine scalar_stats_update(this, k)
331 class(scalar_stats_t), intent(inout) :: this
332 real(kind=rp), intent(in) :: k
333 integer :: n
334
335 associate(stats_work => this%stats_work, stats_ss => this%stats_ss, &
336 stats_uiuj => this%stats_uiuj)
337 n = stats_work%dof%size()
338
340 if (neko_bcknd_device .eq. 1) then
341
342 call this%s_mean%update(k)
343
344 call device_col3(stats_work%x_d, this%u%x_d, this%s%x_d, n)
345 call this%us%update(k)
346 call device_col3(stats_work%x_d, this%v%x_d, this%s%x_d, n)
347 call this%vs%update(k)
348 call device_col3(stats_work%x_d, this%w%x_d, this%s%x_d, n)
349 call this%ws%update(k)
350
351 call device_col3(stats_ss%x_d, this%s%x_d, this%s%x_d, n)
352 call this%ss%update(k)
353
354 if (this%n_stats .eq. 5) return
355
356 call device_col3(stats_work%x_d, this%stats_ss%x_d, this%s%x_d, n)
357 call this%sss%update(k)
358 call device_col3(stats_work%x_d, this%stats_ss%x_d, this%stats_ss%x_d, n)
359 call this%ssss%update(k)
360
361 call device_col3(stats_work%x_d, this%u%x_d, this%stats_ss%x_d, n)
362 call this%uss%update(k)
363 call device_col3(stats_work%x_d, this%v%x_d, this%stats_ss%x_d, n)
364 call this%vss%update(k)
365 call device_col3(stats_work%x_d, this%w%x_d, this%stats_ss%x_d, n)
366 call this%wss%update(k)
367
368 call device_col3(stats_uiuj%x_d, this%u%x_d, this%u%x_d, n)
369 call device_col3(stats_work%x_d, stats_uiuj%x_d, this%s%x_d, n)
370 call this%uus%update(k)
371 call device_col3(stats_uiuj%x_d, this%v%x_d, this%v%x_d, n)
372 call device_col3(stats_work%x_d, stats_uiuj%x_d, this%s%x_d, n)
373 call this%vvs%update(k)
374 call device_col3(stats_uiuj%x_d, this%w%x_d, this%w%x_d, n)
375 call device_col3(stats_work%x_d, stats_uiuj%x_d, this%s%x_d, n)
376 call this%wws%update(k)
377 call device_col3(stats_uiuj%x_d, this%u%x_d, this%v%x_d, n)
378 call device_col3(stats_work%x_d, stats_uiuj%x_d, this%s%x_d, n)
379 call this%uvs%update(k)
380 call device_col3(stats_uiuj%x_d, this%u%x_d, this%w%x_d, n)
381 call device_col3(stats_work%x_d, stats_uiuj%x_d, this%s%x_d, n)
382 call this%uws%update(k)
383 call device_col3(stats_uiuj%x_d, this%v%x_d, this%w%x_d, n)
384 call device_col3(stats_work%x_d, stats_uiuj%x_d, this%s%x_d, n)
385 call this%vws%update(k)
386
387 call device_col3(stats_work%x_d, this%p%x_d, this%s%x_d, n)
388 call this%ps%update(k)
389
390 else
391
392 call this%s_mean%update(k)
393
394 call col3(stats_work%x, this%u%x, this%s%x, n)
395 call this%us%update(k)
396 call col3(stats_work%x, this%v%x, this%s%x, n)
397 call this%vs%update(k)
398 call col3(stats_work%x, this%w%x, this%s%x, n)
399 call this%ws%update(k)
400
401 call col3(stats_ss%x, this%s%x, this%s%x, n)
402 call this%ss%update(k)
403
404 if (this%n_stats .eq. 5) return
405
406 call col3(stats_work%x, this%stats_ss%x, this%s%x, n)
407 call this%sss%update(k)
408 call col3(stats_work%x, this%stats_ss%x, this%stats_ss%x, n)
409 call this%ssss%update(k)
410
411 call col3(stats_work%x, this%u%x, this%stats_ss%x, n)
412 call this%uss%update(k)
413 call col3(stats_work%x, this%v%x, this%stats_ss%x, n)
414 call this%vss%update(k)
415 call col3(stats_work%x, this%w%x, this%stats_ss%x, n)
416 call this%wss%update(k)
417
418 call col3(stats_uiuj%x, this%u%x, this%u%x, n)
419 call col3(stats_work%x, stats_uiuj%x, this%s%x, n)
420 call this%uus%update(k)
421 call col3(stats_uiuj%x, this%v%x, this%v%x, n)
422 call col3(stats_work%x, stats_uiuj%x, this%s%x, n)
423 call this%vvs%update(k)
424 call col3(stats_uiuj%x, this%w%x, this%w%x, n)
425 call col3(stats_work%x, stats_uiuj%x, this%s%x, n)
426 call this%wws%update(k)
427 call col3(stats_uiuj%x, this%u%x, this%v%x, n)
428 call col3(stats_work%x, stats_uiuj%x, this%s%x, n)
429 call this%uvs%update(k)
430 call col3(stats_uiuj%x, this%u%x, this%w%x, n)
431 call col3(stats_work%x, stats_uiuj%x, this%s%x, n)
432 call this%uws%update(k)
433 call col3(stats_uiuj%x, this%v%x, this%w%x, n)
434 call col3(stats_work%x, stats_uiuj%x, this%s%x, n)
435 call this%vws%update(k)
436
437 call col3(stats_work%x, this%p%x, this%s%x, n)
438 call this%ps%update(k)
439
440 end if
441
442 call opgrad(this%dsdx%x, this%dsdy%x, this%dsdz%x, this%s%x, this%coef)
443 call opgrad(this%dudx%x, this%dudy%x, this%dudz%x, this%u%x, this%coef)
444 call opgrad(this%dvdx%x, this%dvdy%x, this%dvdz%x, this%v%x, this%coef)
445 call opgrad(this%dwdx%x, this%dwdy%x, this%dwdz%x, this%w%x, this%coef)
446
447 if (neko_bcknd_device .eq. 1) then
448 call device_col3(stats_work%x_d, this%dsdx%x_d, this%p%x_d, n)
449 call this%pdsdx%update(k)
450 call device_col3(stats_work%x_d, this%dsdy%x_d, this%p%x_d, n)
451 call this%pdsdy%update(k)
452 call device_col3(stats_work%x_d, this%dsdz%x_d, this%p%x_d, n)
453 call this%pdsdz%update(k)
454
455 call device_col3(stats_work%x_d, this%dsdx%x_d, this%u%x_d, n)
456 call this%udsdx%update(k)
457 call device_col3(stats_work%x_d, this%dsdy%x_d, this%u%x_d, n)
458 call this%udsdy%update(k)
459 call device_col3(stats_work%x_d, this%dsdz%x_d, this%u%x_d, n)
460 call this%udsdz%update(k)
461
462 call device_col3(stats_work%x_d, this%dsdx%x_d, this%v%x_d, n)
463 call this%vdsdx%update(k)
464 call device_col3(stats_work%x_d, this%dsdy%x_d, this%v%x_d, n)
465 call this%vdsdy%update(k)
466 call device_col3(stats_work%x_d, this%dsdz%x_d, this%v%x_d, n)
467 call this%vdsdz%update(k)
468
469 call device_col3(stats_work%x_d, this%dsdx%x_d, this%w%x_d, n)
470 call this%wdsdx%update(k)
471 call device_col3(stats_work%x_d, this%dsdy%x_d, this%w%x_d, n)
472 call this%wdsdy%update(k)
473 call device_col3(stats_work%x_d, this%dsdz%x_d, this%w%x_d, n)
474 call this%wdsdz%update(k)
475
476 call device_col3(stats_work%x_d, this%dudx%x_d, this%s%x_d, n)
477 call this%sdudx%update(k)
478 call device_col3(stats_work%x_d, this%dudy%x_d, this%s%x_d, n)
479 call this%sdudy%update(k)
480 call device_col3(stats_work%x_d, this%dudz%x_d, this%s%x_d, n)
481 call this%sdudz%update(k)
482
483 call device_col3(stats_work%x_d, this%dvdx%x_d, this%s%x_d, n)
484 call this%sdvdx%update(k)
485 call device_col3(stats_work%x_d, this%dvdy%x_d, this%s%x_d, n)
486 call this%sdvdy%update(k)
487 call device_col3(stats_work%x_d, this%dvdz%x_d, this%s%x_d, n)
488 call this%sdvdz%update(k)
489
490 call device_col3(stats_work%x_d, this%dwdx%x_d, this%s%x_d, n)
491 call this%sdwdx%update(k)
492 call device_col3(stats_work%x_d, this%dwdy%x_d, this%s%x_d, n)
493 call this%sdwdy%update(k)
494 call device_col3(stats_work%x_d, this%dwdz%x_d, this%s%x_d, n)
495 call this%sdwdz%update(k)
496
497 call device_col3(stats_work%x_d, this%dsdx%x_d, &
498 this%dsdx%x_d, n)
499 call device_addcol3(stats_work%x_d, this%dsdy%x_d, &
500 this%dsdy%x_d, n)
501 call device_addcol3(stats_work%x_d, this%dsdz%x_d, &
502 this%dsdz%x_d, n)
503 call this%ess%update(k)
504
505 call device_col3(stats_work%x_d, this%dudx%x_d, &
506 this%dsdx%x_d, n)
507 call device_addcol3(stats_work%x_d, this%dudy%x_d, &
508 this%dsdy%x_d, n)
509 call device_addcol3(stats_work%x_d, this%dudz%x_d, &
510 this%dsdz%x_d, n)
511 call this%eus%update(k)
512
513 call device_col3(stats_work%x_d, this%dvdx%x_d, &
514 this%dsdx%x_d, n)
515 call device_addcol3(stats_work%x_d, this%dvdy%x_d, &
516 this%dsdy%x_d, n)
517 call device_addcol3(stats_work%x_d, this%dvdz%x_d, &
518 this%dsdz%x_d, n)
519 call this%evs%update(k)
520
521 call device_col3(stats_work%x_d, this%dwdx%x_d, &
522 this%dsdx%x_d, n)
523 call device_addcol3(stats_work%x_d, this%dwdy%x_d, &
524 this%dsdy%x_d, n)
525 call device_addcol3(stats_work%x_d, this%dwdz%x_d, &
526 this%dsdz%x_d, n)
527 call this%ews%update(k)
528
529 else
530
531 call col3(stats_work%x, this%dsdx%x, this%p%x, n)
532 call this%pdsdx%update(k)
533 call col3(stats_work%x, this%dsdy%x, this%p%x, n)
534 call this%pdsdy%update(k)
535 call col3(stats_work%x, this%dsdz%x, this%p%x, n)
536 call this%pdsdz%update(k)
537
538 call col3(stats_work%x, this%dsdx%x, this%u%x, n)
539 call this%udsdx%update(k)
540 call col3(stats_work%x, this%dsdy%x, this%u%x, n)
541 call this%udsdy%update(k)
542 call col3(stats_work%x, this%dsdz%x, this%u%x, n)
543 call this%udsdz%update(k)
544
545 call col3(stats_work%x, this%dsdx%x, this%v%x, n)
546 call this%vdsdx%update(k)
547 call col3(stats_work%x, this%dsdy%x, this%v%x, n)
548 call this%vdsdy%update(k)
549 call col3(stats_work%x, this%dsdz%x, this%v%x, n)
550 call this%vdsdz%update(k)
551
552 call col3(stats_work%x, this%dsdx%x, this%w%x, n)
553 call this%wdsdx%update(k)
554 call col3(stats_work%x, this%dsdy%x, this%w%x, n)
555 call this%wdsdy%update(k)
556 call col3(stats_work%x, this%dsdz%x, this%w%x, n)
557 call this%wdsdz%update(k)
558
559 call col3(stats_work%x, this%dudx%x, this%s%x, n)
560 call this%sdudx%update(k)
561 call col3(stats_work%x, this%dudy%x, this%s%x, n)
562 call this%sdudy%update(k)
563 call col3(stats_work%x, this%dudz%x, this%s%x, n)
564 call this%sdudz%update(k)
565
566 call col3(stats_work%x, this%dvdx%x, this%s%x, n)
567 call this%sdvdx%update(k)
568 call col3(stats_work%x, this%dvdy%x, this%s%x, n)
569 call this%sdvdy%update(k)
570 call col3(stats_work%x, this%dvdz%x, this%s%x, n)
571 call this%sdvdz%update(k)
572
573 call col3(stats_work%x, this%dwdx%x, this%s%x, n)
574 call this%sdwdx%update(k)
575 call col3(stats_work%x, this%dwdy%x, this%s%x, n)
576 call this%sdwdy%update(k)
577 call col3(stats_work%x, this%dwdz%x, this%s%x, n)
578 call this%sdwdz%update(k)
579
580 call col3(stats_work%x, this%dsdx%x, &
581 this%dsdx%x, n)
582 call addcol3(stats_work%x, this%dsdy%x, &
583 this%dsdy%x, n)
584 call addcol3(stats_work%x, this%dsdz%x, &
585 this%dsdz%x, n)
586 call this%ess%update(k)
587
588 call col3(stats_work%x, this%dudx%x, &
589 this%dsdx%x, n)
590 call addcol3(stats_work%x, this%dudy%x, &
591 this%dsdy%x, n)
592 call addcol3(stats_work%x, this%dudz%x, &
593 this%dsdz%x, n)
594 call this%eus%update(k)
595
596 call col3(stats_work%x, this%dvdx%x, &
597 this%dsdx%x, n)
598 call addcol3(stats_work%x, this%dvdy%x, &
599 this%dsdy%x, n)
600 call addcol3(stats_work%x, this%dvdz%x, &
601 this%dsdz%x, n)
602 call this%evs%update(k)
603
604 call col3(stats_work%x, this%dwdx%x, &
605 this%dsdx%x, n)
606 call addcol3(stats_work%x, this%dwdy%x, &
607 this%dsdy%x, n)
608 call addcol3(stats_work%x, this%dwdz%x, &
609 this%dsdz%x, n)
610 call this%ews%update(k)
611
612 end if
613 end associate
614
615 end subroutine scalar_stats_update
616
617
619 subroutine scalar_stats_free(this)
620 class(scalar_stats_t), intent(inout) :: this
621
622 call this%stats_work%free()
623 call this%stats_ss%free()
624 call this%stats_uiuj%free()
625
626 call this%s_mean%free()
627
628 call this%us%free()
629 call this%vs%free()
630 call this%ws%free()
631
632 call this%ss%free()
633
634 if (this%n_stats .eq. 42) then
635 call this%sss%free()
636 call this%ssss%free()
637
638 call this%uss%free()
639 call this%vss%free()
640 call this%wss%free()
641
642 call this%uus%free()
643 call this%vvs%free()
644 call this%wws%free()
645 call this%uvs%free()
646 call this%uws%free()
647 call this%vws%free()
648
649 call this%ps%free()
650
651 call this%pdsdx%free()
652 call this%pdsdy%free()
653 call this%pdsdz%free()
654
655 call this%udsdx%free()
656 call this%udsdy%free()
657 call this%udsdz%free()
658 call this%vdsdx%free()
659 call this%vdsdy%free()
660 call this%vdsdz%free()
661 call this%wdsdx%free()
662 call this%wdsdy%free()
663 call this%wdsdz%free()
664
665 call this%sdudx%free()
666 call this%sdudy%free()
667 call this%sdudz%free()
668 call this%sdvdx%free()
669 call this%sdvdy%free()
670 call this%sdvdz%free()
671 call this%sdwdx%free()
672 call this%sdwdy%free()
673 call this%sdwdz%free()
674
675 call this%ess%free()
676 call this%eus%free()
677 call this%evs%free()
678 call this%ews%free()
679
680 call this%dsdx%free()
681 call this%dsdy%free()
682 call this%dsdz%free()
683 call this%dudx%free()
684 call this%dudy%free()
685 call this%dudz%free()
686 call this%dvdx%free()
687 call this%dvdy%free()
688 call this%dvdz%free()
689 call this%dwdx%free()
690 call this%dwdy%free()
691 call this%dwdz%free()
692 end if
693
694 nullify(this%coef)
695 nullify(this%s)
696 nullify(this%u)
697 nullify(this%v)
698 nullify(this%w)
699 nullify(this%p)
700
701 end subroutine scalar_stats_free
702
704 subroutine scalar_stats_reset(this)
705 class(scalar_stats_t), intent(inout), target:: this
706
707 call this%s_mean%reset()
708
709 call this%us%reset()
710 call this%vs%reset()
711 call this%ws%reset()
712
713 call this%ss%reset()
714
715 if (this%n_stats .eq. 42) then
716 call this%sss%reset()
717 call this%ssss%reset()
718
719 call this%uss%reset()
720 call this%vss%reset()
721 call this%wss%reset()
722
723 call this%uus%reset()
724 call this%vvs%reset()
725 call this%wws%reset()
726 call this%uvs%reset()
727 call this%uws%reset()
728 call this%vws%reset()
729
730 call this%ps%reset()
731
732 call this%pdsdx%reset()
733 call this%pdsdy%reset()
734 call this%pdsdz%reset()
735
736 call this%udsdx%reset()
737 call this%udsdy%reset()
738 call this%udsdz%reset()
739 call this%vdsdx%reset()
740 call this%vdsdy%reset()
741 call this%vdsdz%reset()
742 call this%wdsdx%reset()
743 call this%wdsdy%reset()
744 call this%wdsdz%reset()
745
746 call this%sdudx%reset()
747 call this%sdudy%reset()
748 call this%sdudz%reset()
749 call this%sdvdx%reset()
750 call this%sdvdy%reset()
751 call this%sdvdz%reset()
752 call this%sdwdx%reset()
753 call this%sdwdy%reset()
754 call this%sdwdz%reset()
755
756 call this%ess%reset()
757 call this%eus%reset()
758 call this%evs%reset()
759 call this%ews%reset()
760 end if
761
762 end subroutine scalar_stats_reset
763
764 ! Convert computed weak gradients to strong.
765 subroutine scalar_stats_make_strong_grad(this)
766 class(scalar_stats_t) :: this
767 integer :: n
768
769 if (this%n_stats .eq. 5) return
770
771 n = size(this%coef%B)
772
773 if (neko_bcknd_device .eq. 1) then
774 call device_cfill(this%stats_work%x_d, 1.0_rp, n)
775 call device_invcol2(this%stats_work%x_d, this%coef%B_d, n)
776
777 call device_col2(this%pdsdx%mf%x_d, this%stats_work%x_d, n)
778 call device_col2(this%pdsdy%mf%x_d, this%stats_work%x_d, n)
779 call device_col2(this%pdsdz%mf%x_d, this%stats_work%x_d, n)
780
781 call device_col2(this%udsdx%mf%x_d, this%stats_work%x_d, n)
782 call device_col2(this%udsdy%mf%x_d, this%stats_work%x_d, n)
783 call device_col2(this%udsdz%mf%x_d, this%stats_work%x_d, n)
784 call device_col2(this%vdsdx%mf%x_d, this%stats_work%x_d, n)
785 call device_col2(this%vdsdy%mf%x_d, this%stats_work%x_d, n)
786 call device_col2(this%vdsdz%mf%x_d, this%stats_work%x_d, n)
787 call device_col2(this%wdsdx%mf%x_d, this%stats_work%x_d, n)
788 call device_col2(this%wdsdy%mf%x_d, this%stats_work%x_d, n)
789 call device_col2(this%wdsdz%mf%x_d, this%stats_work%x_d, n)
790
791 call device_col2(this%sdudx%mf%x_d, this%stats_work%x_d, n)
792 call device_col2(this%sdudy%mf%x_d, this%stats_work%x_d, n)
793 call device_col2(this%sdudz%mf%x_d, this%stats_work%x_d, n)
794 call device_col2(this%sdvdx%mf%x_d, this%stats_work%x_d, n)
795 call device_col2(this%sdvdy%mf%x_d, this%stats_work%x_d, n)
796 call device_col2(this%sdvdz%mf%x_d, this%stats_work%x_d, n)
797 call device_col2(this%sdwdx%mf%x_d, this%stats_work%x_d, n)
798 call device_col2(this%sdwdy%mf%x_d, this%stats_work%x_d, n)
799 call device_col2(this%sdwdz%mf%x_d, this%stats_work%x_d, n)
800
801 call device_col2(this%stats_work%x_d, this%stats_work%x_d, n)
802
803 call device_col2(this%ess%mf%x_d, this%stats_work%x_d, n)
804 call device_col2(this%eus%mf%x_d, this%stats_work%x_d, n)
805 call device_col2(this%evs%mf%x_d, this%stats_work%x_d, n)
806 call device_col2(this%ews%mf%x_d, this%stats_work%x_d, n)
807
808 else
809
810 call invers2(this%stats_work%x, this%coef%B, n)
811
812 call col2(this%pdsdx%mf%x, this%stats_work%x, n)
813 call col2(this%pdsdy%mf%x, this%stats_work%x, n)
814 call col2(this%pdsdz%mf%x, this%stats_work%x, n)
815
816 call col2(this%udsdx%mf%x, this%stats_work%x, n)
817 call col2(this%udsdy%mf%x, this%stats_work%x, n)
818 call col2(this%udsdz%mf%x, this%stats_work%x, n)
819 call col2(this%vdsdx%mf%x, this%stats_work%x, n)
820 call col2(this%vdsdy%mf%x, this%stats_work%x, n)
821 call col2(this%vdsdz%mf%x, this%stats_work%x, n)
822 call col2(this%wdsdx%mf%x, this%stats_work%x, n)
823 call col2(this%wdsdy%mf%x, this%stats_work%x, n)
824 call col2(this%wdsdz%mf%x, this%stats_work%x, n)
825
826 call col2(this%sdudx%mf%x, this%stats_work%x, n)
827 call col2(this%sdudy%mf%x, this%stats_work%x, n)
828 call col2(this%sdudz%mf%x, this%stats_work%x, n)
829 call col2(this%sdvdx%mf%x, this%stats_work%x, n)
830 call col2(this%sdvdy%mf%x, this%stats_work%x, n)
831 call col2(this%sdvdz%mf%x, this%stats_work%x, n)
832 call col2(this%sdwdx%mf%x, this%stats_work%x, n)
833 call col2(this%sdwdy%mf%x, this%stats_work%x, n)
834 call col2(this%sdwdz%mf%x, this%stats_work%x, n)
835
836 call col2(this%stats_work%x, this%stats_work%x, n)
837
838 call col2(this%ess%mf%x, this%stats_work%x, n)
839 call col2(this%eus%mf%x, this%stats_work%x, n)
840 call col2(this%evs%mf%x, this%stats_work%x, n)
841 call col2(this%ews%mf%x, this%stats_work%x, n)
842
843 end if
844
845 end subroutine scalar_stats_make_strong_grad
846
847end module scalar_stats
device::device_memcpy
Copy data between host and device (or device and device)
Definition device.F90:71
coefs
Coefficients.
Definition coef.f90:34
device_math
Definition device_math.F90:33
device_math::device_addcol3
subroutine, public device_addcol3(a_d, b_d, c_d, n, strm)
Returns .
Definition device_math.F90:912
device_math::device_col2
subroutine, public device_col2(a_d, b_d, n, strm)
Vector multiplication .
Definition device_math.F90:783
device_math::device_invcol2
subroutine, public device_invcol2(a_d, b_d, n, strm)
Vector division .
Definition device_math.F90:732
device_math::device_col3
subroutine, public device_col3(a_d, b_d, c_d, n, strm)
Vector multiplication with 3 vectors .
Definition device_math.F90:808
device_math::device_cfill
subroutine, public device_cfill(a_d, c, n, strm)
Set all elements to a constant c .
Definition device_math.F90:440
device
Device abstraction, common interface for various accelerators.
Definition device.F90:34
device::host_to_device
integer, parameter, public host_to_device
Definition device.F90:47
device::device_to_host
integer, parameter, public device_to_host
Definition device.F90:47
field_list
Definition field_list.f90:1
field
Defines a field.
Definition field.f90:34
math
Definition math.f90:60
math::invers2
subroutine, public invers2(a, b, n)
Compute inverted vector .
Definition math.f90:639
math::subcol3
subroutine, public subcol3(a, b, c, n)
Returns .
Definition math.f90:881
math::addcol3
subroutine, public addcol3(a, b, c, n)
Returns .
Definition math.f90:958
math::col2
subroutine, public col2(a, b, n)
Vector multiplication .
Definition math.f90:854
math::copy
subroutine, public copy(a, b, n)
Copy a vector .
Definition math.f90:249
math::col3
subroutine, public col3(a, b, c, n)
Vector multiplication with 3 vectors .
Definition math.f90:867
mean_field
Implements mean_field_t.
Definition mean_field.f90:35
neko_config
Build configurations.
Definition neko_config.f90:34
neko_config::neko_bcknd_device
integer, parameter neko_bcknd_device
Definition neko_config.f90:44
num_types
Definition num_types.f90:1
num_types::rp
integer, parameter, public rp
Global precision used in computations.
Definition num_types.f90:12
operators
Operators.
Definition operators.f90:34
operators::opgrad
subroutine, public opgrad(ux, uy, uz, u, coef, es, ee)
Compute the weak gradient of a scalar field, i.e. the gradient multiplied by the mass matrix.
Definition operators.f90:185
scalar_stats
Computes various statistics for the scalar fields. We use the Reynolds decomposition for a field u = ...
Definition scalar_stats.f90:36
scalar_stats::scalar_stats_free
subroutine scalar_stats_free(this)
Destructor.
Definition scalar_stats.f90:620
scalar_stats::scalar_stats_make_strong_grad
subroutine scalar_stats_make_strong_grad(this)
Definition scalar_stats.f90:766
scalar_stats::scalar_stats_update
subroutine scalar_stats_update(this, k)
Updates all fields with a new sample.
Definition scalar_stats.f90:331
scalar_stats::scalar_stats_init
subroutine scalar_stats_init(this, coef, s, u, v, w, p, set)
Constructor. Initialize the fields associated with scalar_stats.
Definition scalar_stats.f90:167
scalar_stats::scalar_stats_reset
subroutine scalar_stats_reset(this)
Resets all the computed means values and sampling times to zero.
Definition scalar_stats.f90:705
stats_quant
Defines a statistical quantity.
Definition stats_quant.f90:34
utils
Utilities.
Definition utils.f90:35
utils::neko_warning
subroutine, public neko_warning(warning_msg)
Reports a warning to standard output.
Definition utils.f90:346
coefs::coef_t
Coefficients defined on a given (mesh, ) tuple. Arrays use indices (i,j,k,e): element e,...
Definition coef.f90:56
field::field_t
Definition field.f90:47
field_list::field_list_t
field_list_t, To be able to group fields together
Definition field_list.f90:15
mean_field::mean_field_t
Computes the temporal mean of a field.
Definition mean_field.f90:45
scalar_stats::scalar_stats_t
Definition scalar_stats.f90:53
stats_quant::stats_quant_t
Abstract type defining a statistical quantity.
Definition stats_quant.f90:40