struct drm_crtc *tmp;
        int ret;
 
+       WARN_ON(drm_drv_uses_atomic_modeset(crtc->dev));
+
        /*
         * NOTE: ->set_config can also disable other crtcs (if we steal all
         * connectors from it), hence we need to refcount the fbs across all
        if (ret == 0) {
                struct drm_plane *plane = crtc->primary;
 
-               if (!plane->state) {
-                       plane->crtc = fb ? crtc : NULL;
-                       plane->fb = fb;
-               }
+               plane->crtc = fb ? crtc : NULL;
+               plane->fb = fb;
        }
 
        drm_for_each_crtc(tmp, crtc->dev) {
 
        return ret;
 }
+
 /**
  * drm_mode_set_config_internal - helper to call &drm_mode_config_funcs.set_config
  * @set: modeset config to set
        set.connectors = connector_set;
        set.num_connectors = crtc_req->count_connectors;
        set.fb = fb;
-       ret = __drm_mode_set_config_internal(&set, &ctx);
+
+       if (drm_drv_uses_atomic_modeset(dev))
+               ret = crtc->funcs->set_config(&set, &ctx);
+       else
+               ret = __drm_mode_set_config_internal(&set, &ctx);
 
 out:
        if (fb)