if (drm_WARN_ON(display->drm, crtc->active))
                return;
-       for_each_pipe_crtc_modeset_enable(display, pipe_crtc, new_crtc_state, i)
-               intel_dmc_enable_pipe(display, pipe_crtc->pipe);
+       for_each_pipe_crtc_modeset_enable(display, pipe_crtc, new_crtc_state, i) {
+               const struct intel_crtc_state *new_pipe_crtc_state =
+                       intel_atomic_get_new_crtc_state(state, pipe_crtc);
+
+               intel_dmc_enable_pipe(new_pipe_crtc_state);
+       }
 
        intel_encoders_pre_pll_enable(state, crtc);
 
 
        intel_encoders_post_pll_disable(state, crtc);
 
-       for_each_pipe_crtc_modeset_disable(display, pipe_crtc, old_crtc_state, i)
-               intel_dmc_disable_pipe(display, pipe_crtc->pipe);
+       for_each_pipe_crtc_modeset_disable(display, pipe_crtc, old_crtc_state, i) {
+               const struct intel_crtc_state *old_pipe_crtc_state =
+                       intel_atomic_get_old_crtc_state(state, pipe_crtc);
+
+               intel_dmc_disable_pipe(old_pipe_crtc_state);
+       }
 }
 
 /* Prefer intel_encoder_is_combo() */
 
        return false;
 }
 
-void intel_dmc_enable_pipe(struct intel_display *display, enum pipe pipe)
+void intel_dmc_enable_pipe(const struct intel_crtc_state *crtc_state)
 {
+       struct intel_display *display = to_intel_display(crtc_state);
+       struct intel_crtc *crtc = to_intel_crtc(crtc_state->uapi.crtc);
+       enum pipe pipe = crtc->pipe;
        enum intel_dmc_id dmc_id = PIPE_TO_DMC_ID(pipe);
 
        if (!is_valid_dmc_id(dmc_id) || !has_dmc_id_fw(display, dmc_id))
                intel_de_rmw(display, PIPEDMC_CONTROL(pipe), 0, PIPEDMC_ENABLE);
 }
 
-void intel_dmc_disable_pipe(struct intel_display *display, enum pipe pipe)
+void intel_dmc_disable_pipe(const struct intel_crtc_state *crtc_state)
 {
+       struct intel_display *display = to_intel_display(crtc_state);
+       struct intel_crtc *crtc = to_intel_crtc(crtc_state->uapi.crtc);
+       enum pipe pipe = crtc->pipe;
        enum intel_dmc_id dmc_id = PIPE_TO_DMC_ID(pipe);
 
        if (!is_valid_dmc_id(dmc_id) || !has_dmc_id_fw(display, dmc_id))
 
 
 enum pipe;
 struct drm_printer;
+struct intel_crtc_state;
 struct intel_display;
 struct intel_dmc_snapshot;
 
 void intel_dmc_init(struct intel_display *display);
 void intel_dmc_load_program(struct intel_display *display);
 void intel_dmc_disable_program(struct intel_display *display);
-void intel_dmc_enable_pipe(struct intel_display *display, enum pipe pipe);
-void intel_dmc_disable_pipe(struct intel_display *display, enum pipe pipe);
+void intel_dmc_enable_pipe(const struct intel_crtc_state *crtc_state);
+void intel_dmc_disable_pipe(const struct intel_crtc_state *crtc_state);
 void intel_dmc_block_pkgc(struct intel_display *display, enum pipe pipe,
                          bool block);
 void intel_dmc_start_pkgc_exit_at_start_of_undelayed_vblank(struct intel_display *display,
 
                drm_crtc_vblank_reset(&crtc->base);
 
                if (crtc_state->hw.active) {
-                       intel_dmc_enable_pipe(display, crtc->pipe);
+                       intel_dmc_enable_pipe(crtc_state);
                        intel_crtc_vblank_on(crtc_state);
                }
        }