1 #ifndef VGRID_DISTRIBUTION_HPP
2 #define VGRID_DISTRIBUTION_HPP
10 template<
class Device>
16 View<double****,CLayout, Device>
f;
19 View<double*,CLayout, Device, Kokkos::MemoryTraits<Kokkos::Unmanaged>>
f_1D;
34 :
f(
"f", maxwellians.
size(), vgrid.nvr, pol_decomp.nnodes, vgrid.nvz),
42 for (
int i=0; i<
f.extent(0); i++){
43 for (
int imu=0; imu<
f.extent(1); imu++){
44 double vperp = imu*
dsmu;
45 for (
int inode=0; inode<
f.extent(2); inode++){
46 for (
int ivp=0; ivp<
f.extent(3); ivp++){
48 f(i,imu,inode,ivp) = maxwellians[i].get_f(vpara, vperp);
56 :
f(
"f", nsp, vgrid.nvr, pol_decomp.nnodes, vgrid.nvz),
66 KOKKOS_INLINE_FUNCTION
double&
operator()(
int isp,
int ivr,
int inode,
int ivz)
const {
return f(isp, ivr, inode, ivz);}
71 double*
data(){
return f.data();}
90 Kokkos::parallel_for(label, Kokkos::MDRangePolicy<Kokkos::Rank<4>,
exec_space>({0, 0, 0, 0}, {
f.extent(0),
f.extent(1),
f.extent(2),
f.extent(3)}), lambda_func);
96 constexpr
int ISP_ZERO = 0;
97 Kokkos::atomic_add(&(
f(ISP_ZERO, wt.
i_vr+0,i_node,wt.
i_vz+0)),value*wt.
w_00);
98 Kokkos::atomic_add(&(
f(ISP_ZERO, wt.
i_vr+0,i_node,wt.
i_vz+1)),value*wt.
w_01);
99 Kokkos::atomic_add(&(
f(ISP_ZERO, wt.
i_vr+1,i_node,wt.
i_vz+0)),value*wt.
w_10);
100 Kokkos::atomic_add(&(
f(ISP_ZERO, wt.
i_vr+1,i_node,wt.
i_vz+1)),value*wt.
w_11);
104 KOKKOS_INLINE_FUNCTION
static void scatter(
const View<double***,CLayout,Device>& view,
int i_node,
const VGridWeights& wt,
double value){
105 Kokkos::atomic_add(&(view(wt.
i_vr+0,i_node,wt.
i_vz+0)),value*wt.
w_00);
106 Kokkos::atomic_add(&(view(wt.
i_vr+0,i_node,wt.
i_vz+1)),value*wt.
w_01);
107 Kokkos::atomic_add(&(view(wt.
i_vr+1,i_node,wt.
i_vz+0)),value*wt.
w_10);
108 Kokkos::atomic_add(&(view(wt.
i_vr+1,i_node,wt.
i_vz+1)),value*wt.
w_11);
112 KOKKOS_INLINE_FUNCTION
static void scatter(
const View<double****,CLayout,Device>& view,
int i_node,
const VGridWeights& wt,
double value){
113 constexpr
int ISP_ZERO = 0;
114 Kokkos::atomic_add(&(view(ISP_ZERO,wt.
i_vr+0,i_node,wt.
i_vz+0)),value*wt.
w_00);
115 Kokkos::atomic_add(&(view(ISP_ZERO,wt.
i_vr+0,i_node,wt.
i_vz+1)),value*wt.
w_01);
116 Kokkos::atomic_add(&(view(ISP_ZERO,wt.
i_vr+1,i_node,wt.
i_vz+0)),value*wt.
w_10);
117 Kokkos::atomic_add(&(view(ISP_ZERO,wt.
i_vr+1,i_node,wt.
i_vz+1)),value*wt.
w_11);
122 constexpr
int ISP_ZERO = 0;
153 KOKKOS_INLINE_FUNCTION
int n_vr()
const {
161 KOKKOS_INLINE_FUNCTION
int n_vz()
const{
165 KOKKOS_INLINE_FUNCTION
int size()
const{
KOKKOS_INLINE_FUNCTION int n_vr() const
Definition: vgrid_distribution.hpp:153
KOKKOS_INLINE_FUNCTION int size() const
Definition: vgrid_distribution.hpp:165
double w_00
Definition: vgrid_weights.hpp:10
Definition: velocity_grid.hpp:8
double get_smu_n(int imu) const
Definition: vgrid_distribution.hpp:170
VGridDistribution(const VelocityGrid &vgrid, const DomainDecomposition< DeviceType > &pol_decomp, const std::vector< Maxwellian > &maxwellians)
Definition: vgrid_distribution.hpp:33
KOKKOS_INLINE_FUNCTION double gather(int i_node, const VGridWeights &wt) const
Definition: vgrid_distribution.hpp:121
double smu_max
max mu
Definition: vgrid_distribution.hpp:26
static KOKKOS_INLINE_FUNCTION double gather(const T &view, int i_node, const VGridWeights &wt)
Definition: vgrid_distribution.hpp:131
double dvp
grid spacing in parallel velocity
Definition: vgrid_distribution.hpp:25
double w_01
Definition: vgrid_weights.hpp:11
View< double ****, CLayout, Device > f
Definition: vgrid_distribution.hpp:16
KOKKOS_INLINE_FUNCTION int n_species() const
Definition: vgrid_distribution.hpp:149
KOKKOS_INLINE_FUNCTION double & operator()(int isp, int ivr, int inode, int ivz) const
Definition: vgrid_distribution.hpp:66
static KOKKOS_INLINE_FUNCTION void scatter(const View< double ***, CLayout, Device > &view, int i_node, const VGridWeights &wt, double value)
Definition: vgrid_distribution.hpp:104
double inv_mu0_factor
Definition: vgrid_distribution.hpp:28
double w_11
Definition: vgrid_weights.hpp:13
VGridDistribution()
Definition: vgrid_distribution.hpp:30
KOKKOS_INLINE_FUNCTION int n_vz() const
Definition: vgrid_distribution.hpp:161
View< double *, CLayout, Device, Kokkos::MemoryTraits< Kokkos::Unmanaged > > f_1D
Definition: vgrid_distribution.hpp:19
VGridDistribution(int nsp, const VelocityGrid &vgrid, const DomainDecomposition< DeviceType > &pol_decomp)
Definition: vgrid_distribution.hpp:55
static KOKKOS_INLINE_FUNCTION void scatter(const View< double ****, CLayout, Device > &view, int i_node, const VGridWeights &wt, double value)
Definition: vgrid_distribution.hpp:112
typename HostType::execution_space exec_space
Definition: vgrid_distribution.hpp:14
Definition: vgrid_distribution.hpp:11
int i_vr
Definition: vgrid_weights.hpp:8
int i_vz
Definition: vgrid_weights.hpp:9
double dsmu
grid spacing in mu
Definition: vgrid_distribution.hpp:27
void for_each_element(const std::string label, F lambda_func) const
Definition: vgrid_distribution.hpp:89
Definition: vgrid_weights.hpp:7
double * data()
Definition: vgrid_distribution.hpp:71
void parallel_for(const std::string name, int n_ptl, Function func, Option option, HostAoSoA aosoa_h, DeviceAoSoA aosoa_d)
Definition: streamed_parallel_for.hpp:252
KOKKOS_INLINE_FUNCTION int n_nodes() const
Definition: vgrid_distribution.hpp:157
void for_all_elements(const std::string label, F lambda_func) const
Definition: vgrid_distribution.hpp:79
KOKKOS_INLINE_FUNCTION double & operator[](int i) const
Definition: vgrid_distribution.hpp:69
double w_10
Definition: vgrid_weights.hpp:12
static KOKKOS_INLINE_FUNCTION double normed_gather(const T &view, int iw, int i_node, const VGridWeights &wt, const T &norm_view)
Definition: vgrid_distribution.hpp:141
double vp_max
max parallel velocity
Definition: vgrid_distribution.hpp:24
KOKKOS_INLINE_FUNCTION void scatter(int i_node, const VGridWeights &wt, double value) const
Definition: vgrid_distribution.hpp:95