#include "vmwgfx_drv.h"
 #include "vmwgfx_devcaps.h"
 #include <drm/vmwgfx_drm.h>
+#include <drm/drm_drv.h>
 #include "vmwgfx_kms.h"
 
 int vmw_getparam_ioctl(struct drm_device *dev, void *data,
        struct drm_vmw_rect __user *clips_ptr;
        struct drm_vmw_rect *clips = NULL;
        struct drm_framebuffer *fb;
+       struct drm_modeset_acquire_ctx ctx;
        struct vmw_framebuffer *vfb;
        struct vmw_resource *res;
        uint32_t num_clips;
                goto out_no_copy;
        }
 
-       drm_modeset_lock_all(dev);
+       DRM_MODESET_LOCK_ALL_BEGIN(dev, ctx, 0, ret);
 
        fb = drm_framebuffer_lookup(dev, file_priv, arg->fb_id);
        if (!fb) {
 out_no_surface:
        drm_framebuffer_put(fb);
 out_no_fb:
-       drm_modeset_unlock_all(dev);
+       DRM_MODESET_LOCK_ALL_END(dev, ctx, ret);
 out_no_copy:
        kfree(clips);
 out_clips:
        struct drm_vmw_rect __user *clips_ptr;
        struct drm_vmw_rect *clips = NULL;
        struct drm_framebuffer *fb;
+       struct drm_modeset_acquire_ctx ctx;
        struct vmw_framebuffer *vfb;
        uint32_t num_clips;
        int ret;
                goto out_no_copy;
        }
 
-       drm_modeset_lock_all(dev);
+       DRM_MODESET_LOCK_ALL_BEGIN(dev, ctx, 0, ret);
 
        fb = drm_framebuffer_lookup(dev, file_priv, arg->fb_id);
        if (!fb) {
 out_no_ttm_lock:
        drm_framebuffer_put(fb);
 out_no_fb:
-       drm_modeset_unlock_all(dev);
+       DRM_MODESET_LOCK_ALL_END(dev, ctx, ret);
 out_no_copy:
        kfree(clips);
 out_clips:
 
 #include <drm/drm_rect.h>
 #include <drm/drm_sysfs.h>
 #include <drm/drm_vblank.h>
+#include <drm/drm_drv.h>
 
 #include "vmwgfx_kms.h"
 
        struct drm_device *dev = &dev_priv->drm;
        struct vmw_display_unit *du;
        struct drm_crtc *crtc;
+       struct drm_modeset_acquire_ctx ctx;
+       int ret;
 
-       drm_modeset_lock_all(dev);
+       DRM_MODESET_LOCK_ALL_BEGIN(dev, ctx, 0, ret);
        drm_for_each_crtc(crtc, dev) {
                du = vmw_crtc_to_du(crtc);
 
                du->hotspot_x = 0;
                du->hotspot_y = 0;
        }
-       drm_modeset_unlock_all(dev);
+       DRM_MODESET_LOCK_ALL_END(dev, ctx, ret);
 }
 
 void vmw_kms_cursor_post_execbuf(struct vmw_private *dev_priv)
        struct vmw_framebuffer_bo *vfbd =
                vmw_framebuffer_to_vfbd(framebuffer);
        struct drm_clip_rect norect;
+       struct drm_modeset_acquire_ctx ctx;
        int ret, increment = 1;
 
-       drm_modeset_lock_all(&dev_priv->drm);
+       DRM_MODESET_LOCK_ALL_BEGIN((&dev_priv->drm), ctx, 0, ret);
 
        if (!num_clips) {
                num_clips = 1;
 
        vmw_cmd_flush(dev_priv, false);
 
-       drm_modeset_unlock_all(&dev_priv->drm);
+       DRM_MODESET_LOCK_ALL_END((&dev_priv->drm), ctx, ret);
 
        return ret;
 }