ECP-WarpX / WarpX

WarpX is an advanced electromagnetic & electrostatic Particle-In-Cell code.
https://ecp-warpx.github.io
Other
303 stars 191 forks source link

Comparison of computing speed between yee solver and psatd solver in RZ #2564

Open wangjia-ai opened 2 years ago

wangjia-ai commented 2 years ago

Dear developers: Recently, I compiled warpx with psatd in RZ. But the calculation speed is too slow. The input file and output with psatd in RZ are inputs_2d.txt
image The input file and output with yee in RZ are inputs_2d.txt image The compilation choice is -DWarpX_DIMS=RZ -DWarpX_COMPUTE=CUDA -DWarpX_OPENPMD=OFF -DWarpX_PSATD=ON -DWarpX_QED=OFF. The GPU in the server is NVIDIA RTX3090 24GB. Something wrong with the code or the input file above ?? Thanks!!

dpgrote commented 2 years ago

It is expected that the RZ spectral solver will be slower than the FDTD version since it entails significantly more operations. At the end of the simulation, WarpX will print out profiling information showing the timings of the most expensive routines. Can you post that here for the PSATD case?

wangjia-ai commented 2 years ago

In order to improve the calculation speed, I have reduced the number of meshes and the number of macro particles in the cell. inputs_2d_psatd.txt

image image image image image