1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
index 6904d794d60a..db0cbf8d219d 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
@@ -3234,8 +3234,10 @@ void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
         }
         rbtree_postorder_for_each_entry_safe(mapping, tmp,
                                              &vm->va.rb_root, rb) {
+                /* Don't remove the mapping here, we don't want to trigger a
+                 * rebalance and the tree is about to be destroyed anyway.
+                 */
                 list_del(&mapping->list);
-                amdgpu_vm_it_remove(mapping, &vm->va);
                 kfree(mapping);
         }
         list_for_each_entry_safe(mapping, tmp, &vm->freed, list) {