update for gpu