1 #ifndef ERF_SHOCINTERFACE_H
2 #define ERF_SHOCINTERFACE_H
9 #include "physics/shoc/shoc_constants.hpp"
10 #include "physics/shoc/shoc_functions.hpp"
11 #include "share/util/eamxx_common_physics_functions.hpp"
13 #include "ekat_subview_utils.hpp"
14 #include "ekat_parameter_list.hpp"
15 #include "ekat_assert.hpp"
16 #include "ekat_team_policy_utils.hpp"
17 #include "ekat_reduction_utils.hpp"
24 #include <Kokkos_Core.hpp>
25 #include <AMReX_ParmParse.H>
26 #include <AMReX_MultiFabUtil.H>
37 using SHF = scream::shoc::Functions<Real, KokkosDefaultDevice>;
38 using PF = scream::PhysicsFunctions<KokkosDefaultDevice>;
39 using C = scream::physics::Constants<Real>;
40 using KT = ekat::KokkosTypes<KokkosDefaultDevice>;
41 using SC = scream::shoc::Constants<Real>;
43 using Spack =
typename SHF::Spack;
45 using Smask =
typename SHF::Smask;
47 using view_1d =
typename SHF::view_1d<Real>;
49 using view_2d =
typename SHF::view_2d<SHF::Spack>;
53 using view_3d =
typename SHF::view_3d<Spack>;
57 using WSM = ekat::WorkspaceManager<Spack, KT::Device>;
59 template<
typename ScalarT>
60 using uview_1d = ekat::Unmanaged<typename KT::template view_1d<ScalarT>>;
61 template<
typename ScalarT>
62 using uview_2d = ekat::Unmanaged<typename KT::template view_2d<ScalarT>>;
73 const amrex::BoxArray& ba,
74 amrex::Geometry& geom,
75 amrex::MultiFab*
cons,
76 amrex::MultiFab* z_phys);
107 std::string
name ()
const {
return "shoc"; }
109 #ifndef KOKKOS_ENABLE_CUDA
122 KOKKOS_INLINE_FUNCTION
123 void operator()(
const Kokkos::TeamPolicy<KT::ExeSpace>::member_type& team)
const
125 const int i = team.league_rank();
127 const Real zvir = C::ZVIR;
128 const Real cpair = C::Cpair;
129 const Real ggr = C::gravit;
130 const Real inv_ggr = 1/ggr;
131 const Real mintke = SC::mintke;
133 const int nlev_packs = ekat::npack<Spack>(
nlev);
135 Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev_packs), [&] (
const Int& k)
140 const Spack exner = PF::exner_function(
p_mid(i,k));
141 const Smask nonzero = (exner != 0);
142 EKAT_KERNEL_ASSERT((nonzero || !(ekat::range<IntSmallPack>(k*Spack::n) <
nlev)).all());
145 tke(i,k) = ekat::max(mintke,
tke(i,k));
156 qw(i,k) =
qv(i,k) +
qc(i,k);
160 const auto theta_zt = PF::calculate_theta_from_T(
T_mid(i,k),
p_mid(i,k));
161 thlm(i,k) = PF::calculate_thetal_from_theta(theta_zt,
T_mid(i,k),
qc(i,k));
162 thv(i,k) = theta_zt*(1 + zvir*
qv(i,k) -
qc(i,k));
173 const auto dz_s = ekat::subview(
dz, i);
174 const auto z_int_s = ekat::subview(
z_int, i);
175 const auto z_mid_s = ekat::subview(
z_mid, i);
176 PF::calculate_z_int(team,
nlev,dz_s,
z_surf,z_int_s);
178 PF::calculate_z_mid(team,
nlev,z_int_s,z_mid_s);
181 const int nlevi_v =
nlev/Spack::n;
182 const int nlevi_p =
nlev%Spack::n;
183 Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev_packs), [&] (
const Int& k)
191 if (k+1 == nlev_packs)
zi_grid(i,nlevi_v)[nlevi_p] = 0;
195 const auto zt_grid_s = ekat::subview(
zt_grid, i);
196 const auto zi_grid_s = ekat::subview(
zi_grid, i);
197 const auto rrho_s = ekat::subview(
rrho, i);
198 const auto rrho_i_s = ekat::subview(
rrho_i, i);
199 SHF::linear_interp(team,zt_grid_s,zi_grid_s,rrho_s,rrho_i_s,
nlev,
nlev+1,0);
202 const auto exner_int = PF::exner_function(
p_int(i,nlevi_v)[nlevi_p]);
203 const auto inv_exner_int_surf = 1/exner_int;
318 KOKKOS_INLINE_FUNCTION
319 void operator()(
const Kokkos::TeamPolicy<KT::ExeSpace>::member_type& team)
const
321 const int i = team.league_rank();
323 const Real inv_qc_relvar_max = 10;
324 const Real inv_qc_relvar_min = 0.001;
326 const int nlev_packs = ekat::npack<Spack>(
nlev);
327 Kokkos::parallel_for(Kokkos::TeamVectorRange(team, nlev_packs), [&] (
const Int& k)
333 qv(i,k) =
qw(i,k) -
qc(i,k);
340 const auto condition = (
qc(i,k) != 0 &&
qc2(i,k) != 0);
341 if (condition.any()) {
343 ekat::min(inv_qc_relvar_max,
344 ekat::max(inv_qc_relvar_min,
345 ekat::square(
qc(i,k))/
qc2(i,k))));
352 T_mid(i,k) = PF::calculate_temperature_from_dse(dse_ik,z_mid_ik,phis_i);
430 #ifndef SCREAM_SHOC_SMALL_KERNELS
436 #ifndef SCREAM_SHOC_SMALL_KERNELS
449 #ifdef SCREAM_SHOC_SMALL_KERNELS
496 #ifdef SCREAM_SHOC_SMALL_KERNELS
509 #ifndef KOKKOS_ENABLE_CUDA
574 #ifdef SCREAM_SHOC_SMALL_KERNELS
575 SHF::SHOCTemporaries temporaries;
int Int
Definition: ERF_ShocInterface.H:20
amrex::Real Real
Definition: ERF_ShocInterface.H:19
Definition: ERF_ShocInterface.H:36
view_2d tabs
Definition: ERF_ShocInterface.H:686
view_2d p_int
Definition: ERF_ShocInterface.H:603
typename SHF::view_1d< const Real > view_1d_const
Definition: ERF_ShocInterface.H:48
bool check_flux_state
Definition: ERF_ShocInterface.H:588
SHOCPreprocess shoc_preprocess
Definition: ERF_ShocInterface.H:579
view_2d wqw_sec
Definition: ERF_ShocInterface.H:658
scream::shoc::Constants< Real > SC
Definition: ERF_ShocInterface.H:41
view_1d surf_drag_coeff_tms
Definition: ERF_ShocInterface.H:683
view_2d dz_zi
Definition: ERF_ShocInterface.H:688
view_2d inv_exner
Definition: ERF_ShocInterface.H:623
amrex::Geometry m_geom
Definition: ERF_ShocInterface.H:551
void set_computed_group_impl()
view_1d wl_b
Definition: ERF_ShocInterface.H:675
void alloc_buffers()
Definition: ERF_ShocInterface.cpp:154
view_1d wv_a
Definition: ERF_ShocInterface.H:678
view_2d wtke_sec
Definition: ERF_ShocInterface.H:659
view_2d thl_sec
Definition: ERF_ShocInterface.H:654
view_2d tkh
Definition: ERF_ShocInterface.H:648
view_2d cldfrac_liq
Definition: ERF_ShocInterface.H:627
amrex::MultiFab * m_z_phys
Definition: ERF_ShocInterface.H:560
Int m_num_layers
Definition: ERF_ShocInterface.H:537
view_2d vw_sec
Definition: ERF_ShocInterface.H:661
view_2d thv
Definition: ERF_ShocInterface.H:616
view_2d shoc_qv
Definition: ERF_ShocInterface.H:685
size_t requested_buffer_size_in_bytes() const
view_2d shoc_cond
Definition: ERF_ShocInterface.H:666
typename ekat::KokkosTypes< KokkosDefaultDevice >::template view_2d< Real > sview_2d
Definition: ERF_ShocInterface.H:51
view_1d vpwp_sfc
Definition: ERF_ShocInterface.H:596
view_2d brunt
Definition: ERF_ShocInterface.H:664
Int m_num_vel_comp
Definition: ERF_ShocInterface.H:541
bool apply_tms
Definition: ERF_ShocInterface.H:587
view_2d dens
Definition: ERF_ShocInterface.H:604
view_1d surf_sens_flux
Definition: ERF_ShocInterface.H:597
view_1d pblh
Definition: ERF_ShocInterface.H:645
Int m_num_cols
Definition: ERF_ShocInterface.H:536
view_1d wstar
Definition: ERF_ShocInterface.H:682
view_1d se_b
Definition: ERF_ShocInterface.H:672
view_2d sgs_buoy_flux
Definition: ERF_ShocInterface.H:639
SHF::SHOCOutput output
Definition: ERF_ShocInterface.H:571
typename KT::template view_1d< Int > view_1d_int
Definition: ERF_ShocInterface.H:46
ekat::WorkspaceManager< Spack, KT::Device > workspace_mgr
Definition: ERF_ShocInterface.H:583
view_2d p_mid
Definition: ERF_ShocInterface.H:602
view_2d rho_zt
Definition: ERF_ShocInterface.H:684
view_2d uw_sec
Definition: ERF_ShocInterface.H:660
view_2d dse
Definition: ERF_ShocInterface.H:618
Int m_nadv
Definition: ERF_ShocInterface.H:539
view_2d qc
Definition: ERF_ShocInterface.H:607
view_1d kbfs
Definition: ERF_ShocInterface.H:680
view_2d dz
Definition: ERF_ShocInterface.H:617
view_2d qwthl_sec
Definition: ERF_ShocInterface.H:656
view_1d ke_a
Definition: ERF_ShocInterface.H:677
void run_impl(const Real dt)
Definition: ERF_ShocInterface.cpp:652
view_1d wl_a
Definition: ERF_ShocInterface.H:679
Int m_npbl
Definition: ERF_ShocInterface.H:538
view_2d shoc_mix
Definition: ERF_ShocInterface.H:652
view_2d qw_sec
Definition: ERF_ShocInterface.H:655
view_2d cldfrac_liq_prev
Definition: ERF_ShocInterface.H:628
scream::physics::Constants< Real > C
Definition: ERF_ShocInterface.H:39
view_2d qw
Definition: ERF_ShocInterface.H:625
SHF::SHOCInput input
Definition: ERF_ShocInterface.H:569
SHOCInterface(const int &lev, SolverChoice &sc)
Definition: ERF_ShocInterface.cpp:6
view_2d cloud_frac
Definition: ERF_ShocInterface.H:626
view_2d omega
Definition: ERF_ShocInterface.H:605
typename SHF::view_2d< const Spack > view_2d_const
Definition: ERF_ShocInterface.H:50
ekat::Unmanaged< typename KT::template view_1d< ScalarT > > uview_1d
Definition: ERF_ShocInterface.H:60
typename SHF::Smask Smask
Definition: ERF_ShocInterface.H:45
typename SHF::view_3d_strided< Spack > view_3d_strided
Definition: ERF_ShocInterface.H:55
view_1d ustar2
Definition: ERF_ShocInterface.H:681
view_2d rrho_i
Definition: ERF_ShocInterface.H:615
view_1d obklen
Definition: ERF_ShocInterface.H:647
void kokkos_buffers_to_mf()
Definition: ERF_ShocInterface.cpp:533
view_2d z_mid
Definition: ERF_ShocInterface.H:609
view_2d qc_copy
Definition: ERF_ShocInterface.H:608
scream::shoc::Functions< Real, KokkosDefaultDevice > SHF
Definition: ERF_ShocInterface.H:37
SHF::SHOCRuntime runtime_options
Definition: ERF_ShocInterface.H:573
view_2d rrho
Definition: ERF_ShocInterface.H:614
view_2d inv_qc_relvar
Definition: ERF_ShocInterface.H:635
view_1d surf_evap
Definition: ERF_ShocInterface.H:598
view_2d w_sec
Definition: ERF_ShocInterface.H:653
void set_grids(int &level, const amrex::BoxArray &ba, amrex::Geometry &geom, amrex::MultiFab *cons, amrex::MultiFab *z_phys)
Definition: ERF_ShocInterface.cpp:112
view_1d phis
Definition: ERF_ShocInterface.H:592
view_2d shoc_evap
Definition: ERF_ShocInterface.H:667
typename ekat::KokkosTypes< KokkosDefaultDevice >::template view_2d< const Real > sview_2d_const
Definition: ERF_ShocInterface.H:52
view_3d horiz_wind
Definition: ERF_ShocInterface.H:641
void check_flux_state_consistency(const double dt)
Definition: ERF_ShocInterface.cpp:751
typename SHF::view_1d< Real > view_1d
Definition: ERF_ShocInterface.H:47
view_2d wqls_sec
Definition: ERF_ShocInterface.H:663
amrex::MultiFab * m_cons
Definition: ERF_ShocInterface.H:557
view_1d ke_b
Definition: ERF_ShocInterface.H:673
view_3d_strided qtracers
Definition: ERF_ShocInterface.H:630
ekat::KokkosTypes< KokkosDefaultDevice > KT
Definition: ERF_ShocInterface.H:40
view_1d upwp_sfc
Definition: ERF_ShocInterface.H:595
view_1d ustar
Definition: ERF_ShocInterface.H:646
view_2d tke_copy
Definition: ERF_ShocInterface.H:613
view_2d wm_zt
Definition: ERF_ShocInterface.H:622
scream::PhysicsFunctions< KokkosDefaultDevice > PF
Definition: ERF_ShocInterface.H:38
void finalize_impl()
Definition: ERF_ShocInterface.cpp:722
view_2d w3
Definition: ERF_ShocInterface.H:662
view_1d se_a
Definition: ERF_ShocInterface.H:676
void initialize_impl()
Definition: ERF_ShocInterface.cpp:538
ekat::Unmanaged< typename KT::template view_2d< ScalarT > > uview_2d
Definition: ERF_ShocInterface.H:62
view_1d wv_b
Definition: ERF_ShocInterface.H:674
ekat::WorkspaceManager< Spack, KT::Device > WSM
Definition: ERF_ShocInterface.H:57
typename SHF::view_3d< const Spack > view_3d_const
Definition: ERF_ShocInterface.H:54
bool m_first_step
Definition: ERF_ShocInterface.H:563
Int m_num_tracers
Definition: ERF_ShocInterface.H:540
std::string name() const
Definition: ERF_ShocInterface.H:107
int m_step
Definition: ERF_ShocInterface.H:548
sview_2d surf_mom_flux
Definition: ERF_ShocInterface.H:599
typename SHF::view_2d< SHF::Spack > view_2d
Definition: ERF_ShocInterface.H:49
view_2d zi_grid
Definition: ERF_ShocInterface.H:620
view_2d thlm
Definition: ERF_ShocInterface.H:624
int m_lev
Definition: ERF_ShocInterface.H:545
view_2d wtracer_sfc
Definition: ERF_ShocInterface.H:621
SHOCPostprocess shoc_postprocess
Definition: ERF_ShocInterface.H:580
typename SHF::Spack Spack
Definition: ERF_ShocInterface.H:43
SHF::SHOCInputOutput input_output
Definition: ERF_ShocInterface.H:570
Buffer m_buffer
Definition: ERF_ShocInterface.H:566
void mf_to_kokkos_buffers()
Definition: ERF_ShocInterface.cpp:376
view_2d shoc_ql2
Definition: ERF_ShocInterface.H:634
view_2d tk
Definition: ERF_ShocInterface.H:640
view_2d qv
Definition: ERF_ShocInterface.H:606
view_2d wthl_sec
Definition: ERF_ShocInterface.H:657
amrex::BoxArray m_ba
Definition: ERF_ShocInterface.H:554
view_2d tke
Definition: ERF_ShocInterface.H:612
view_2d zt_grid
Definition: ERF_ShocInterface.H:619
view_2d T_mid
Definition: ERF_ShocInterface.H:601
view_2d shoc_s
Definition: ERF_ShocInterface.H:611
void dealloc_buffers()
Definition: ERF_ShocInterface.cpp:268
Int hdtime
Definition: ERF_ShocInterface.H:542
view_2d z_int
Definition: ERF_ShocInterface.H:610
void apply_turbulent_mountain_stress()
Definition: ERF_ShocInterface.cpp:735
typename SHF::view_3d< Spack > view_3d
Definition: ERF_ShocInterface.H:53
view_2d isotropy
Definition: ERF_ShocInterface.H:665
typename SHF::IntSmallPack IntSmallPack
Definition: ERF_ShocInterface.H:44
view_2d dz_zt
Definition: ERF_ShocInterface.H:687
amrex::Vector< int > m_col_offsets
Definition: ERF_ShocInterface.H:532
SHF::SHOCHistoryOutput history_output
Definition: ERF_ShocInterface.H:572
view_1d wpthlp_sfc
Definition: ERF_ShocInterface.H:593
view_1d wprtp_sfc
Definition: ERF_ShocInterface.H:594
@ cons
Definition: ERF_IndexDefines.H:140
Definition: ERF_ShocInterface.H:429
uview_2d< Spack > w3
Definition: ERF_ShocInterface.H:493
uview_2d< Spack > qc_copy
Definition: ERF_ShocInterface.H:480
uview_2d< Spack > dse
Definition: ERF_ShocInterface.H:478
uview_1d< Real > upwp_sfc
Definition: ERF_ShocInterface.H:447
uview_2d< Spack > vw_sec
Definition: ERF_ShocInterface.H:492
uview_1d< Real > wpthlp_sfc
Definition: ERF_ShocInterface.H:445
uview_2d< Spack > thv
Definition: ERF_ShocInterface.H:469
uview_2d< Spack > wqls_sec
Definition: ERF_ShocInterface.H:494
uview_2d< Spack > wthl_sec
Definition: ERF_ShocInterface.H:488
uview_2d< Spack > thlm
Definition: ERF_ShocInterface.H:476
uview_2d< Spack > uw_sec
Definition: ERF_ShocInterface.H:491
uview_2d< Spack > z_mid
Definition: ERF_ShocInterface.H:465
uview_1d< Real > vpwp_sfc
Definition: ERF_ShocInterface.H:448
uview_2d< Spack > rrho
Definition: ERF_ShocInterface.H:467
uview_2d< Spack > wtke_sec
Definition: ERF_ShocInterface.H:490
uview_2d< Spack > rrho_i
Definition: ERF_ShocInterface.H:468
uview_2d< Spack > unused
Definition: ERF_ShocInterface.H:464
Spack * wsm_data
Definition: ERF_ShocInterface.H:505
static constexpr int num_2d_vector_int
Definition: ERF_ShocInterface.H:438
uview_2d< Spack > wtracer_sfc
Definition: ERF_ShocInterface.H:473
static constexpr int num_2d_vector_tr
Definition: ERF_ShocInterface.H:443
static constexpr int num_1d_scalar_nlev
Definition: ERF_ShocInterface.H:435
uview_2d< Spack > inv_exner
Definition: ERF_ShocInterface.H:475
uview_2d< Spack > qwthl_sec
Definition: ERF_ShocInterface.H:487
uview_2d< Spack > qw
Definition: ERF_ShocInterface.H:477
uview_2d< Spack > brunt
Definition: ERF_ShocInterface.H:495
uview_2d< Spack > shoc_mix
Definition: ERF_ShocInterface.H:482
uview_2d< Spack > tke_copy
Definition: ERF_ShocInterface.H:479
static constexpr int num_1d_scalar_ncol
Definition: ERF_ShocInterface.H:431
uview_2d< Spack > isotropy
Definition: ERF_ShocInterface.H:483
uview_2d< Spack > zt_grid
Definition: ERF_ShocInterface.H:471
uview_2d< Spack > w_sec
Definition: ERF_ShocInterface.H:484
uview_2d< Spack > thl_sec
Definition: ERF_ShocInterface.H:485
uview_2d< Spack > wm_zt
Definition: ERF_ShocInterface.H:474
uview_2d< Spack > qw_sec
Definition: ERF_ShocInterface.H:486
uview_2d< Spack > wqw_sec
Definition: ERF_ShocInterface.H:489
uview_2d< Spack > z_int
Definition: ERF_ShocInterface.H:466
uview_2d< Spack > shoc_ql2
Definition: ERF_ShocInterface.H:481
uview_1d< Real > wprtp_sfc
Definition: ERF_ShocInterface.H:446
uview_2d< Spack > zi_grid
Definition: ERF_ShocInterface.H:472
static constexpr int num_2d_vector_mid
Definition: ERF_ShocInterface.H:437
uview_1d< Spack > pref_mid
Definition: ERF_ShocInterface.H:462
uview_2d< Spack > dz
Definition: ERF_ShocInterface.H:470
Definition: ERF_ShocInterface.H:315
view_1d_const surf_sens_flux
Definition: ERF_ShocInterface.H:380
view_1d ice_flux
Definition: ERF_ShocInterface.H:383
view_1d vapor_flux
Definition: ERF_ShocInterface.H:381
view_1d_const phis
Definition: ERF_ShocInterface.H:377
int nlev
Definition: ERF_ShocInterface.H:367
view_1d heat_flux
Definition: ERF_ShocInterface.H:384
view_2d cldfrac_liq
Definition: ERF_ShocInterface.H:373
view_2d_const rrho
Definition: ERF_ShocInterface.H:368
KOKKOS_INLINE_FUNCTION void operator()(const Kokkos::TeamPolicy< KT::ExeSpace >::member_type &team) const
Definition: ERF_ShocInterface.H:319
view_3d_strided qtracers
Definition: ERF_ShocInterface.H:371
void set_variables(const int ncol_, const int nlev_, const view_2d_const &rrho_, const view_2d &qv_, const view_2d_const &qw_, const view_2d &qc_, const view_2d_const &qc_copy_, const view_2d &tke_, const view_2d_const &tke_copy_, const view_3d_strided &qtracers_, const view_2d_const &qc2_, const view_2d &cldfrac_liq_, const view_2d &inv_qc_relvar_, const view_2d &T_mid_, const view_2d_const &dse_, const view_2d_const &z_mid_, const view_1d_const phis_)
Definition: ERF_ShocInterface.H:387
view_2d_const qw
Definition: ERF_ShocInterface.H:370
bool compute_mass_and_energy_fluxes
Definition: ERF_ShocInterface.H:378
view_1d water_flux
Definition: ERF_ShocInterface.H:382
view_1d_const surf_evap
Definition: ERF_ShocInterface.H:379
view_2d_const qc_copy
Definition: ERF_ShocInterface.H:370
view_2d_const z_mid
Definition: ERF_ShocInterface.H:376
view_2d_const dse
Definition: ERF_ShocInterface.H:376
view_2d tke
Definition: ERF_ShocInterface.H:369
view_2d T_mid
Definition: ERF_ShocInterface.H:375
view_2d inv_qc_relvar
Definition: ERF_ShocInterface.H:374
view_2d qc
Definition: ERF_ShocInterface.H:369
void set_mass_and_energy_fluxes(const view_1d_const &surf_evap_, const view_1d_const &surf_sens_flux_, const view_1d &vapor_flux_, const view_1d &water_flux_, const view_1d &ice_flux_, const view_1d &heat_flux_)
Definition: ERF_ShocInterface.H:413
view_2d qv
Definition: ERF_ShocInterface.H:369
SHOCPostprocess()=default
int ncol
Definition: ERF_ShocInterface.H:367
view_2d_const qc2
Definition: ERF_ShocInterface.H:372
view_2d_const tke_copy
Definition: ERF_ShocInterface.H:370
Definition: ERF_ShocInterface.H:119
int ncol
Definition: ERF_ShocInterface.H:212
view_3d_strided qtracers
Definition: ERF_ShocInterface.H:223
int nlev
Definition: ERF_ShocInterface.H:212
view_2d cldfrac_liq_prev
Definition: ERF_ShocInterface.H:249
view_2d rrho
Definition: ERF_ShocInterface.H:232
view_2d_const pseudo_density
Definition: ERF_ShocInterface.H:217
sview_2d_const surf_mom_flux
Definition: ERF_ShocInterface.H:222
view_2d_const qc
Definition: ERF_ShocInterface.H:225
view_2d thlm
Definition: ERF_ShocInterface.H:245
view_1d_const phis
Definition: ERF_ShocInterface.H:219
view_1d_const surf_evap
Definition: ERF_ShocInterface.H:221
view_2d zt_grid
Definition: ERF_ShocInterface.H:236
view_2d cldfrac_liq
Definition: ERF_ShocInterface.H:248
view_2d inv_exner
Definition: ERF_ShocInterface.H:244
view_1d_const surf_sens_flux
Definition: ERF_ShocInterface.H:220
view_2d wm_zt
Definition: ERF_ShocInterface.H:243
KOKKOS_INLINE_FUNCTION void operator()(const Kokkos::TeamPolicy< KT::ExeSpace >::member_type &team) const
Definition: ERF_ShocInterface.H:123
view_2d qc_copy
Definition: ERF_ShocInterface.H:226
view_2d wtracer_sfc
Definition: ERF_ShocInterface.H:242
view_2d z_mid
Definition: ERF_ShocInterface.H:227
view_2d rrho_i
Definition: ERF_ShocInterface.H:233
view_2d_const omega
Definition: ERF_ShocInterface.H:218
view_2d_const T_mid
Definition: ERF_ShocInterface.H:214
view_2d cloud_frac
Definition: ERF_ShocInterface.H:247
Real z_surf
Definition: ERF_ShocInterface.H:213
view_2d_const p_int
Definition: ERF_ShocInterface.H:216
view_2d tke
Definition: ERF_ShocInterface.H:230
view_1d wprtp_sfc
Definition: ERF_ShocInterface.H:239
view_2d qw
Definition: ERF_ShocInterface.H:246
view_1d upwp_sfc
Definition: ERF_ShocInterface.H:240
view_1d wpthlp_sfc
Definition: ERF_ShocInterface.H:238
view_2d z_int
Definition: ERF_ShocInterface.H:228
view_2d_const p_mid
Definition: ERF_ShocInterface.H:215
view_1d vpwp_sfc
Definition: ERF_ShocInterface.H:241
view_2d dz
Definition: ERF_ShocInterface.H:235
view_2d shoc_s
Definition: ERF_ShocInterface.H:229
view_2d tke_copy
Definition: ERF_ShocInterface.H:231
view_2d zi_grid
Definition: ERF_ShocInterface.H:237
view_2d qv
Definition: ERF_ShocInterface.H:224
view_2d thv
Definition: ERF_ShocInterface.H:234
void set_variables(const int ncol_, const int nlev_, const Real z_surf_, const view_2d_const &T_mid_, const view_2d_const &p_mid_, const view_2d_const &p_int_, const view_2d_const &pseudo_density_, const view_2d_const &omega_, const view_1d_const &phis_, const view_1d_const &surf_sens_flux_, const view_1d_const &surf_evap_, const sview_2d_const &surf_mom_flux_, const view_3d_strided &qtracers_, const view_2d &qv_, const view_2d_const &qc_, const view_2d &qc_copy_, const view_2d &tke_, const view_2d &tke_copy_, const view_2d &z_mid_, const view_2d &z_int_, const view_2d &dse_, const view_2d &rrho_, const view_2d &rrho_i_, const view_2d &thv_, const view_2d &dz_, const view_2d &zt_grid_, const view_2d &zi_grid_, const view_1d &wpthlp_sfc_, const view_1d &wprtp_sfc_, const view_1d &upwp_sfc_, const view_1d &vpwp_sfc_, const view_2d &wtracer_sfc_, const view_2d &wm_zt_, const view_2d &inv_exner_, const view_2d &thlm_, const view_2d &qw_, const view_2d &cldfrac_liq_, const view_2d &cldfrac_liq_prev_)
Definition: ERF_ShocInterface.H:252
Definition: ERF_DataStruct.H:123