3 #include <Cabana_AoSoA.hpp>
4 #include <Cabana_DeepCopy.hpp>
5 #include <Kokkos_Core.hpp>
74 template<
class Device>
105 Cabana::AoSoA<ParticleDataTypes,HostType,VEC_LEN>
particles;
121 Cabana::AoSoA<PhaseDataTypes,HostType,VEC_LEN>
phase0;
122 Cabana::AoSoA<PhaseDataTypes,Device,VEC_LEN>
phase0_d;
153 Species(
int idx_in,
int nonadiabatic_idx_in,
bool is_electron_in,
bool is_adiabatic_in,
KinType kintype_in,
double mass_in,
double charge_in,
double charge_eu_in,
bool is_deltaf_in,
209 auto ph = Cabana::slice<PtlSlice::Ph>(
particles);
210 auto ct = Cabana::slice<PtlSlice::Ct>(
particles);
211 auto gid = Cabana::slice<PtlSlice::Gid>(
particles);
213 auto flag = Cabana::slice<PtlSlice::Flag>(
particles);
220 long long int gid_offset = 0;
224 for (
int i=0;i<
n_ptl;i++){
226 gid(i) = gid_offset + i+1;
231 for (
int j=0;j<6;j++) ph(i, j) = gid(i) + (j)*0.1;
232 for (
int j=0;j<3;j++) ct(i, j) = gid(i) + (j+6)*0.1;
239 for (
int j=0;j<6;j++) ph(i, j) = gid(i) + (j)*0.1;
240 for (
int j=0;j<3;j++) ct(i, j) = gid(i) + (j+6)*0.1;
255 particles_d = Cabana::AoSoA<ParticleDataTypes,Device,VEC_LEN>();
287 particles = Cabana::AoSoA<ParticleDataTypes,HostType,VEC_LEN>();
381 int last_ptl_index =
n_ptl - 1;
382 auto ph = Cabana::slice<PtlSlice::Ph>(
particles_d);
383 auto ct = Cabana::slice<PtlSlice::Ct>(
particles_d);
384 auto gid = Cabana::slice<PtlSlice::Gid>(
particles_d);
386 auto flag = Cabana::slice<PtlSlice::Flag>(
particles_d);
391 for (
int j=0;j<6;j++) ph(i, j) = ph(last_ptl_index, j);
392 for (
int j=0;j<3;j++) ct(i, j) = ct(last_ptl_index, j);
395 flag(i) = flag(last_ptl_index);
410 int last_ptl_index =
n_ptl - 1;
411 auto ph = Cabana::slice<PtlSlice::Ph>(
phase0_d);
415 for (
int j=0;j<6;j++) ph(i, j) = ph(last_ptl_index, j);
437 inline void back_up_SoA(Cabana::AoSoA<ParticleDataTypes,Device,VEC_LEN>& backup_SoA,
int offset,
int n)
const{
438 auto ph_b = Cabana::slice<PtlSlice::Ph>(backup_SoA);
439 auto ct_b = Cabana::slice<PtlSlice::Ct>(backup_SoA);
440 auto gid_b = Cabana::slice<PtlSlice::Gid>(backup_SoA);
442 auto flag_b = Cabana::slice<PtlSlice::Flag>(backup_SoA);
445 auto ph = Cabana::slice<PtlSlice::Ph>(
particles_d);
446 auto ct = Cabana::slice<PtlSlice::Ct>(
particles_d);
447 auto gid = Cabana::slice<PtlSlice::Gid>(
particles_d);
449 auto flag = Cabana::slice<PtlSlice::Flag>(
particles_d);
452 Kokkos::parallel_for(
"backup_first_soa", Kokkos::RangePolicy<ExSpace>( 0, n ), KOKKOS_LAMBDA(
const int i ){
453 int i_offset = i + offset;
455 for (
int j=0;j<6;j++) ph_b(i, j) = ph(i_offset, j);
456 for (
int j=0;j<3;j++) ct_b(i, j) = ct(i_offset, j);
457 gid_b(i) = gid(i_offset);
459 flag_b(i) = flag(i_offset);
466 inline void restore_backup_SoA(Cabana::AoSoA<ParticleDataTypes,Device,VEC_LEN>& backup_SoA,
int offset,
int n)
const{
467 auto ph_b = Cabana::slice<PtlSlice::Ph>(backup_SoA);
468 auto ct_b = Cabana::slice<PtlSlice::Ct>(backup_SoA);
469 auto gid_b = Cabana::slice<PtlSlice::Gid>(backup_SoA);
471 auto flag_b = Cabana::slice<PtlSlice::Flag>(backup_SoA);
474 auto ph = Cabana::slice<PtlSlice::Ph>(
particles_d);
475 auto ct = Cabana::slice<PtlSlice::Ct>(
particles_d);
476 auto gid = Cabana::slice<PtlSlice::Gid>(
particles_d);
478 auto flag = Cabana::slice<PtlSlice::Flag>(
particles_d);
481 Kokkos::parallel_for(
"backup_first_soa", Kokkos::RangePolicy<ExSpace>( 0, n ), KOKKOS_LAMBDA(
const int i ){
482 int i_offset = i + offset;
484 for (
int j=0;j<6;j++) ph(i_offset, j) = ph_b(i, j);
485 for (
int j=0;j<3;j++) ct(i_offset, j) = ct_b(i, j);
486 gid(i_offset) = gid_b(i);
488 flag(i_offset) = flag_b(i);
499 inline void for_particle_range(
int begin_idx,
int end_idx,
const std::string label, F lambda_func)
const {
500 if(end_idx <= begin_idx)
return;
503 int first_soa = begin_idx/VEC_LEN;
504 int n_other_ptl_in_first_soa = begin_idx - first_soa*VEC_LEN;
505 bool first_soa_is_partial = (n_other_ptl_in_first_soa>0);
506 int last_soa = (end_idx-1)/VEC_LEN;
507 int n_other_ptl_in_last_soa = (last_soa+1)*VEC_LEN - end_idx;
508 bool last_soa_is_partial = (n_other_ptl_in_last_soa>0);
511 int first_item_in_shifted_range = first_soa*VEC_LEN;
513 int first_item_in_shifted_range = first_soa;
516 Cabana::AoSoA<ParticleDataTypes,Device,VEC_LEN> ptl_first_soa;
517 Cabana::AoSoA<ParticleDataTypes,Device,VEC_LEN> ptl_last_soa;
518 if(first_soa_is_partial){
520 ptl_first_soa = Cabana::AoSoA<ParticleDataTypes,Device,VEC_LEN>(
"ptl_first_soa", n_other_ptl_in_first_soa);
521 back_up_SoA(ptl_first_soa, first_soa*VEC_LEN, n_other_ptl_in_first_soa);
523 if(last_soa_is_partial){
525 ptl_last_soa = Cabana::AoSoA<ParticleDataTypes,Device,VEC_LEN>(
"ptl_last_soa", n_other_ptl_in_last_soa);
526 back_up_SoA(ptl_last_soa, end_idx, n_other_ptl_in_last_soa);
530 Kokkos::RangePolicy<ExSpace> particle_range_policy( first_item_in_shifted_range,
p_range<DeviceType>(end_idx) );
534 if(first_soa_is_partial){
537 if(last_soa_is_partial){
552 if(!
owns_particles_d)
exit_XGC(
"\nSpecies tried to loop over particles on device, but doesn't own the device array.");
556 use_streaming =
false;
565 if((!send_ptl) && (!return_ptl)) use_streaming =
false;
580 #ifdef USE_EPUSH_LAUNCH_BOUNDS
581 # if !defined(PUSH_MAX_THREADS_PER_BLOCK) || !defined(PUSH_MIN_WARPS_PER_EU)
582 # error "USE_EPUSH_LAUNCH_BOUNDS requires PUSH_MAX_THREADS_PER_BLOCK and PUSH_MIN_WARPS_PER_EU to be defined"
584 Kokkos::RangePolicy<ExSpace, Kokkos::LaunchBounds<PUSH_MAX_THREADS_PER_BLOCK, PUSH_MIN_WARPS_PER_EU>>
588 exit_XGC(
"\nERROR: LaunchBounds::Custom specified, but USE_EPUSH_LAUNCH_BOUNDS is not defined\n");
591 Kokkos::RangePolicy<ExSpace>
613 for (
int i_simd = 0; i_simd<SIMD_SIZE; i_simd++){
614 int p_vec = inds.
a + i_simd;
615 ph0_loc[inds.
s].
r[p_vec] = ptl_loc[inds.
s].
ph.
r[p_vec];
616 ph0_loc[inds.
s].
z[p_vec] = ptl_loc[inds.
s].
ph.
z[p_vec];
617 ph0_loc[inds.
s].
phi[p_vec] = ptl_loc[inds.
s].
ph.
phi[p_vec];
618 ph0_loc[inds.
s].
rho[p_vec] = ptl_loc[inds.
s].
ph.
rho[p_vec];
619 ph0_loc[inds.
s].
w1[p_vec] = ptl_loc[inds.
s].
ph.
w1[p_vec];
620 ph0_loc[inds.
s].
w2[p_vec] = ptl_loc[inds.
s].
ph.
w2[p_vec];
668 backup_particles = Cabana::AoSoA<ParticleDataTypes,HostType,VEC_LEN>(
"backup_particles", 0);
686 for (
int i_simd = 0; i_simd<SIMD_SIZE; i_simd++){
687 int p_vec = inds.
a + i_simd;
688 part_one.
ph.
r[i_simd] = ph0_loc[inds.
s].
r[p_vec];
689 part_one.
ph.
z[i_simd] = ph0_loc[inds.
s].
z[p_vec];
690 part_one.
ph.
phi[i_simd] = ph0_loc[inds.
s].
phi[p_vec];
691 part_one.
ph.
rho[i_simd] = ph0_loc[inds.
s].
rho[p_vec];
692 part_one.
ph.
w1[i_simd] = ph0_loc[inds.
s].
w1[p_vec];
693 part_one.
ph.
w2[i_simd] = ph0_loc[inds.
s].
w2[p_vec];
699 long long int tmp_n_ptl =
n_ptl;
700 long long int out_n_ptl = 0;
701 MPI_Allreduce(&tmp_n_ptl, &out_n_ptl, 1, MPI_LONG_LONG_INT, MPI_SUM,
SML_COMM_WORLD);
704 return (
long long int)(
n_ptl);
710 int tmp_n_ptl =
n_ptl;
712 MPI_Allreduce(&tmp_n_ptl, &out_n_ptl, 1, MPI_INT, MPI_MAX,
SML_COMM_WORLD);
761 for (
int i_simd = 0; i_simd<SIMD_SIZE; i_simd++){
762 not_in_triangle[i_simd] = !grid_wts0.is_valid(i_simd);
768 for (
int i_simd = 0; i_simd<SIMD_SIZE; i_simd++){
769 if(!grid_wts0.is_valid(i_simd))
continue;
771 nearest_node[i_simd]=grid_wts0.node[i_simd] - pol_decomp.
node_offset;
772 not_in_poloidal_domain[i_simd] = (nearest_node[i_simd]<0 || nearest_node[i_simd]>=pol_decomp.
nnodes);
774 double temp_ev_norm = not_in_poloidal_domain[i_simd] ?
f0.fg_temp_ev(0) :
f0.fg_temp_ev(nearest_node[i_simd]);
777 const double& B = bmag[i_simd];
void calculate_global_f0_arrays(const Grid< DeviceType > &grid, const MagneticField< DeviceType > &magnetic_field)
Definition: species.cpp:493
Cabana::AoSoA< PhaseDataTypes, HostType, VEC_LEN > phase0
Definition: species.hpp:121
WeightEvoEq weight_evo_eq
Definition: species.hpp:91
bool stream_particles
Whether to stream particles between host and device if possible.
Definition: species.hpp:113
Definition: globals.hpp:84
KOKKOS_INLINE_FUNCTION VecPhase * ph0() const
Definition: species.hpp:604
KOKKOS_INLINE_FUNCTION int divide_and_round_up(int a, int b)
Definition: globals.hpp:203
bool owns_particles_d
Whether the species owns the device particle allocation right now.
Definition: species.hpp:110
void back_up_SoA(Cabana::AoSoA< ParticleDataTypes, Device, VEC_LEN > &backup_SoA, int offset, int n) const
Definition: species.hpp:437
KOKKOS_INLINE_FUNCTION double normalized_v_para(double c_m, double mass, double B, double temp_ev, double rho)
Definition: basic_physics.hpp:80
subroutine adjust_n_ptl_for_core_ptl(n_ptl)
Definition: load.F90:54
void set_spall_num_and_ptr(int idx, int n_ptl, int n_vecs, VecParticles *ptl)
void for_particle_range(int begin_idx, int end_idx, const std::string label, F lambda_func) const
Definition: species.hpp:499
constexpr double PROTON_MASS
Definition: constants.hpp:7
MarkerType
Definition: globals.hpp:110
Distribution< Device > f0
Species distribution in velocity space on local mesh nodes.
Definition: species.hpp:129
MPI_Comm SML_COMM_WORLD
Definition: my_mpi.cpp:4
Cabana::AoSoA< ParticleDataTypes, HostType, VEC_LEN > backup_particles
Copy of particles to be restored for RK2.
Definition: species.hpp:125
bool is_electron
Whether this species is the electrons.
Definition: species.hpp:79
Eq::Profile< Device > eq_mk_temp
Definition: species.hpp:143
void for_all_particles(const std::string label, F lambda_func, const PtlMvmt mvmt, LaunchBounds launch_bounds=LaunchBounds::Default)
Definition: species.hpp:550
bool dynamic_f0
Whether f0 can evolve in time.
Definition: species.hpp:96
void save_backup_particles()
Definition: species.hpp:625
void update_decomposed_f0_calculations(const DomainDecomposition< DeviceType > &pol_decomp, const Grid< DeviceType > &grid, const MagneticField< DeviceType > &magnetic_field, const VelocityGrid &vgrid)
Definition: species.cpp:438
double c2_2m
c2/2m
Definition: species.hpp:87
double rho[VEC_LEN]
Definition: particles.hpp:96
void copy_to_phase0(Species< Device > &species)
Definition: species.hpp:608
Definition: species.hpp:58
Simd< double > w1
Definition: particles.hpp:22
double c_m
c/m
Definition: species.hpp:86
Definition: species.hpp:57
constexpr double EV_2_J
Conversion rate ev to J.
Definition: constants.hpp:5
Definition: velocity_grid.hpp:8
bool default_streaming_option()
Definition: species.hpp:28
Eq::Profile< Device > eq_den
Definition: species.hpp:134
Definition: globals.hpp:89
KOKKOS_INLINE_FUNCTION VecParticles * ptl() const
Definition: species.hpp:600
Definition: grid_weights.hpp:47
KOKKOS_INLINE_FUNCTION double thermal_velocity(double mass, double temp_ev)
Definition: basic_physics.hpp:58
Definition: NamelistReader.hpp:193
KinType kintype
Whether the species is gyrokinetic or drift kinetic.
Definition: species.hpp:82
Eq::Profile< Device > eq_fg_temp
Definition: species.hpp:138
Definition: magnetic_field.hpp:12
void get_ptl_write_total_and_offsets(const DomainDecomposition< DeviceType > &pol_decomp, long long int &inum_total, long long int &ioff) const
Definition: species.cpp:578
int add_vec_buffer(int n_ptl)
Definition: particles.hpp:194
int idx
Index in all_species.
Definition: species.hpp:78
Definition: particles.hpp:92
int a
The index in the inner array of the AoSoA.
Definition: particles.hpp:150
KOKKOS_INLINE_FUNCTION double get_f0_fg_unit_velocity_lnode_h(int inode) const
Definition: species.hpp:746
Definition: particles.hpp:109
KOKKOS_INLINE_FUNCTION void bmag_interpol(const SimdVector &v, Simd< double > &bmag) const
Definition: magnetic_field.tpp:239
bool particles_are_backed_up
Whether particles are currently backed up.
Definition: species.hpp:118
int nonadiabatic_idx
Index of species skipping adiabatic species (for compatibility with fortran arrays) ...
Definition: species.hpp:81
bool default_residence_option()
Definition: species.hpp:35
int n_ptl
Number of particles.
Definition: species.hpp:104
Eq::Profile< Device > eq_mk_flow
Definition: species.hpp:145
KOKKOS_INLINE_FUNCTION void get_particle_velocity_and_nearest_node(const Grid< DeviceType > &grid, const MagneticField< DeviceType > &magnetic_field, const DomainDecomposition< DeviceType > &pol_decomp, SimdParticles &part, Simd< double > &smu, Simd< double > &vp, Simd< int > &nearest_node, Simd< bool > ¬_in_triangle, Simd< bool > ¬_in_poloidal_domain) const
Definition: species.hpp:751
Definition: streamed_parallel_for.hpp:16
KOKKOS_INLINE_FUNCTION double get_fg_gyro_radius(int inode, double smu_n, double bfield) const
Definition: species.hpp:723
int node_offset
Offset of first mesh node belonging to this MPI rank.
Definition: domain_decomposition.hpp:55
Definition: streamed_parallel_for.hpp:14
Eq::Profile< Device > eq_fg_flow
Definition: species.hpp:140
void set_buffer_phase0_d()
Definition: species.hpp:408
long long int get_total_n_ptl()
Definition: species.hpp:697
void set_buffer_particles_d()
Definition: species.hpp:379
Simd< double > rho
Definition: particles.hpp:21
Definition: species.hpp:59
int p_range< DeviceType >(int num_particle)
Definition: particles.hpp:187
int eq_flow_type
Definition: species.hpp:136
double charge_eu
Particle charge in eu.
Definition: species.hpp:85
Definition: species.hpp:51
int nnodes
Number of nodes belonging to this MPI rank.
Definition: domain_decomposition.hpp:56
void resize_particles(int new_n_ptl)
Definition: species.hpp:249
double mass
Particle mass.
Definition: species.hpp:83
KOKKOS_INLINE_FUNCTION double get_f0_eq_thermal_velocity(int inode) const
Definition: species.hpp:730
Species(int idx_in, int nonadiabatic_idx_in, bool is_electron_in, bool is_adiabatic_in, KinType kintype_in, double mass_in, double charge_in, double charge_eu_in, bool is_deltaf_in, int ncycles_in)
Definition: species.cpp:25
void write_ptl_checkpoint_files(const DomainDecomposition< DeviceType > &pol_decomp, const XGC_IO_Stream &stream, std::string sp_name)
Definition: species.cpp:603
void for_all_particles(const std::string label, F lambda_func) const
Definition: species.hpp:432
Cabana::AoSoA< ParticleDataTypes, Device, VEC_LEN > particles_d
Particles on device.
Definition: species.hpp:108
Definition: species.hpp:48
double w2[VEC_LEN]
Definition: particles.hpp:98
#define TIMER(N, F)
Definition: timer_macro.hpp:24
FAnalyticShape
Definition: globals.hpp:116
Eq::Profile< Device > eq_mk_den
Definition: species.hpp:144
RKRestorationMethod
Definition: species.hpp:68
void copy_particles_to_device_if_not_resident()
Definition: species.hpp:362
RKRestorationMethod RK_restoration_method
Currently, electrons must use first method and ions must use second.
Definition: species.hpp:116
Simd< double > r
Definition: particles.hpp:18
void resize_host_particles_to_match_device()
Definition: species.hpp:269
Definition: species.hpp:69
KOKKOS_INLINE_FUNCTION SimdVector & v()
Definition: particles.hpp:39
ReturnOpt return_opt
Definition: species.hpp:63
ReturnOpt
Definition: species.hpp:56
Option
Definition: streamed_parallel_for.hpp:13
void restore_particles_from_backup()
Definition: species.hpp:646
MarkerType marker_type
Marker type: reduced delta-f, total-f, full-f, or none (placeholder for adiabatic species) ...
Definition: species.hpp:89
Definition: globals.hpp:90
static std::vector< MemoryPrediction > estimate_memory_usage(NLReader::NamelistReader &nlr, const Grid< DeviceType > &grid, const DomainDecomposition< DeviceType > &pol_decomp, int species_idx)
Definition: species.cpp:56
SendOpt send_opt
Definition: species.hpp:62
double charge
Particle charge.
Definition: species.hpp:84
SimdPhase ph
Definition: particles.hpp:59
void copy_particles_from_device()
Definition: species.hpp:337
KOKKOS_INLINE_FUNCTION double get_f0_eq_thermal_velocity_lnode(int inode) const
Definition: species.hpp:736
void copy_particles_from_device_if_not_resident()
Definition: species.hpp:368
KOKKOS_INLINE_FUNCTION void wedge_modulo_phi(Simd< double > &phi_mod) const
Definition: grid.tpp:86
void unassign_host_particles()
Definition: species.hpp:286
void read_ptl_checkpoint_files(const DomainDecomposition< DeviceType > &pol_decomp, const XGC_IO_Stream &stream, std::string sp_name, bool n_ranks_is_same, int version)
Definition: species.cpp:653
int ncycles_between_sorts
Number of subcycles between sorts.
Definition: species.hpp:101
Definition: particles.hpp:58
Cabana::AoSoA< PhaseDataTypes, Device, VEC_LEN > phase0_d
Definition: species.hpp:122
int SML_COMM_RANK
Definition: my_mpi.cpp:5
KinType
Definition: globals.hpp:88
Species(SpeciesType sp_type, int n_ptl)
Definition: species.hpp:159
bool is_deltaf
Whether this species is deltaf.
Definition: species.hpp:98
Definition: xgc_io.hpp:24
VecPhase ph
Definition: particles.hpp:110
Definition: species.hpp:44
Definition: species.hpp:70
void set_min_max_num(int isp, int n_ptl)
int minimum_ptl_reservation
The minimum reservation size for particles.
Definition: species.hpp:103
int s
The index in the outer array of the AoSoA.
Definition: particles.hpp:149
KOKKOS_INLINE_FUNCTION double normalized_sqrt_mu(double B, double temp_ev, double mu)
Definition: basic_physics.hpp:90
Simd< double > z
Definition: particles.hpp:19
void copy_particles_to_device_if_resident()
Definition: species.hpp:350
Definition: species.hpp:52
void resize_device_particles(int new_n_ptl)
Definition: species.hpp:307
Definition: species.hpp:53
constexpr double UNIT_CHARGE
Charge of an electron (C)
Definition: constants.hpp:4
long long int get_max_gid() const
Definition: species.cpp:555
KOKKOS_INLINE_FUNCTION double get_f0_eq_thermal_velocity_lnode_h(int inode) const
Definition: species.hpp:742
void exit_XGC(std::string msg)
Definition: globals.hpp:37
void copy_particles_from_device_if_resident()
Definition: species.hpp:356
bool is_adiabatic
Whether this species is adiabatic.
Definition: species.hpp:80
Simd< double > phi
Definition: particles.hpp:20
FAnalyticShape f_analytic_shape
f_analytic_shape shape: Maxwellian, SlowingDown or None
Definition: species.hpp:90
Definition: magnetic_field.F90:1
static constexpr const Kokkos::Experimental::WorkItemProperty::HintLightWeight_t Async
Definition: space_settings.hpp:82
int n_backup_particles
Definition: species.hpp:126
Eq::Profile< Device > eq_flow
Definition: species.hpp:135
Definition: streamed_parallel_for.hpp:15
SendOpt
Definition: species.hpp:50
void read_f0_checkpoint_files(const DomainDecomposition< DeviceType > &pol_decomp, const XGC_IO_Stream &stream, std::string sp_name)
Definition: species.cpp:748
SimdConstants ct
Definition: particles.hpp:60
int eq_fg_flow_type
Definition: species.hpp:141
void write_f0_checkpoint_files(const DomainDecomposition< DeviceType > &pol_decomp, const XGC_IO_Stream &stream, std::string sp_name)
Definition: species.cpp:712
void copy_particles_to_device()
Definition: species.hpp:321
KOKKOS_INLINE_FUNCTION void restore_phase_from_phase0(const AoSoAIndices< Device > &inds, SimdParticles &part_one) const
Definition: species.hpp:684
Species(int n_ptl_in)
Definition: species.hpp:195
double phi[VEC_LEN]
Definition: particles.hpp:95
Simd< double > w2
Definition: particles.hpp:23
double r[VEC_LEN]
Definition: particles.hpp:93
GyroAverageMatrices< HostType > gyro_avg_matrices
Definition: species.hpp:149
Definition: species.hpp:75
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 void get_grid_weights(const MagneticField< Device > &magnetic_field, const SimdVector &v, const Simd< double > &psi, SimdVector2D &xff, SimdGridWeights< Order::One, PIT > &grid_wts) const
Definition: grid.tpp:32
Definition: species.hpp:44
Eq::Profile< Device > eq_temp
Definition: species.hpp:133
bool particles_resident_on_device
Whether the particles can reside on device.
Definition: species.hpp:112
Simd< double > mu
Definition: particles.hpp:52
Eq::Profile< Device > eq_fg_den
Definition: species.hpp:139
PtlMvmt(SendOpt send_opt, ReturnOpt return_opt)
Definition: species.hpp:65
static int get_initial_n_ptl(NLReader::NamelistReader &nlr, const Grid< DeviceType > &grid, const DomainDecomposition< DeviceType > &pol_decomp, int sml_special, int species_idx, bool verbose)
Definition: species.cpp:113
int ncycles
Number of subcycles.
Definition: species.hpp:100
int eq_mk_flow_type
Definition: species.hpp:146
WeightEvoEq
Definition: globals.hpp:121
Definition: profile.hpp:171
int collision_grid_index
Which collision grid to use.
Definition: species.hpp:131
Definition: particles.hpp:148
SpeciesType
Definition: globals.hpp:83
double z[VEC_LEN]
Definition: particles.hpp:94
void restore_backup_SoA(Cabana::AoSoA< ParticleDataTypes, Device, VEC_LEN > &backup_SoA, int offset, int n) const
Definition: species.hpp:466
Definition: distribution.hpp:10
void resize_device_particles()
Definition: species.hpp:292
int get_max_n_ptl()
Definition: species.hpp:708
LaunchBounds
Definition: species.hpp:421
Cabana::AoSoA< ParticleDataTypes, HostType, VEC_LEN > particles
Particles.
Definition: species.hpp:105
double w1[VEC_LEN]
Definition: particles.hpp:97
Definition: species.hpp:44