From 4388fc2ab0e83fd6247c48c7e97064cc75511848 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Christian=20K=C3=B6nig?= Date: Mon, 13 Mar 2017 10:13:36 +0100 Subject: [PATCH] drm/amdgpu: make set_prt callback optional and fix error handling MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit PRT support is completely implemented now and we left it turned on accidentially in the error path. Signed-off-by: Christian König Reviewed-by: Junwei Zhang Signed-off-by: Alex Deucher --- drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c | 29 +++++++++++++------------- 1 file changed, 15 insertions(+), 14 deletions(-) diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c index a45de6e6a0f7..f9f4e2055084 100644 --- a/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c +++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_vm.c @@ -1194,10 +1194,13 @@ static void amdgpu_vm_update_prt_state(struct amdgpu_device *adev) } /** - * amdgpu_vm_prt_put - add a PRT user + * amdgpu_vm_prt_get - add a PRT user */ static void amdgpu_vm_prt_get(struct amdgpu_device *adev) { + if (!adev->gart.gart_funcs->set_prt) + return; + if (atomic_inc_return(&adev->vm_manager.num_prt_users) == 1) amdgpu_vm_update_prt_state(adev); } @@ -1228,9 +1231,12 @@ static void amdgpu_vm_prt_cb(struct dma_fence *fence, struct dma_fence_cb *_cb) static void amdgpu_vm_add_prt_cb(struct amdgpu_device *adev, struct dma_fence *fence) { - struct amdgpu_prt_cb *cb = kmalloc(sizeof(struct amdgpu_prt_cb), - GFP_KERNEL); + struct amdgpu_prt_cb *cb; + if (!adev->gart.gart_funcs->set_prt) + return; + + cb = kmalloc(sizeof(struct amdgpu_prt_cb), GFP_KERNEL); if (!cb) { /* Last resort when we are OOM */ if (fence) @@ -1445,14 +1451,6 @@ int amdgpu_vm_bo_map(struct amdgpu_device *adev, size == 0 || size & AMDGPU_GPU_PAGE_MASK) return -EINVAL; - if (flags & AMDGPU_PTE_PRT) { - /* Check if we have PRT hardware support */ - if (!adev->gart.gart_funcs->set_prt) - return -EINVAL; - - amdgpu_vm_prt_get(adev); - } - /* make sure object fit at this offset */ eaddr = saddr + size - 1; if (saddr >= eaddr || @@ -1533,6 +1531,9 @@ int amdgpu_vm_bo_map(struct amdgpu_device *adev, vm->page_tables[pt_idx].addr = 0; } + if (flags & AMDGPU_PTE_PRT) + amdgpu_vm_prt_get(adev); + return 0; error_free: @@ -1753,7 +1754,7 @@ err: void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm) { struct amdgpu_bo_va_mapping *mapping, *tmp; - bool prt_fini_called = false; + bool prt_fini_needed = !!adev->gart.gart_funcs->set_prt; int i; amd_sched_entity_fini(vm->entity.sched, &vm->entity); @@ -1767,9 +1768,9 @@ void amdgpu_vm_fini(struct amdgpu_device *adev, struct amdgpu_vm *vm) kfree(mapping); } list_for_each_entry_safe(mapping, tmp, &vm->freed, list) { - if (mapping->flags & AMDGPU_PTE_PRT && !prt_fini_called) { + if (mapping->flags & AMDGPU_PTE_PRT && prt_fini_needed) { amdgpu_vm_prt_fini(adev, vm); - prt_fini_called = true; + prt_fini_needed = false; } list_del(&mapping->list); -- 2.39.2