i965: Use brw_stage_state for WM data as well.