Jimmy Walker warmed up for next week's defence of his PGA Championship title by taking a two-shot lead at the WGC Bridgestone Invitational.