drm/amdgpu: add reserved vmid field in vm struct v2

v2: rename dedicated_vmid to reserved_vmid

Signed-off-by: Chunming Zhou <David1.Zhou@amd.com>
Reviewed-by: Junwei Zhang <Jerry.Zhang@amd.com>
Reviewed-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

2 files changed, 18 insertions, 1 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
index e4e2bacdb230..6eaeed08237c 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c
@@ -2147,10 +2147,12 @@ int amdgpu_vm_init(struct amdgpu_device *adev, struct amdgpu_vm *vm)
         unsigned ring_instance;
         struct amdgpu_ring *ring;
         struct amd_sched_rq *rq;
-        int r;
+        int r, i;
 
         vm->va = RB_ROOT;
         vm->client_id = atomic64_inc_return(&adev->vm_manager.client_counter);
+        for (i = 0; i < AMDGPU_MAX_VMHUBS; i++)
+                vm->reserved_vmid[i] = NULL;
         spin_lock_init(&vm->status_lock);
         INIT_LIST_HEAD(&vm->invalidated);
         INIT_LIST_HEAD(&vm->cleared);
@@ -2235,6 +2237,7 @@ void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
 {
         struct amdgpu_bo_va_mapping *mapping, *tmp;
         bool prt_fini_needed = !!adev->gart.gart_funcs->set_prt;
+        int i;
 
         amd_sched_entity_fini(vm->entity.sched, &vm->entity);
 
@@ -2258,6 +2261,18 @@ void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm)
 
         amdgpu_vm_free_levels(&vm->root);
         dma_fence_put(vm->last_dir_update);
+        for (i = 0; i < AMDGPU_MAX_VMHUBS; i++) {
+                struct amdgpu_vm_id_manager *id_mgr =
+                        &adev->vm_manager.id_mgr[i];
+
+                mutex_lock(&id_mgr->lock);
+                if (vm->reserved_vmid[i]) {
+                        list_add(&vm->reserved_vmid[i]->list,
+                                 &id_mgr->ids_lru);
+                        vm->reserved_vmid[i] = NULL;
+                }
+                mutex_unlock(&id_mgr->lock);
+        }
 }
 
 /**
diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.h b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.h
index b10ce2d08685..24ccf677f5a5 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.h
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.h
@@ -123,6 +123,8 @@ struct amdgpu_vm {
 
         /* client id */
         u64                     client_id;
+        /* dedicated to vm */
+        struct amdgpu_vm_id     *reserved_vmid[AMDGPU_MAX_VMHUBS];
         /* each VM will map on CSA */
         struct amdgpu_bo_va *csa_bo_va;
 };