Neko 1.99.1
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 end subroutine scalar_stats_free
695
697 subroutine scalar_stats_reset(this)
698 class(scalar_stats_t), intent(inout), target:: this
699
700 call this%s_mean%reset()
701
702 call this%us%reset()
703 call this%vs%reset()
704 call this%ws%reset()
705
706 call this%ss%reset()
707
708 if (this%n_stats .eq. 42) then
709 call this%sss%reset()
710 call this%ssss%reset()
711
712 call this%uss%reset()
713 call this%vss%reset()
714 call this%wss%reset()
715
716 call this%uus%reset()
717 call this%vvs%reset()
718 call this%wws%reset()
719 call this%uvs%reset()
720 call this%uws%reset()
721 call this%vws%reset()
722
723 call this%ps%reset()
724
725 call this%pdsdx%reset()
726 call this%pdsdy%reset()
727 call this%pdsdz%reset()
728
729 call this%udsdx%reset()
730 call this%udsdy%reset()
731 call this%udsdz%reset()
732 call this%vdsdx%reset()
733 call this%vdsdy%reset()
734 call this%vdsdz%reset()
735 call this%wdsdx%reset()
736 call this%wdsdy%reset()
737 call this%wdsdz%reset()
738
739 call this%sdudx%reset()
740 call this%sdudy%reset()
741 call this%sdudz%reset()
742 call this%sdvdx%reset()
743 call this%sdvdy%reset()
744 call this%sdvdz%reset()
745 call this%sdwdx%reset()
746 call this%sdwdy%reset()
747 call this%sdwdz%reset()
748
749 call this%ess%reset()
750 call this%eus%reset()
751 call this%evs%reset()
752 call this%ews%reset()
753 end if
754
755 end subroutine scalar_stats_reset
756
757 ! Convert computed weak gradients to strong.
758 subroutine scalar_stats_make_strong_grad(this)
759 class(scalar_stats_t) :: this
760 integer :: n
761
762 if (this%n_stats .eq. 5) return
763
764 n = size(this%coef%B)
765
766 if (neko_bcknd_device .eq. 1) then
767 call device_cfill(this%stats_work%x_d, 1.0_rp, n)
768 call device_invcol2(this%stats_work%x_d, this%coef%B_d, n)
769
770 call device_col2(this%pdsdx%mf%x_d, this%stats_work%x_d, n)
771 call device_col2(this%pdsdy%mf%x_d, this%stats_work%x_d, n)
772 call device_col2(this%pdsdz%mf%x_d, this%stats_work%x_d, n)
773
774 call device_col2(this%udsdx%mf%x_d, this%stats_work%x_d, n)
775 call device_col2(this%udsdy%mf%x_d, this%stats_work%x_d, n)
776 call device_col2(this%udsdz%mf%x_d, this%stats_work%x_d, n)
777 call device_col2(this%vdsdx%mf%x_d, this%stats_work%x_d, n)
778 call device_col2(this%vdsdy%mf%x_d, this%stats_work%x_d, n)
779 call device_col2(this%vdsdz%mf%x_d, this%stats_work%x_d, n)
780 call device_col2(this%wdsdx%mf%x_d, this%stats_work%x_d, n)
781 call device_col2(this%wdsdy%mf%x_d, this%stats_work%x_d, n)
782 call device_col2(this%wdsdz%mf%x_d, this%stats_work%x_d, n)
783
784 call device_col2(this%sdudx%mf%x_d, this%stats_work%x_d, n)
785 call device_col2(this%sdudy%mf%x_d, this%stats_work%x_d, n)
786 call device_col2(this%sdudz%mf%x_d, this%stats_work%x_d, n)
787 call device_col2(this%sdvdx%mf%x_d, this%stats_work%x_d, n)
788 call device_col2(this%sdvdy%mf%x_d, this%stats_work%x_d, n)
789 call device_col2(this%sdvdz%mf%x_d, this%stats_work%x_d, n)
790 call device_col2(this%sdwdx%mf%x_d, this%stats_work%x_d, n)
791 call device_col2(this%sdwdy%mf%x_d, this%stats_work%x_d, n)
792 call device_col2(this%sdwdz%mf%x_d, this%stats_work%x_d, n)
793
794 call device_col2(this%stats_work%x_d, this%stats_work%x_d, n)
795
796 call device_col2(this%ess%mf%x_d, this%stats_work%x_d, n)
797 call device_col2(this%eus%mf%x_d, this%stats_work%x_d, n)
798 call device_col2(this%evs%mf%x_d, this%stats_work%x_d, n)
799 call device_col2(this%ews%mf%x_d, this%stats_work%x_d, n)
800
801 else
802
803 call invers2(this%stats_work%x, this%coef%B, n)
804
805 call col2(this%pdsdx%mf%x, this%stats_work%x, n)
806 call col2(this%pdsdy%mf%x, this%stats_work%x, n)
807 call col2(this%pdsdz%mf%x, this%stats_work%x, n)
808
809 call col2(this%udsdx%mf%x, this%stats_work%x, n)
810 call col2(this%udsdy%mf%x, this%stats_work%x, n)
811 call col2(this%udsdz%mf%x, this%stats_work%x, n)
812 call col2(this%vdsdx%mf%x, this%stats_work%x, n)
813 call col2(this%vdsdy%mf%x, this%stats_work%x, n)
814 call col2(this%vdsdz%mf%x, this%stats_work%x, n)
815 call col2(this%wdsdx%mf%x, this%stats_work%x, n)
816 call col2(this%wdsdy%mf%x, this%stats_work%x, n)
817 call col2(this%wdsdz%mf%x, this%stats_work%x, n)
818
819 call col2(this%sdudx%mf%x, this%stats_work%x, n)
820 call col2(this%sdudy%mf%x, this%stats_work%x, n)
821 call col2(this%sdudz%mf%x, this%stats_work%x, n)
822 call col2(this%sdvdx%mf%x, this%stats_work%x, n)
823 call col2(this%sdvdy%mf%x, this%stats_work%x, n)
824 call col2(this%sdvdz%mf%x, this%stats_work%x, n)
825 call col2(this%sdwdx%mf%x, this%stats_work%x, n)
826 call col2(this%sdwdy%mf%x, this%stats_work%x, n)
827 call col2(this%sdwdz%mf%x, this%stats_work%x, n)
828
829 call col2(this%stats_work%x, this%stats_work%x, n)
830
831 call col2(this%ess%mf%x, this%stats_work%x, n)
832 call col2(this%eus%mf%x, this%stats_work%x, n)
833 call col2(this%evs%mf%x, this%stats_work%x, n)
834 call col2(this%ews%mf%x, this%stats_work%x, n)
835
836 end if
837
838 end subroutine scalar_stats_make_strong_grad
839
840end 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:177
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:759
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:698
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:55
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