include/pops/coupling/schur/core/schur_source_kernels.hpp Source File

adc_cpp 0.3.0 Model-free C++23 core for coupled hyperbolic-elliptic systems on adaptive (AMR) meshes, with MPI and GPU (Kokkos) backends

• include
• pops
• coupling
• schur
• core

schur_source_kernels.hpp

Go to the documentation of this file.

1#pragma once

2

3#include <pops/core/foundation/types.hpp> // Real, POPS_HD

4#include <pops/mesh/storage/fab2d.hpp> // Array4, ConstArray4

5

6#include <stdexcept>

7#include <string>

8

25

26namespace pops {

27

28namespace detail {

29

32struct SchurExtrapolateScalarKernel {

33 ConstArray4 f_n;

34 Array4 f;

35 Real inv_theta;

36 POPS_HD void operator()(int i, int j) const {

37 f(i, j, 0) = f_n(i, j, 0) + inv_theta * (f(i, j, 0) - f_n(i, j, 0));

38 }

39};

40

43struct SchurExtrapolateVelocityKernel {

44 ConstArray4 vx_n, vy_n;

45 Array4 vx, vy;

46 Array4 st;

47 Real inv_theta;

48 int c_rho, c_mx, c_my;

49 POPS_HD void operator()(int i, int j) const {

50 const Real nx = vx_n(i, j, 0) + inv_theta * (vx(i, j, 0) - vx_n(i, j, 0));

51 const Real ny = vy_n(i, j, 0) + inv_theta * (vy(i, j, 0) - vy_n(i, j, 0));

52 vx(i, j, 0) = nx;

53 vy(i, j, 0) = ny;

54 const Real rho = st(i, j, c_rho);

55 st(i, j, c_mx) = rho * nx;

56 st(i, j, c_my) = rho * ny;

57 }

58};

59

62struct SchurEnergyKernel {

63 ConstArray4 vx_n, vy_n;

64 ConstArray4 vx, vy;

65 Array4 st;

66 int c_rho, c_E;

67 POPS_HD void operator()(int i, int j) const {

68 const Real rho = st(i, j, c_rho);

69 const Real ke_old =

70 Real(0.5) * rho * (vx_n(i, j, 0) * vx_n(i, j, 0) + vy_n(i, j, 0) * vy_n(i, j, 0));

71 const Real ke_new = Real(0.5) * rho * (vx(i, j, 0) * vx(i, j, 0) + vy(i, j, 0) * vy(i, j, 0));

72 st(i, j, c_E) += ke_new - ke_old;

73 }

74};

75

79struct ExtractVelocityKernel {

80 ConstArray4 st;

81 Array4 vx, vy;

82 int c_rho, c_mx, c_my;

83 POPS_HD void operator()(int i, int j) const {

84 const Real rho = st(i, j, c_rho);

85 const Real inv = rho != Real(0) ? Real(1) / rho : Real(0);

86 vx(i, j, 0) = st(i, j, c_mx) * inv;

87 vy(i, j, 0) = st(i, j, c_my) * inv;

88 }

89};

90

93struct CopyBzKernel {

94 ConstArray4 bz_src;

95 Array4 bz_dst;

96 int c_bz;

97 POPS_HD void operator()(int i, int j) const { bz_dst(i, j, 0) = bz_src(i, j, c_bz); }

98};

99

103inline void validate_krylov_params(Real tol, int max_iters, const char* who) {

104 if (!(tol > Real(0)) || max_iters < 1)

105 throw std::invalid_argument(std::string(who) + ": tol > 0, max_iters >= 1");

106}

107

108} // namespace detail

109

110} // namespace pops

fab2d.hpp

Fab2D: single-grid data on a Box2D (in-house equivalent of AMReX's FArrayBox); Array4 / ConstArray4: ...

pops::detail::validate_krylov_params

void validate_krylov_params(Real tol, int max_iters, const char *who)

Validates the Krylov tolerance / iteration budget shared by the Schur source steppers (historical con...

Definition schur_source_kernels.hpp:103

pops

Definition amr_hierarchy.hpp:29

pops::Real

double Real

Definition types.hpp:30

pops::Array4

WRITE POD handle (raw pointer + strides) over a Fab2D buffer, indexed by (i, j, c) IN GLOBAL INDICES ...

Definition fab2d.hpp:29

pops::ConstArray4

READ-only handle (const counterpart of Array4): same layout and same contract (POD device-copyable,...

Definition fab2d.hpp:44

pops::detail::CopyBzKernel

Copies the B_z field (aux channel) into an internal scalar MultiFab (0 ghost is enough,...

Definition schur_source_kernels.hpp:93

pops::detail::CopyBzKernel::operator()

POPS_HD void operator()(int i, int j) const

Definition schur_source_kernels.hpp:97

pops::detail::CopyBzKernel::bz_dst

Array4 bz_dst

Definition schur_source_kernels.hpp:95

pops::detail::CopyBzKernel::bz_src

ConstArray4 bz_src

Definition schur_source_kernels.hpp:94

pops::detail::CopyBzKernel::c_bz

int c_bz

Definition schur_source_kernels.hpp:96

pops::detail::ExtractVelocityKernel

Extracts the velocity v = (mx, my) / rho from the state (Density / MomentumX / MomentumY roles) into ...

Definition schur_source_kernels.hpp:79

pops::detail::ExtractVelocityKernel::vy

Array4 vy

Definition schur_source_kernels.hpp:81

pops::detail::ExtractVelocityKernel::c_mx

int c_mx

Definition schur_source_kernels.hpp:82

pops::detail::ExtractVelocityKernel::operator()

POPS_HD void operator()(int i, int j) const

Definition schur_source_kernels.hpp:83

pops::detail::ExtractVelocityKernel::st

ConstArray4 st

Definition schur_source_kernels.hpp:80

pops::detail::ExtractVelocityKernel::c_my

int c_my

Definition schur_source_kernels.hpp:82

pops::detail::ExtractVelocityKernel::c_rho

int c_rho

Definition schur_source_kernels.hpp:82

pops::detail::ExtractVelocityKernel::vx

Array4 vx

Definition schur_source_kernels.hpp:81

pops::detail::SchurEnergyKernel

Energy update: E^{n+1} = E^n + (1/2) rho^n (|v^{n+1}|^2 - |v^n|^2).

Definition schur_source_kernels.hpp:62

pops::detail::SchurEnergyKernel::st

Array4 st

state (WRITE E; READ rho)

Definition schur_source_kernels.hpp:65

pops::detail::SchurEnergyKernel::c_E

int c_E

Definition schur_source_kernels.hpp:66

pops::detail::SchurEnergyKernel::vy_n

ConstArray4 vy_n

v^n

Definition schur_source_kernels.hpp:63

pops::detail::SchurEnergyKernel::vx_n

ConstArray4 vx_n

Definition schur_source_kernels.hpp:63

pops::detail::SchurEnergyKernel::c_rho

int c_rho

Definition schur_source_kernels.hpp:66

pops::detail::SchurEnergyKernel::vx

ConstArray4 vx

Definition schur_source_kernels.hpp:64

pops::detail::SchurEnergyKernel::operator()

POPS_HD void operator()(int i, int j) const

Definition schur_source_kernels.hpp:67

pops::detail::SchurEnergyKernel::vy

ConstArray4 vy

v^{n+1}

Definition schur_source_kernels.hpp:64

pops::detail::SchurExtrapolateScalarKernel

Linear extrapolation of a SCALAR field from the theta-stage to the full step: f^{n+1} = f^n + (1/thet...

Definition schur_source_kernels.hpp:32

pops::detail::SchurExtrapolateScalarKernel::f_n

ConstArray4 f_n

f^n

Definition schur_source_kernels.hpp:33

pops::detail::SchurExtrapolateScalarKernel::operator()

POPS_HD void operator()(int i, int j) const

Definition schur_source_kernels.hpp:36

pops::detail::SchurExtrapolateScalarKernel::inv_theta

Real inv_theta

1 / theta

Definition schur_source_kernels.hpp:35

pops::detail::SchurExtrapolateScalarKernel::f

Array4 f

IN: f^{n+theta}; OUT: f^{n+1}.

Definition schur_source_kernels.hpp:34

pops::detail::SchurExtrapolateVelocityKernel

Linear extrapolation of the VELOCITY (vx, vy) from the theta-stage to the full step,...

Definition schur_source_kernels.hpp:43

pops::detail::SchurExtrapolateVelocityKernel::c_rho

int c_rho

Definition schur_source_kernels.hpp:48

pops::detail::SchurExtrapolateVelocityKernel::operator()

POPS_HD void operator()(int i, int j) const

Definition schur_source_kernels.hpp:49

pops::detail::SchurExtrapolateVelocityKernel::st

Array4 st

state (WRITE mx, my; READ rho)

Definition schur_source_kernels.hpp:46

pops::detail::SchurExtrapolateVelocityKernel::vy

Array4 vy

IN: v^{n+theta}; OUT: v^{n+1}.

Definition schur_source_kernels.hpp:45

pops::detail::SchurExtrapolateVelocityKernel::c_my

int c_my

Definition schur_source_kernels.hpp:48

pops::detail::SchurExtrapolateVelocityKernel::vy_n

ConstArray4 vy_n

v^n

Definition schur_source_kernels.hpp:44

pops::detail::SchurExtrapolateVelocityKernel::vx

Array4 vx

Definition schur_source_kernels.hpp:45

pops::detail::SchurExtrapolateVelocityKernel::inv_theta

Real inv_theta

1 / theta

Definition schur_source_kernels.hpp:47

pops::detail::SchurExtrapolateVelocityKernel::vx_n

ConstArray4 vx_n

Definition schur_source_kernels.hpp:44

pops::detail::SchurExtrapolateVelocityKernel::c_mx

int c_mx

Definition schur_source_kernels.hpp:48

types.hpp

Base scalar types and the POPS_HD macro (host+device portability).

POPS_HD

#define POPS_HD

Definition types.hpp:25

---

Generated by  1.9.8