struct intel_connector;
 struct intel_encoder;
+struct intel_atomic_state;
 struct intel_crtc_state;
 struct intel_initial_plane_config;
 struct intel_crtc;
        int (*compute_intermediate_wm)(struct drm_device *dev,
                                       struct intel_crtc *intel_crtc,
                                       struct intel_crtc_state *newstate);
-       void (*initial_watermarks)(struct intel_crtc_state *cstate);
-       void (*optimize_watermarks)(struct intel_crtc_state *cstate);
+       void (*initial_watermarks)(struct intel_atomic_state *state,
+                                  struct intel_crtc_state *cstate);
+       void (*atomic_update_watermarks)(struct intel_atomic_state *state,
+                                        struct intel_crtc_state *cstate);
+       void (*optimize_watermarks)(struct intel_atomic_state *state,
+                                   struct intel_crtc_state *cstate);
        int (*compute_global_watermarks)(struct drm_atomic_state *state);
        void (*update_wm)(struct intel_crtc *crtc);
        int (*modeset_calc_cdclk)(struct drm_atomic_state *state);
 
        struct drm_plane_state *old_pri_state =
                drm_atomic_get_existing_plane_state(old_state, primary);
        bool modeset = needs_modeset(&pipe_config->base);
+       struct intel_atomic_state *old_intel_state =
+               to_intel_atomic_state(old_state);
 
        if (old_pri_state) {
                struct intel_plane_state *primary_state =
         * us to.
         */
        if (dev_priv->display.initial_watermarks != NULL)
-               dev_priv->display.initial_watermarks(pipe_config);
+               dev_priv->display.initial_watermarks(old_intel_state,
+                                                    pipe_config);
        else if (pipe_config->update_wm_pre)
                intel_update_watermarks(crtc);
 }
        struct drm_i915_private *dev_priv = to_i915(dev);
        struct intel_crtc *intel_crtc = to_intel_crtc(crtc);
        int pipe = intel_crtc->pipe;
+       struct intel_atomic_state *old_intel_state =
+               to_intel_atomic_state(old_state);
 
        if (WARN_ON(intel_crtc->active))
                return;
        intel_color_load_luts(&pipe_config->base);
 
        if (dev_priv->display.initial_watermarks != NULL)
-               dev_priv->display.initial_watermarks(intel_crtc->config);
+               dev_priv->display.initial_watermarks(old_intel_state, intel_crtc->config);
        intel_enable_pipe(intel_crtc);
 
        if (intel_crtc->config->has_pch_encoder)
        struct intel_crtc *intel_crtc = to_intel_crtc(crtc);
        int pipe = intel_crtc->pipe, hsw_workaround_pipe;
        enum transcoder cpu_transcoder = intel_crtc->config->cpu_transcoder;
+       struct intel_atomic_state *old_intel_state =
+               to_intel_atomic_state(old_state);
 
        if (WARN_ON(intel_crtc->active))
                return;
                intel_ddi_enable_transcoder_func(crtc);
 
        if (dev_priv->display.initial_watermarks != NULL)
-               dev_priv->display.initial_watermarks(pipe_config);
+               dev_priv->display.initial_watermarks(old_intel_state,
+                                                    pipe_config);
        else
                intel_update_watermarks(intel_crtc);
 
                intel_cstate = to_intel_crtc_state(crtc->state);
 
                if (dev_priv->display.optimize_watermarks)
-                       dev_priv->display.optimize_watermarks(intel_cstate);
+                       dev_priv->display.optimize_watermarks(intel_state,
+                                                             intel_cstate);
        }
 
        for_each_crtc_in_state(state, crtc, old_crtc_state, i) {
        struct intel_crtc *intel_crtc = to_intel_crtc(crtc);
        struct intel_crtc_state *intel_cstate =
                to_intel_crtc_state(crtc->state);
-       struct intel_crtc_state *old_intel_state =
+       struct intel_crtc_state *old_intel_cstate =
                to_intel_crtc_state(old_crtc_state);
+       struct intel_atomic_state *old_intel_state =
+               to_intel_atomic_state(old_crtc_state->state);
        bool modeset = needs_modeset(crtc->state);
-       enum pipe pipe = intel_crtc->pipe;
 
        /* Perform vblank evasion around commit operation */
        intel_pipe_update_start(intel_crtc);
                intel_color_load_luts(crtc->state);
        }
 
-       if (intel_cstate->update_pipe) {
-               intel_update_pipe_config(intel_crtc, old_intel_state);
-       } else if (INTEL_GEN(dev_priv) >= 9) {
+       if (intel_cstate->update_pipe)
+               intel_update_pipe_config(intel_crtc, old_intel_cstate);
+       else if (INTEL_GEN(dev_priv) >= 9)
                skl_detach_scalers(intel_crtc);
 
-               I915_WRITE(PIPE_WM_LINETIME(pipe),
-                          intel_cstate->wm.skl.optimal.linetime);
-       }
+       if (dev_priv->display.atomic_update_watermarks)
+               dev_priv->display.atomic_update_watermarks(old_intel_state,
+                                                          intel_cstate);
 }
 
 static void intel_finish_crtc_commit(struct drm_crtc *crtc,
 {
        struct drm_i915_private *dev_priv = to_i915(dev);
        struct drm_atomic_state *state;
+       struct intel_atomic_state *intel_state;
        struct drm_crtc *crtc;
        struct drm_crtc_state *cstate;
        struct drm_modeset_acquire_ctx ctx;
        if (WARN_ON(IS_ERR(state)))
                goto fail;
 
+       intel_state = to_intel_atomic_state(state);
+
        /*
         * Hardware readout is the only time we don't want to calculate
         * intermediate watermarks (since we don't trust the current
         * watermarks).
         */
-       to_intel_atomic_state(state)->skip_intermediate_wm = true;
+       intel_state->skip_intermediate_wm = true;
 
        ret = intel_atomic_check(dev, state);
        if (ret) {
                struct intel_crtc_state *cs = to_intel_crtc_state(cstate);
 
                cs->wm.need_postvbl_update = true;
-               dev_priv->display.optimize_watermarks(cs);
+               dev_priv->display.optimize_watermarks(intel_state, cs);
        }
 
 put_state:
 
        return 0;
 }
 
+static void skl_atomic_update_crtc_wm(struct intel_atomic_state *state,
+                                     struct intel_crtc_state *cstate)
+{
+       struct intel_crtc *crtc = to_intel_crtc(cstate->base.crtc);
+       struct drm_i915_private *dev_priv = to_i915(state->base.dev);
+       struct skl_pipe_wm *pipe_wm = &cstate->wm.skl.optimal;
+       enum pipe pipe = crtc->pipe;
+
+       I915_WRITE(PIPE_WM_LINETIME(pipe), pipe_wm->linetime);
+}
+
 static void skl_update_wm(struct intel_crtc *intel_crtc)
 {
        struct drm_device *dev = intel_crtc->base.dev;
        ilk_write_wm_values(dev_priv, &results);
 }
 
-static void ilk_initial_watermarks(struct intel_crtc_state *cstate)
+static void ilk_initial_watermarks(struct intel_atomic_state *state,
+                                  struct intel_crtc_state *cstate)
 {
        struct drm_i915_private *dev_priv = to_i915(cstate->base.crtc->dev);
        struct intel_crtc *intel_crtc = to_intel_crtc(cstate->base.crtc);
        mutex_unlock(&dev_priv->wm.wm_mutex);
 }
 
-static void ilk_optimize_watermarks(struct intel_crtc_state *cstate)
+static void ilk_optimize_watermarks(struct intel_atomic_state *state,
+                                   struct intel_crtc_state *cstate)
 {
        struct drm_i915_private *dev_priv = to_i915(cstate->base.crtc->dev);
        struct intel_crtc *intel_crtc = to_intel_crtc(cstate->base.crtc);
        if (INTEL_GEN(dev_priv) >= 9) {
                skl_setup_wm_latency(dev_priv);
                dev_priv->display.update_wm = skl_update_wm;
+               dev_priv->display.atomic_update_watermarks = skl_atomic_update_crtc_wm;
                dev_priv->display.compute_global_watermarks = skl_compute_wm;
        } else if (HAS_PCH_SPLIT(dev_priv)) {
                ilk_setup_wm_latency(dev_priv);