add wb stage to part pipe for test