Faster version without GPU array dump in each step