.mode_set_nofb = fsl_dcu_drm_crtc_mode_set_nofb,
 };
 
+static int fsl_dcu_drm_crtc_enable_vblank(struct drm_crtc *crtc)
+{
+       struct drm_device *dev = crtc->dev;
+       struct fsl_dcu_drm_device *fsl_dev = dev->dev_private;
+       unsigned int value;
+
+       regmap_read(fsl_dev->regmap, DCU_INT_MASK, &value);
+       value &= ~DCU_INT_MASK_VBLANK;
+       regmap_write(fsl_dev->regmap, DCU_INT_MASK, value);
+
+       return 0;
+}
+
+static void fsl_dcu_drm_crtc_disable_vblank(struct drm_crtc *crtc)
+{
+       struct drm_device *dev = crtc->dev;
+       struct fsl_dcu_drm_device *fsl_dev = dev->dev_private;
+       unsigned int value;
+
+       regmap_read(fsl_dev->regmap, DCU_INT_MASK, &value);
+       value |= DCU_INT_MASK_VBLANK;
+       regmap_write(fsl_dev->regmap, DCU_INT_MASK, value);
+}
+
 static const struct drm_crtc_funcs fsl_dcu_drm_crtc_funcs = {
        .atomic_duplicate_state = drm_atomic_helper_crtc_duplicate_state,
        .atomic_destroy_state = drm_atomic_helper_crtc_destroy_state,
        .page_flip = drm_atomic_helper_page_flip,
        .reset = drm_atomic_helper_crtc_reset,
        .set_config = drm_atomic_helper_set_config,
+       .enable_vblank = fsl_dcu_drm_crtc_enable_vblank,
+       .disable_vblank = fsl_dcu_drm_crtc_disable_vblank,
 };
 
 int fsl_dcu_drm_crtc_create(struct fsl_dcu_drm_device *fsl_dev)
 
        return IRQ_HANDLED;
 }
 
-static int fsl_dcu_drm_enable_vblank(struct drm_device *dev, unsigned int pipe)
-{
-       struct fsl_dcu_drm_device *fsl_dev = dev->dev_private;
-       unsigned int value;
-
-       regmap_read(fsl_dev->regmap, DCU_INT_MASK, &value);
-       value &= ~DCU_INT_MASK_VBLANK;
-       regmap_write(fsl_dev->regmap, DCU_INT_MASK, value);
-
-       return 0;
-}
-
-static void fsl_dcu_drm_disable_vblank(struct drm_device *dev,
-                                      unsigned int pipe)
-{
-       struct fsl_dcu_drm_device *fsl_dev = dev->dev_private;
-       unsigned int value;
-
-       regmap_read(fsl_dev->regmap, DCU_INT_MASK, &value);
-       value |= DCU_INT_MASK_VBLANK;
-       regmap_write(fsl_dev->regmap, DCU_INT_MASK, value);
-}
-
 static void fsl_dcu_drm_lastclose(struct drm_device *dev)
 {
        struct fsl_dcu_drm_device *fsl_dev = dev->dev_private;
        .load                   = fsl_dcu_load,
        .unload                 = fsl_dcu_unload,
        .irq_handler            = fsl_dcu_drm_irq,
-       .enable_vblank          = fsl_dcu_drm_enable_vblank,
-       .disable_vblank         = fsl_dcu_drm_disable_vblank,
        .gem_free_object_unlocked = drm_gem_cma_free_object,
        .gem_vm_ops             = &drm_gem_cma_vm_ops,
        .prime_handle_to_fd     = drm_gem_prime_handle_to_fd,