void (*cancel_pending_vblank)(struct drm_crtc *crtc, struct drm_file *file_priv);
 };
 
-struct rockchip_atomic_commit {
-       struct work_struct      work;
-       struct drm_atomic_state *state;
-       struct drm_device *dev;
-       struct mutex lock;
-};
-
 struct rockchip_crtc_state {
        struct drm_crtc_state base;
        int output_type;
        struct drm_fb_helper fbdev_helper;
        struct drm_gem_object *fbdev_bo;
        const struct rockchip_crtc_funcs *crtc_funcs[ROCKCHIP_MAX_CRTC];
-
-       struct rockchip_atomic_commit commit;
 };
 
-void rockchip_drm_atomic_work(struct work_struct *work);
 int rockchip_register_crtc_funcs(struct drm_crtc *crtc,
                                 const struct rockchip_crtc_funcs *crtc_funcs);
 void rockchip_unregister_crtc_funcs(struct drm_crtc *crtc);
 
 }
 
 static void
-rockchip_atomic_commit_complete(struct rockchip_atomic_commit *commit)
+rockchip_atomic_commit_tail(struct drm_atomic_state *state)
 {
-       struct drm_atomic_state *state = commit->state;
-       struct drm_device *dev = commit->dev;
+       struct drm_device *dev = state->dev;
 
-       /*
-        * TODO: do fence wait here.
-        */
-
-       /*
-        * Rockchip crtc support runtime PM, can't update display planes
-        * when crtc is disabled.
-        *
-        * drm_atomic_helper_commit comments detail that:
-        *     For drivers supporting runtime PM the recommended sequence is
-        *
-        *     drm_atomic_helper_commit_modeset_disables(dev, state);
-        *
-        *     drm_atomic_helper_commit_modeset_enables(dev, state);
-        *
-        *     drm_atomic_helper_commit_planes(dev, state, true);
-        *
-        * See the kerneldoc entries for these three functions for more details.
-        */
        drm_atomic_helper_commit_modeset_disables(dev, state);
 
        drm_atomic_helper_commit_modeset_enables(dev, state);
 
        drm_atomic_helper_commit_planes(dev, state, true);
 
+       drm_atomic_helper_commit_hw_done(state);
+
        rockchip_atomic_wait_for_complete(dev, state);
 
        drm_atomic_helper_cleanup_planes(dev, state);
-
-       drm_atomic_state_free(state);
-}
-
-void rockchip_drm_atomic_work(struct work_struct *work)
-{
-       struct rockchip_atomic_commit *commit = container_of(work,
-                                       struct rockchip_atomic_commit, work);
-
-       rockchip_atomic_commit_complete(commit);
 }
 
-int rockchip_drm_atomic_commit(struct drm_device *dev,
-                              struct drm_atomic_state *state,
-                              bool nonblock)
-{
-       struct rockchip_drm_private *private = dev->dev_private;
-       struct rockchip_atomic_commit *commit = &private->commit;
-       int ret;
-
-       ret = drm_atomic_helper_prepare_planes(dev, state);
-       if (ret)
-               return ret;
-
-       /* serialize outstanding nonblocking commits */
-       mutex_lock(&commit->lock);
-       flush_work(&commit->work);
-
-       drm_atomic_helper_swap_state(state, true);
-
-       commit->dev = dev;
-       commit->state = state;
-
-       if (nonblock)
-               schedule_work(&commit->work);
-       else
-               rockchip_atomic_commit_complete(commit);
-
-       mutex_unlock(&commit->lock);
-
-       return 0;
-}
+struct drm_mode_config_helper_funcs rockchip_mode_config_helpers = {
+       .atomic_commit_tail = rockchip_atomic_commit_tail,
+};
 
 static const struct drm_mode_config_funcs rockchip_drm_mode_config_funcs = {
        .fb_create = rockchip_user_fb_create,
        .output_poll_changed = rockchip_drm_output_poll_changed,
        .atomic_check = drm_atomic_helper_check,
-       .atomic_commit = rockchip_drm_atomic_commit,
+       .atomic_commit = drm_atomic_helper_commit,
 };
 
 struct drm_framebuffer *
        dev->mode_config.max_height = 4096;
 
        dev->mode_config.funcs = &rockchip_drm_mode_config_funcs;
+       dev->mode_config.helper_private = &rockchip_mode_config_helpers;
 }
 
        struct vop *vop = to_vop(crtc);
        int i;
 
+       WARN_ON(vop->event);
+
        /*
         * We need to make sure that all windows are disabled before we
         * disable that crtc. Otherwise we might try to scan from a destroyed
        clk_disable(vop->aclk);
        clk_disable(vop->hclk);
        pm_runtime_put(vop->dev);
+
+       if (crtc->state->event && !crtc->state->active) {
+               spin_lock_irq(&crtc->dev->event_lock);
+               drm_crtc_send_vblank_event(crtc, crtc->state->event);
+               spin_unlock_irq(&crtc->dev->event_lock);
+
+               crtc->state->event = NULL;
+       }
 }
 
 static void vop_plane_destroy(struct drm_plane *plane)
        u16 vact_end = vact_st + vdisplay;
        uint32_t val;
 
+       WARN_ON(vop->event);
+
        vop_enable(crtc);
        /*
         * If dclk rate is zero, mean that scanout is stop,
 {
        struct vop *vop = to_vop(crtc);
 
+       spin_lock_irq(&crtc->dev->event_lock);
        if (crtc->state->event) {
                WARN_ON(drm_crtc_vblank_get(crtc) != 0);
+               WARN_ON(vop->event);
 
                vop->event = crtc->state->event;
                crtc->state->event = NULL;
        }
+       spin_unlock_irq(&crtc->dev->event_lock);
 }
 
 static const struct drm_crtc_helper_funcs vop_crtc_helper_funcs = {
                        return;
        }
 
+       spin_lock_irqsave(&drm->event_lock, flags);
        if (vop->event) {
-               spin_lock_irqsave(&drm->event_lock, flags);
 
                drm_crtc_send_vblank_event(crtc, vop->event);
                drm_crtc_vblank_put(crtc);
                vop->event = NULL;
 
-               spin_unlock_irqrestore(&drm->event_lock, flags);
        }
+       spin_unlock_irqrestore(&drm->event_lock, flags);
+
        if (!completion_done(&vop->wait_update_complete))
                complete(&vop->wait_update_complete);
 }