1 files changed, 3 insertions, 3 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
index 1c3dd6e0ed33..bd6296a6dab1 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
@@ -1285,11 +1285,11 @@ void amdgpu_vm_get_entry(struct amdgpu_pte_update_params *p, uint64_t addr,
         *parent = NULL;
         *entry = &p->vm->root;
         while ((*entry)->entries) {
-                unsigned idx = addr >> amdgpu_vm_level_shift(p->adev, level++);
+                unsigned shift = amdgpu_vm_level_shift(p->adev, level++);
 
-                idx %= amdgpu_bo_size((*entry)->base.bo) / 8;
                 *parent = *entry;
-                *entry = &(*entry)->entries[idx];
+                *entry = &(*entry)->entries[addr >> shift];
+                addr &= (1ULL << shift) - 1;
         }
 
         if (level != p->adev->vm_manager.num_level)