}
 
 static int vc4_crtc_disable(struct drm_crtc *crtc,
+                           struct drm_encoder *encoder,
                            struct drm_atomic_state *state,
                            unsigned int channel)
 {
-       struct drm_encoder *encoder = vc4_get_crtc_encoder(crtc, state,
-                                                          drm_atomic_get_old_connector_state);
        struct vc4_encoder *vc4_encoder = to_vc4_encoder(encoder);
        struct vc4_crtc *vc4_crtc = to_vc4_crtc(crtc);
        struct drm_device *dev = crtc->dev;
        return 0;
 }
 
+static struct drm_encoder *vc4_crtc_get_encoder_by_type(struct drm_crtc *crtc,
+                                                       enum vc4_encoder_type type)
+{
+       struct drm_encoder *encoder;
+
+       drm_for_each_encoder(encoder, crtc->dev) {
+               struct vc4_encoder *vc4_encoder = to_vc4_encoder(encoder);
+
+               if (vc4_encoder->type == type)
+                       return encoder;
+       }
+
+       return NULL;
+}
+
 int vc4_crtc_disable_at_boot(struct drm_crtc *crtc)
 {
        struct drm_device *drm = crtc->dev;
        struct vc4_crtc *vc4_crtc = to_vc4_crtc(crtc);
+       enum vc4_encoder_type encoder_type;
+       const struct vc4_pv_data *pv_data;
+       struct drm_encoder *encoder;
+       unsigned encoder_sel;
        int channel;
 
        if (!(of_device_is_compatible(vc4_crtc->pdev->dev.of_node,
        if (channel < 0)
                return 0;
 
-       return vc4_crtc_disable(crtc, NULL, channel);
+       encoder_sel = VC4_GET_FIELD(CRTC_READ(PV_CONTROL), PV_CONTROL_CLK_SELECT);
+       if (WARN_ON(encoder_sel != 0))
+               return 0;
+
+       pv_data = vc4_crtc_to_vc4_pv_data(vc4_crtc);
+       encoder_type = pv_data->encoder_types[encoder_sel];
+       encoder = vc4_crtc_get_encoder_by_type(crtc, encoder_type);
+       if (WARN_ON(!encoder))
+               return 0;
+
+       return vc4_crtc_disable(crtc, encoder, NULL, channel);
 }
 
 static void vc4_crtc_atomic_disable(struct drm_crtc *crtc,
        struct drm_crtc_state *old_state = drm_atomic_get_old_crtc_state(state,
                                                                         crtc);
        struct vc4_crtc_state *old_vc4_state = to_vc4_crtc_state(old_state);
+       struct drm_encoder *encoder = vc4_get_crtc_encoder(crtc, state,
+                                                          drm_atomic_get_old_connector_state);
        struct drm_device *dev = crtc->dev;
 
        require_hvs_enabled(dev);
        /* Disable vblank irq handling before crtc is disabled. */
        drm_crtc_vblank_off(crtc);
 
-       vc4_crtc_disable(crtc, state, old_vc4_state->assigned_channel);
+       vc4_crtc_disable(crtc, encoder, state, old_vc4_state->assigned_channel);
 
        /*
         * Make sure we issue a vblank event after disabling the CRTC if