Update PlasmaParticleContainer.cpp

Hi-PACE · Feb 21, 2025 · f00732d · f00732d
1 parent 4a27702
commit f00732d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/particles/plasma/PlasmaParticleContainer.cpp b/src/particles/plasma/PlasmaParticleContainer.cpp
@@ -789,10 +789,11 @@ PlasmaToBeam (MultiBeam& beams, const amrex::Vector< std::string > beamnames, co
         auto ptd_beam = beam_elec->getBeamSlice(WhichBeamSlice::This).getParticleTileData();
 
         amrex::Gpu::DeviceScalar<uint32_t> ip_elec(0); 
+        uint32_t * AMREX_RESTRICT p_ip_elec = ip_elec.dataPtr();
         amrex::ParallelFor(num_particles,
             [=] AMREX_GPU_DEVICE (int ip) {
                 if (ptd_plasma.id(ip) != 2){
-                    const long pid_beam = amrex::Gpu::Atomic::Add(ip_elec.dataPtr(), 1u);
+                    const long pid_beam = amrex::Gpu::Atomic::Add(ip_elec, 1u);
                     const long pidx_beam = pid + old_size;
                     ptd_beam.rdata(BeamIdx::x)[pidx_beam] = ptd_plasma.pos(0, ip);
                     ptd_beam.rdata(BeamIdx::y)[pidx_beam] = ptd_plasma.pos(1, ip);