Improved overlapping of mpi communcation with GPU computation