US counting on 'diplomatic surge' for Iraq