1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_display.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_display.c
index b0832da2ef7e..0b5f3accb1e4 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_display.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_display.c
@@ -240,7 +240,7 @@ int amdgpu_crtc_page_flip(struct drm_crtc *crtc,
 
         work->base = base;
 
-        r = drm_vblank_get(crtc->dev, amdgpu_crtc->crtc_id);
+        r = drm_crtc_vblank_get(crtc);
         if (r) {
                 DRM_ERROR("failed to get vblank before flip\n");
                 goto pflip_cleanup;
@@ -268,7 +268,7 @@ int amdgpu_crtc_page_flip(struct drm_crtc *crtc,
         return 0;
 
 vblank_cleanup:
-        drm_vblank_put(crtc->dev, amdgpu_crtc->crtc_id);
+        drm_crtc_vblank_put(&amdgpu_crtc->base);
 
 pflip_cleanup:
         if (unlikely(amdgpu_bo_reserve(new_rbo, false) != 0)) {