1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_gem.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_gem.c
index 6daf004955b0..0240f108f90e 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_gem.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_gem.c
@@ -631,8 +631,12 @@ int amdgpu_gem_va_ioctl(struct drm_device *dev, void *data,
 
         switch (args->operation) {
         case AMDGPU_VA_OP_MAP:
-                va_flags = amdgpu_vm_get_pte_flags(adev, args->flags);
+                r = amdgpu_vm_alloc_pts(adev, bo_va->vm, args->va_address,
+                                        args->map_size);
+                if (r)
+                        goto error_backoff;
 
+                va_flags = amdgpu_vm_get_pte_flags(adev, args->flags);
                 r = amdgpu_vm_bo_map(adev, bo_va, args->va_address,
                                      args->offset_in_bo, args->map_size,
                                      va_flags);