3D finite difference code based on pycuda and mpi