Cross-platform implementation of Particle-In-Cell method for simulation of high-temperature and fusion plasma by means of hybrid supercomputers equipped with GPU or Intel Xeon Phi accelerators

Abstract

A new Python-based Particle-In-Cell code is presented. The code uses leapfrog particle pusher. The important feature of our code is that all the particles are pushed at once, thus the code is vectorized to improve performance. Electric field is given by Poisson equation with Least squares solver. The code involves collision simulation by PIC-MC method. Both large-scale (MPI) and fine-grain parallelization are being used. The implementation is based on the efficient NumPy library in Python language with the help of Dask package to improve Numpy performance. GPU implementation involves PyCUDA and the performance with Intel Xeon processors and Intel Xeon Phi accelerators is supported by high-performance Intel Python.