| // SPDX-License-Identifier: MIT | |
| /* | |
| * Copyright 2023 Advanced Micro Devices, Inc. | |
| * | |
| * Permission is hereby granted, free of charge, to any person obtaining a | |
| * copy of this software and associated documentation files (the "Software"), | |
| * to deal in the Software without restriction, including without limitation | |
| * the rights to use, copy, modify, merge, publish, distribute, sublicense, | |
| * and/or sell copies of the Software, and to permit persons to whom the | |
| * Software is furnished to do so, subject to the following conditions: | |
| * | |
| * The above copyright notice and this permission notice shall be included in | |
| * all copies or substantial portions of the Software. | |
| * | |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
| * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL | |
| * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR | |
| * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, | |
| * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR | |
| * OTHER DEALINGS IN THE SOFTWARE. | |
| * | |
| */ | |
| #include <drm/drm_auth.h> | |
| #include <drm/drm_exec.h> | |
| #include <linux/pm_runtime.h> | |
| #include <drm/drm_drv.h> | |
| #include "amdgpu.h" | |
| #include "amdgpu_reset.h" | |
| #include "amdgpu_vm.h" | |
| #include "amdgpu_userq.h" | |
| #include "amdgpu_hmm.h" | |
| #include "amdgpu_userq_fence.h" | |
| u32 amdgpu_userq_get_supported_ip_mask(struct amdgpu_device *adev) | |
| { | |
| int i; | |
| u32 userq_ip_mask = 0; | |
| for (i = 0; i < AMDGPU_HW_IP_NUM; i++) { | |
| if (adev->userq_funcs[i]) | |
| userq_ip_mask |= (1 << i); | |
| } | |
| return userq_ip_mask; | |
| } | |
| static bool amdgpu_userq_is_reset_type_supported(struct amdgpu_device *adev, | |
| enum amdgpu_ring_type ring_type, int reset_type) | |
| { | |
| if (ring_type < 0 || ring_type >= AMDGPU_RING_TYPE_MAX) | |
| return false; | |
| switch (ring_type) { | |
| case AMDGPU_RING_TYPE_GFX: | |
| if (adev->gfx.gfx_supported_reset & reset_type) | |
| return true; | |
| break; | |
| case AMDGPU_RING_TYPE_COMPUTE: | |
| if (adev->gfx.compute_supported_reset & reset_type) | |
| return true; | |
| break; | |
| case AMDGPU_RING_TYPE_SDMA: | |
| if (adev->sdma.supported_reset & reset_type) | |
| return true; | |
| break; | |
| case AMDGPU_RING_TYPE_VCN_DEC: | |
| case AMDGPU_RING_TYPE_VCN_ENC: | |
| if (adev->vcn.supported_reset & reset_type) | |
| return true; | |
| break; | |
| case AMDGPU_RING_TYPE_VCN_JPEG: | |
| if (adev->jpeg.supported_reset & reset_type) | |
| return true; | |
| break; | |
| default: | |
| break; | |
| } | |
| return false; | |
| } | |
| static void amdgpu_userq_mgr_reset_work(struct work_struct *work) | |
| { | |
| struct amdgpu_userq_mgr *uq_mgr = | |
| container_of(work, struct amdgpu_userq_mgr, | |
| reset_work); | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| const int queue_types[] = { | |
| AMDGPU_RING_TYPE_COMPUTE, | |
| AMDGPU_RING_TYPE_GFX, | |
| AMDGPU_RING_TYPE_SDMA | |
| }; | |
| const int num_queue_types = ARRAY_SIZE(queue_types); | |
| bool gpu_reset = false; | |
| int i, r; | |
| if (unlikely(adev->debug_disable_gpu_ring_reset)) { | |
| dev_err(adev->dev, "userq reset disabled by debug mask\n"); | |
| return; | |
| } | |
| /* | |
| * If GPU recovery feature is disabled system-wide, | |
| * skip all reset detection logic | |
| */ | |
| if (!amdgpu_gpu_recovery) | |
| return; | |
| /* | |
| * Iterate through all queue types to detect and reset problematic queues | |
| * Process each queue type in the defined order | |
| */ | |
| for (i = 0; i < num_queue_types; i++) { | |
| int ring_type = queue_types[i]; | |
| const struct amdgpu_userq_funcs *funcs = | |
| adev->userq_funcs[ring_type]; | |
| if (!amdgpu_userq_is_reset_type_supported(adev, ring_type, | |
| AMDGPU_RESET_TYPE_PER_QUEUE)) | |
| continue; | |
| if (atomic_read(&uq_mgr->userq_count[ring_type]) > 0 && | |
| funcs && funcs->detect_and_reset) { | |
| r = funcs->detect_and_reset(adev, ring_type); | |
| if (r) { | |
| gpu_reset = true; | |
| break; | |
| } | |
| } | |
| } | |
| if (gpu_reset) { | |
| struct amdgpu_reset_context reset_context; | |
| memset(&reset_context, 0, sizeof(reset_context)); | |
| reset_context.method = AMD_RESET_METHOD_NONE; | |
| reset_context.reset_req_dev = adev; | |
| reset_context.src = AMDGPU_RESET_SRC_USERQ; | |
| set_bit(AMDGPU_NEED_FULL_RESET, &reset_context.flags); | |
| /*set_bit(AMDGPU_SKIP_COREDUMP, &reset_context.flags);*/ | |
| amdgpu_device_gpu_recover(adev, NULL, &reset_context); | |
| } | |
| } | |
| static void amdgpu_userq_hang_detect_work(struct work_struct *work) | |
| { | |
| struct amdgpu_usermode_queue *queue = | |
| container_of(work, struct amdgpu_usermode_queue, | |
| hang_detect_work.work); | |
| /* | |
| * Don't schedule the work here! Scheduling or queue work from one reset | |
| * handler to another is illegal if you don't take extra precautions! | |
| */ | |
| amdgpu_userq_mgr_reset_work(&queue->userq_mgr->reset_work); | |
| } | |
| /* | |
| * Start hang detection for a user queue fence. A delayed work will be scheduled | |
| * to reset the queues when the fence doesn't signal in time. | |
| */ | |
| void amdgpu_userq_start_hang_detect_work(struct amdgpu_usermode_queue *queue) | |
| { | |
| struct amdgpu_device *adev; | |
| unsigned long timeout_ms; | |
| adev = queue->userq_mgr->adev; | |
| /* Determine timeout based on queue type */ | |
| switch (queue->queue_type) { | |
| case AMDGPU_RING_TYPE_GFX: | |
| timeout_ms = adev->gfx_timeout; | |
| break; | |
| case AMDGPU_RING_TYPE_COMPUTE: | |
| timeout_ms = adev->compute_timeout; | |
| break; | |
| case AMDGPU_RING_TYPE_SDMA: | |
| timeout_ms = adev->sdma_timeout; | |
| break; | |
| default: | |
| timeout_ms = adev->gfx_timeout; | |
| break; | |
| } | |
| queue_delayed_work(adev->reset_domain->wq, &queue->hang_detect_work, | |
| msecs_to_jiffies(timeout_ms)); | |
| } | |
| void amdgpu_userq_process_fence_irq(struct amdgpu_device *adev, u32 doorbell) | |
| { | |
| struct xarray *xa = &adev->userq_doorbell_xa; | |
| struct amdgpu_usermode_queue *queue; | |
| unsigned long flags; | |
| int r; | |
| xa_lock_irqsave(xa, flags); | |
| queue = xa_load(xa, doorbell); | |
| if (queue) { | |
| r = amdgpu_userq_fence_driver_process(queue->fence_drv); | |
| /* | |
| * We are in interrupt context here, this *can't* wait for | |
| * reset work to finish. | |
| */ | |
| if (r >= 0) | |
| cancel_delayed_work(&queue->hang_detect_work); | |
| /* Restart the timer when there are still fences pending */ | |
| if (r == 1) | |
| amdgpu_userq_start_hang_detect_work(queue); | |
| } | |
| xa_unlock_irqrestore(xa, flags); | |
| } | |
| int amdgpu_userq_input_va_validate(struct amdgpu_device *adev, | |
| struct amdgpu_usermode_queue *queue, | |
| u64 addr, u64 expected_size, | |
| u64 *va_out) | |
| { | |
| struct amdgpu_bo_va_mapping *va_map; | |
| struct amdgpu_vm *vm = queue->vm; | |
| u64 user_addr; | |
| u64 size; | |
| /* Caller must hold vm->root.bo reservation */ | |
| dma_resv_assert_held(queue->vm->root.bo->tbo.base.resv); | |
| user_addr = (addr & AMDGPU_GMC_HOLE_MASK) >> AMDGPU_GPU_PAGE_SHIFT; | |
| size = expected_size >> AMDGPU_GPU_PAGE_SHIFT; | |
| va_map = amdgpu_vm_bo_lookup_mapping(vm, user_addr); | |
| if (!va_map) | |
| return -EINVAL; | |
| /* Only validate the userq whether resident in the VM mapping range */ | |
| if (user_addr >= va_map->start && | |
| va_map->last - user_addr + 1 >= size) { | |
| va_map->bo_va->userq_va_mapped = true; | |
| *va_out = user_addr; | |
| return 0; | |
| } | |
| return -EINVAL; | |
| } | |
| static bool amdgpu_userq_buffer_va_mapped(struct amdgpu_vm *vm, u64 addr) | |
| { | |
| struct amdgpu_bo_va_mapping *mapping; | |
| bool r; | |
| dma_resv_assert_held(vm->root.bo->tbo.base.resv); | |
| mapping = amdgpu_vm_bo_lookup_mapping(vm, addr); | |
| if (!IS_ERR_OR_NULL(mapping) && mapping->bo_va->userq_va_mapped) | |
| r = true; | |
| else | |
| r = false; | |
| return r; | |
| } | |
| static bool amdgpu_userq_buffer_vas_mapped(struct amdgpu_usermode_queue *queue) | |
| { | |
| int i, r = 0; | |
| for (i = 0; i < ARRAY_SIZE(queue->userq_vas.va_array); i++) { | |
| if (!queue->userq_vas.va_array[i]) | |
| continue; | |
| r += amdgpu_userq_buffer_va_mapped(queue->vm, | |
| queue->userq_vas.va_array[i]); | |
| dev_dbg(queue->userq_mgr->adev->dev, | |
| "validate the userq mapping:%p va:%llx r:%d\n", | |
| queue, queue->userq_vas.va_array[i], r); | |
| } | |
| if (r != 0) | |
| return true; | |
| return false; | |
| } | |
| static int amdgpu_userq_preempt_helper(struct amdgpu_usermode_queue *queue) | |
| { | |
| struct amdgpu_userq_mgr *uq_mgr = queue->userq_mgr; | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| const struct amdgpu_userq_funcs *userq_funcs = | |
| adev->userq_funcs[queue->queue_type]; | |
| int r; | |
| if (queue->state == AMDGPU_USERQ_STATE_MAPPED) { | |
| r = userq_funcs->preempt(queue); | |
| if (r) { | |
| queue->state = AMDGPU_USERQ_STATE_HUNG; | |
| return r; | |
| } else { | |
| queue->state = AMDGPU_USERQ_STATE_PREEMPTED; | |
| } | |
| } | |
| return 0; | |
| } | |
| static int amdgpu_userq_restore_helper(struct amdgpu_usermode_queue *queue) | |
| { | |
| struct amdgpu_userq_mgr *uq_mgr = queue->userq_mgr; | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| const struct amdgpu_userq_funcs *userq_funcs = | |
| adev->userq_funcs[queue->queue_type]; | |
| int r = 0; | |
| if (queue->state == AMDGPU_USERQ_STATE_PREEMPTED) { | |
| r = userq_funcs->restore(queue); | |
| if (r) { | |
| queue->state = AMDGPU_USERQ_STATE_HUNG; | |
| } else { | |
| queue->state = AMDGPU_USERQ_STATE_MAPPED; | |
| } | |
| } | |
| return r; | |
| } | |
| static int amdgpu_userq_unmap_helper(struct amdgpu_usermode_queue *queue) | |
| { | |
| struct amdgpu_userq_mgr *uq_mgr = queue->userq_mgr; | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| const struct amdgpu_userq_funcs *userq_funcs = | |
| adev->userq_funcs[queue->queue_type]; | |
| int r; | |
| if ((queue->state == AMDGPU_USERQ_STATE_MAPPED) || | |
| (queue->state == AMDGPU_USERQ_STATE_PREEMPTED)) { | |
| r = userq_funcs->unmap(queue); | |
| if (r) { | |
| queue->state = AMDGPU_USERQ_STATE_HUNG; | |
| return r; | |
| } else { | |
| queue->state = AMDGPU_USERQ_STATE_UNMAPPED; | |
| } | |
| } | |
| return 0; | |
| } | |
| static int amdgpu_userq_map_helper(struct amdgpu_usermode_queue *queue) | |
| { | |
| struct amdgpu_userq_mgr *uq_mgr = queue->userq_mgr; | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| const struct amdgpu_userq_funcs *userq_funcs = | |
| adev->userq_funcs[queue->queue_type]; | |
| int r; | |
| if (queue->state == AMDGPU_USERQ_STATE_UNMAPPED) { | |
| r = userq_funcs->map(queue); | |
| if (r) { | |
| queue->state = AMDGPU_USERQ_STATE_HUNG; | |
| return r; | |
| } else { | |
| queue->state = AMDGPU_USERQ_STATE_MAPPED; | |
| } | |
| } | |
| return 0; | |
| } | |
| static void amdgpu_userq_wait_for_last_fence(struct amdgpu_usermode_queue *queue) | |
| { | |
| struct dma_fence *f = queue->last_fence; | |
| if (!f) | |
| return; | |
| dma_fence_wait(f, false); | |
| } | |
| static void amdgpu_userq_cleanup(struct amdgpu_usermode_queue *queue) | |
| { | |
| struct amdgpu_userq_mgr *uq_mgr = queue->userq_mgr; | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| /* Wait for mode-1 reset to complete */ | |
| down_read(&adev->reset_domain->sem); | |
| /* Use interrupt-safe locking since IRQ handlers may access these XArrays */ | |
| xa_erase_irq(&adev->userq_doorbell_xa, queue->doorbell_index); | |
| amdgpu_userq_fence_driver_free(queue); | |
| queue->fence_drv = NULL; | |
| up_read(&adev->reset_domain->sem); | |
| } | |
| /** | |
| * amdgpu_userq_ensure_ev_fence - ensure a valid, unsignaled eviction fence exists | |
| * @uq_mgr: the usermode queue manager for this process | |
| * @evf_mgr: the eviction fence manager to check and rearm | |
| * | |
| * Ensures that a valid and not yet signaled eviction fence is attached to the | |
| * usermode queue before any queue operations proceed. If it is signalled, then | |
| * rearm a new eviction fence. | |
| */ | |
| void | |
| amdgpu_userq_ensure_ev_fence(struct amdgpu_userq_mgr *uq_mgr, | |
| struct amdgpu_eviction_fence_mgr *evf_mgr) | |
| { | |
| struct dma_fence *ev_fence; | |
| retry: | |
| /* Flush any pending resume work to create ev_fence */ | |
| flush_delayed_work(&uq_mgr->resume_work); | |
| mutex_lock(&uq_mgr->userq_mutex); | |
| ev_fence = amdgpu_evf_mgr_get_fence(evf_mgr); | |
| if (dma_fence_is_signaled(ev_fence)) { | |
| dma_fence_put(ev_fence); | |
| mutex_unlock(&uq_mgr->userq_mutex); | |
| /* | |
| * Looks like there was no pending resume work, | |
| * add one now to create a valid eviction fence | |
| */ | |
| schedule_delayed_work(&uq_mgr->resume_work, 0); | |
| goto retry; | |
| } | |
| dma_fence_put(ev_fence); | |
| } | |
| static int | |
| amdgpu_userq_get_doorbell_index(struct amdgpu_userq_mgr *uq_mgr, | |
| struct amdgpu_db_info *db_info, | |
| struct drm_file *filp, | |
| u64 *index) | |
| { | |
| u64 doorbell_index; | |
| struct drm_gem_object *gobj; | |
| struct amdgpu_userq_obj *db_obj = db_info->db_obj; | |
| int r, db_size; | |
| gobj = drm_gem_object_lookup(filp, db_info->doorbell_handle); | |
| if (gobj == NULL) { | |
| drm_file_err(uq_mgr->file, "Can't find GEM object for doorbell\n"); | |
| return -EINVAL; | |
| } | |
| db_obj->obj = amdgpu_bo_ref(gem_to_amdgpu_bo(gobj)); | |
| drm_gem_object_put(gobj); | |
| r = amdgpu_bo_reserve(db_obj->obj, true); | |
| if (r) { | |
| drm_file_err(uq_mgr->file, "[Usermode queues] Failed to pin doorbell object\n"); | |
| goto unref_bo; | |
| } | |
| /* Pin the BO before generating the index, unpin in queue destroy */ | |
| r = amdgpu_bo_pin(db_obj->obj, AMDGPU_GEM_DOMAIN_DOORBELL); | |
| if (r) { | |
| drm_file_err(uq_mgr->file, "[Usermode queues] Failed to pin doorbell object\n"); | |
| goto unresv_bo; | |
| } | |
| switch (db_info->queue_type) { | |
| case AMDGPU_HW_IP_GFX: | |
| case AMDGPU_HW_IP_COMPUTE: | |
| case AMDGPU_HW_IP_DMA: | |
| db_size = sizeof(u64); | |
| break; | |
| default: | |
| drm_file_err(uq_mgr->file, "[Usermode queues] IP %d not support\n", | |
| db_info->queue_type); | |
| r = -EINVAL; | |
| goto unpin_bo; | |
| } | |
| /* Validate doorbell_offset is within the doorbell BO */ | |
| if ((u64)db_info->doorbell_offset * db_size + db_size > | |
| amdgpu_bo_size(db_obj->obj)) { | |
| r = -EINVAL; | |
| goto unpin_bo; | |
| } | |
| doorbell_index = amdgpu_doorbell_index_on_bar(uq_mgr->adev, db_obj->obj, | |
| db_info->doorbell_offset, db_size); | |
| drm_dbg_driver(adev_to_drm(uq_mgr->adev), | |
| "[Usermode queues] doorbell index=%lld\n", doorbell_index); | |
| amdgpu_bo_unreserve(db_obj->obj); | |
| *index = doorbell_index; | |
| return 0; | |
| unpin_bo: | |
| amdgpu_bo_unpin(db_obj->obj); | |
| unresv_bo: | |
| amdgpu_bo_unreserve(db_obj->obj); | |
| unref_bo: | |
| amdgpu_bo_unref(&db_obj->obj); | |
| return r; | |
| } | |
| static int | |
| amdgpu_userq_destroy(struct amdgpu_userq_mgr *uq_mgr, struct amdgpu_usermode_queue *queue) | |
| { | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| const struct amdgpu_userq_funcs *uq_funcs = adev->userq_funcs[queue->queue_type]; | |
| int r = 0; | |
| cancel_delayed_work_sync(&uq_mgr->resume_work); | |
| /* Cancel any pending hang detection work and cleanup */ | |
| cancel_delayed_work_sync(&queue->hang_detect_work); | |
| mutex_lock(&uq_mgr->userq_mutex); | |
| amdgpu_userq_wait_for_last_fence(queue); | |
| #if defined(CONFIG_DEBUG_FS) | |
| debugfs_remove_recursive(queue->debugfs_queue); | |
| #endif | |
| r = amdgpu_userq_unmap_helper(queue); | |
| atomic_dec(&uq_mgr->userq_count[queue->queue_type]); | |
| amdgpu_userq_cleanup(queue); | |
| mutex_unlock(&uq_mgr->userq_mutex); | |
| /* | |
| * A failed unmap means MES could not remove the hung queue and is now | |
| * unresponsive. Recover the GPU here so the wedged MES does not fail | |
| * the next, unrelated queue submission and trigger a reset attributed | |
| * to an innocent workload. | |
| */ | |
| if (r) | |
| queue_work(adev->reset_domain->wq, &uq_mgr->reset_work); | |
| cancel_delayed_work_sync(&queue->hang_detect_work); | |
| uq_funcs->mqd_destroy(queue); | |
| queue->userq_mgr = NULL; | |
| amdgpu_bo_reserve(queue->db_obj.obj, true); | |
| amdgpu_bo_unpin(queue->db_obj.obj); | |
| amdgpu_bo_unreserve(queue->db_obj.obj); | |
| amdgpu_bo_unref(&queue->db_obj.obj); | |
| kfree(queue); | |
| pm_runtime_put_autosuspend(adev_to_drm(adev)->dev); | |
| return r; | |
| } | |
| static void amdgpu_userq_kref_destroy(struct kref *kref) | |
| { | |
| int r; | |
| struct amdgpu_usermode_queue *queue = | |
| container_of(kref, struct amdgpu_usermode_queue, refcount); | |
| struct amdgpu_userq_mgr *uq_mgr = queue->userq_mgr; | |
| r = amdgpu_userq_destroy(uq_mgr, queue); | |
| if (r) | |
| drm_file_err(uq_mgr->file, "Failed to destroy usermode queue %d\n", r); | |
| } | |
| struct amdgpu_usermode_queue *amdgpu_userq_get(struct amdgpu_userq_mgr *uq_mgr, u32 qid) | |
| { | |
| struct amdgpu_usermode_queue *queue; | |
| xa_lock(&uq_mgr->userq_xa); | |
| queue = xa_load(&uq_mgr->userq_xa, qid); | |
| if (queue) | |
| kref_get(&queue->refcount); | |
| xa_unlock(&uq_mgr->userq_xa); | |
| return queue; | |
| } | |
| void amdgpu_userq_put(struct amdgpu_usermode_queue *queue) | |
| { | |
| if (queue) | |
| kref_put(&queue->refcount, amdgpu_userq_kref_destroy); | |
| } | |
| static int amdgpu_userq_priority_permit(struct drm_file *filp, | |
| int priority) | |
| { | |
| if (priority < AMDGPU_USERQ_CREATE_FLAGS_QUEUE_PRIORITY_HIGH) | |
| return 0; | |
| if (capable(CAP_SYS_NICE)) | |
| return 0; | |
| if (drm_is_current_master(filp)) | |
| return 0; | |
| return -EACCES; | |
| } | |
| static int | |
| amdgpu_userq_create(struct drm_file *filp, union drm_amdgpu_userq *args) | |
| { | |
| struct amdgpu_fpriv *fpriv = filp->driver_priv; | |
| struct amdgpu_userq_mgr *uq_mgr = &fpriv->userq_mgr; | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| const struct amdgpu_userq_funcs *uq_funcs; | |
| struct amdgpu_usermode_queue *queue; | |
| struct amdgpu_db_info db_info; | |
| uint64_t index; | |
| int priority; | |
| u32 qid; | |
| int r; | |
| priority = | |
| (args->in.flags & AMDGPU_USERQ_CREATE_FLAGS_QUEUE_PRIORITY_MASK) | |
| >> AMDGPU_USERQ_CREATE_FLAGS_QUEUE_PRIORITY_SHIFT; | |
| r = amdgpu_userq_priority_permit(filp, priority); | |
| if (r) | |
| return r; | |
| r = pm_runtime_resume_and_get(adev_to_drm(adev)->dev); | |
| if (r < 0) { | |
| drm_file_err(uq_mgr->file, "pm_runtime_resume_and_get() failed for userqueue create\n"); | |
| return r; | |
| } | |
| uq_funcs = adev->userq_funcs[args->in.ip_type]; | |
| if (!uq_funcs) { | |
| r = -EINVAL; | |
| goto err_pm_runtime; | |
| } | |
| queue = kzalloc_obj(struct amdgpu_usermode_queue); | |
| if (!queue) { | |
| r = -ENOMEM; | |
| goto err_pm_runtime; | |
| } | |
| kref_init(&queue->refcount); | |
| queue->doorbell_handle = args->in.doorbell_handle; | |
| queue->queue_type = args->in.ip_type; | |
| queue->vm = &fpriv->vm; | |
| queue->priority = priority; | |
| queue->userq_mgr = uq_mgr; | |
| INIT_DELAYED_WORK(&queue->hang_detect_work, | |
| amdgpu_userq_hang_detect_work); | |
| r = amdgpu_userq_fence_driver_alloc(adev, &queue->fence_drv); | |
| if (r) | |
| goto free_queue; | |
| xa_init_flags(&queue->fence_drv_xa, XA_FLAGS_ALLOC); | |
| mutex_init(&queue->fence_drv_lock); | |
| /* Make sure the queue can actually run with those virtual addresses. */ | |
| r = amdgpu_bo_reserve(fpriv->vm.root.bo, false); | |
| if (r) | |
| goto free_fence_drv; | |
| if (amdgpu_userq_input_va_validate(adev, queue, args->in.queue_va, | |
| args->in.queue_size, | |
| &queue->userq_vas.va.queue_rb) || | |
| amdgpu_userq_input_va_validate(adev, queue, args->in.rptr_va, | |
| AMDGPU_GPU_PAGE_SIZE, | |
| &queue->userq_vas.va.rptr) || | |
| amdgpu_userq_input_va_validate(adev, queue, args->in.wptr_va, | |
| AMDGPU_GPU_PAGE_SIZE, | |
| &queue->userq_vas.va.wptr)) { | |
| r = -EINVAL; | |
| amdgpu_bo_unreserve(fpriv->vm.root.bo); | |
| goto free_fence_drv; | |
| } | |
| amdgpu_bo_unreserve(fpriv->vm.root.bo); | |
| /* Convert relative doorbell offset into absolute doorbell index */ | |
| db_info.queue_type = queue->queue_type; | |
| db_info.doorbell_handle = queue->doorbell_handle; | |
| db_info.db_obj = &queue->db_obj; | |
| db_info.doorbell_offset = args->in.doorbell_offset; | |
| r = amdgpu_userq_get_doorbell_index(uq_mgr, &db_info, filp, &index); | |
| if (r) { | |
| drm_file_err(uq_mgr->file, "Failed to get doorbell for queue\n"); | |
| goto free_fence_drv; | |
| } | |
| queue->doorbell_index = index; | |
| r = uq_funcs->mqd_create(queue, &args->in); | |
| if (r) { | |
| drm_file_err(uq_mgr->file, "Failed to create Queue\n"); | |
| goto clean_doorbell_bo; | |
| } | |
| /* Update VM owner at userq submit-time for page-fault attribution. */ | |
| amdgpu_vm_set_task_info(&fpriv->vm); | |
| r = xa_insert_irq(&adev->userq_doorbell_xa, index, queue, | |
| GFP_KERNEL); | |
| if (r) | |
| goto clean_mqd; | |
| amdgpu_userq_ensure_ev_fence(&fpriv->userq_mgr, &fpriv->evf_mgr); | |
| /* don't map the queue if scheduling is halted */ | |
| if (!adev->userq_halt_for_enforce_isolation || | |
| ((queue->queue_type != AMDGPU_HW_IP_GFX) && | |
| (queue->queue_type != AMDGPU_HW_IP_COMPUTE))) { | |
| r = amdgpu_userq_map_helper(queue); | |
| if (r) { | |
| drm_file_err(uq_mgr->file, "Failed to map Queue\n"); | |
| mutex_unlock(&uq_mgr->userq_mutex); | |
| goto erase_doorbell; | |
| } | |
| } | |
| atomic_inc(&uq_mgr->userq_count[queue->queue_type]); | |
| mutex_unlock(&uq_mgr->userq_mutex); | |
| r = xa_alloc(&uq_mgr->userq_xa, &qid, queue, | |
| XA_LIMIT(1, AMDGPU_MAX_USERQ_COUNT), | |
| GFP_KERNEL); | |
| if (r) { | |
| /* | |
| * This drops the last reference which should take care of | |
| * all cleanup. | |
| */ | |
| amdgpu_userq_put(queue); | |
| return r; | |
| } | |
| amdgpu_debugfs_userq_init(filp, queue, qid); | |
| args->out.queue_id = qid; | |
| return 0; | |
| erase_doorbell: | |
| xa_erase_irq(&adev->userq_doorbell_xa, index); | |
| clean_mqd: | |
| uq_funcs->mqd_destroy(queue); | |
| clean_doorbell_bo: | |
| amdgpu_bo_reserve(queue->db_obj.obj, true); | |
| amdgpu_bo_unpin(queue->db_obj.obj); | |
| amdgpu_bo_unreserve(queue->db_obj.obj); | |
| amdgpu_bo_unref(&queue->db_obj.obj); | |
| free_fence_drv: | |
| amdgpu_userq_fence_driver_free(queue); | |
| free_queue: | |
| kfree(queue); | |
| err_pm_runtime: | |
| pm_runtime_put_autosuspend(adev_to_drm(adev)->dev); | |
| return r; | |
| } | |
| static int amdgpu_userq_input_args_validate(struct drm_device *dev, | |
| union drm_amdgpu_userq *args, | |
| struct drm_file *filp) | |
| { | |
| struct amdgpu_device *adev = drm_to_adev(dev); | |
| switch (args->in.op) { | |
| case AMDGPU_USERQ_OP_CREATE: | |
| if (args->in.flags & ~(AMDGPU_USERQ_CREATE_FLAGS_QUEUE_PRIORITY_MASK | | |
| AMDGPU_USERQ_CREATE_FLAGS_QUEUE_SECURE)) | |
| return -EINVAL; | |
| /* Usermode queues are only supported for GFX IP as of now */ | |
| if (args->in.ip_type != AMDGPU_HW_IP_GFX && | |
| args->in.ip_type != AMDGPU_HW_IP_DMA && | |
| args->in.ip_type != AMDGPU_HW_IP_COMPUTE) { | |
| drm_file_err(filp, "Usermode queue doesn't support IP type %u\n", | |
| args->in.ip_type); | |
| return -EINVAL; | |
| } | |
| if ((args->in.flags & AMDGPU_USERQ_CREATE_FLAGS_QUEUE_SECURE) && | |
| (args->in.ip_type != AMDGPU_HW_IP_GFX) && | |
| (args->in.ip_type != AMDGPU_HW_IP_COMPUTE) && | |
| !amdgpu_is_tmz(adev)) { | |
| drm_file_err(filp, "Secure only supported on GFX/Compute queues\n"); | |
| return -EINVAL; | |
| } | |
| if (args->in.queue_va == AMDGPU_BO_INVALID_OFFSET || | |
| args->in.queue_va == 0 || | |
| args->in.queue_size == 0) { | |
| drm_file_err(filp, "invalidate userq queue va or size\n"); | |
| return -EINVAL; | |
| } | |
| if (!is_power_of_2(args->in.queue_size)) { | |
| drm_file_err(filp, "Queue size must be a power of 2\n"); | |
| return -EINVAL; | |
| } | |
| if (args->in.queue_size < AMDGPU_GPU_PAGE_SIZE) { | |
| drm_file_err(filp, "Queue size smaller than AMDGPU_GPU_PAGE_SIZE\n"); | |
| return -EINVAL; | |
| } | |
| if (!args->in.wptr_va || !args->in.rptr_va) { | |
| drm_file_err(filp, "invalidate userq queue rptr or wptr\n"); | |
| return -EINVAL; | |
| } | |
| break; | |
| case AMDGPU_USERQ_OP_FREE: | |
| if (args->in.ip_type || | |
| args->in.doorbell_handle || | |
| args->in.doorbell_offset || | |
| args->in.flags || | |
| args->in.queue_va || | |
| args->in.queue_size || | |
| args->in.rptr_va || | |
| args->in.wptr_va || | |
| args->in.mqd || | |
| args->in.mqd_size) | |
| return -EINVAL; | |
| break; | |
| default: | |
| return -EINVAL; | |
| } | |
| return 0; | |
| } | |
| bool amdgpu_userq_enabled(struct drm_device *dev) | |
| { | |
| struct amdgpu_device *adev = drm_to_adev(dev); | |
| int i; | |
| for (i = 0; i < AMDGPU_HW_IP_NUM; i++) { | |
| if (adev->userq_funcs[i]) | |
| return true; | |
| } | |
| return false; | |
| } | |
| int amdgpu_userq_ioctl(struct drm_device *dev, void *data, | |
| struct drm_file *filp) | |
| { | |
| union drm_amdgpu_userq *args = data; | |
| struct amdgpu_fpriv *fpriv = filp->driver_priv; | |
| struct amdgpu_usermode_queue *queue; | |
| int r = 0; | |
| if (!amdgpu_userq_enabled(dev)) | |
| return -ENOTSUPP; | |
| if (amdgpu_userq_input_args_validate(dev, args, filp) < 0) | |
| return -EINVAL; | |
| switch (args->in.op) { | |
| case AMDGPU_USERQ_OP_CREATE: | |
| r = amdgpu_userq_create(filp, args); | |
| if (r) | |
| drm_file_err(filp, "Failed to create usermode queue\n"); | |
| break; | |
| case AMDGPU_USERQ_OP_FREE: { | |
| xa_lock(&fpriv->userq_mgr.userq_xa); | |
| queue = __xa_erase(&fpriv->userq_mgr.userq_xa, args->in.queue_id); | |
| xa_unlock(&fpriv->userq_mgr.userq_xa); | |
| if (!queue) | |
| return -ENOENT; | |
| amdgpu_userq_put(queue); | |
| break; | |
| } | |
| default: | |
| drm_dbg_driver(dev, "Invalid user queue op specified: %d\n", args->in.op); | |
| return -EINVAL; | |
| } | |
| return r; | |
| } | |
| static int | |
| amdgpu_userq_restore_all(struct amdgpu_userq_mgr *uq_mgr) | |
| { | |
| struct amdgpu_fpriv *fpriv = uq_mgr_to_fpriv(uq_mgr); | |
| struct amdgpu_vm *vm = &fpriv->vm; | |
| struct amdgpu_usermode_queue *queue; | |
| unsigned long queue_id; | |
| int ret = 0, r; | |
| if (amdgpu_bo_reserve(vm->root.bo, false)) | |
| return false; | |
| mutex_lock(&uq_mgr->userq_mutex); | |
| /* Resume all the queues for this process */ | |
| xa_for_each(&uq_mgr->userq_xa, queue_id, queue) { | |
| if (!amdgpu_userq_buffer_vas_mapped(queue)) { | |
| drm_file_err(uq_mgr->file, | |
| "trying restore queue without va mapping\n"); | |
| queue->state = AMDGPU_USERQ_STATE_INVALID_VA; | |
| continue; | |
| } | |
| r = amdgpu_userq_map_helper(queue); | |
| if (r) | |
| ret = r; | |
| } | |
| mutex_unlock(&uq_mgr->userq_mutex); | |
| amdgpu_bo_unreserve(vm->root.bo); | |
| if (ret) | |
| drm_file_err(uq_mgr->file, | |
| "Failed to map all the queues, restore failed ret=%d\n", ret); | |
| return ret; | |
| } | |
| static int amdgpu_userq_validate_vm(void *param, struct amdgpu_bo *bo) | |
| { | |
| struct ttm_operation_ctx ctx = { false, false }; | |
| amdgpu_bo_placement_from_domain(bo, bo->allowed_domains); | |
| return ttm_bo_validate(&bo->tbo, &bo->placement, &ctx); | |
| } | |
| /* Handle all BOs on the invalidated list, validate them and update the PTs */ | |
| static int | |
| amdgpu_userq_bo_validate(struct amdgpu_device *adev, struct drm_exec *exec, | |
| struct amdgpu_vm *vm) | |
| { | |
| struct ttm_operation_ctx ctx = { false, false }; | |
| struct amdgpu_bo_va *bo_va; | |
| struct amdgpu_bo *bo; | |
| int ret; | |
| spin_lock(&vm->individual_lock); | |
| while (!list_empty(&vm->always_valid.evicted)) { | |
| bo_va = list_first_entry(&vm->always_valid.evicted, | |
| struct amdgpu_bo_va, | |
| base.vm_status); | |
| spin_unlock(&vm->individual_lock); | |
| bo = bo_va->base.bo; | |
| ret = drm_exec_prepare_obj(exec, &bo->tbo.base, 2); | |
| if (unlikely(ret)) | |
| return ret; | |
| amdgpu_bo_placement_from_domain(bo, bo->allowed_domains); | |
| ret = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx); | |
| if (ret) | |
| return ret; | |
| /* This moves the bo_va to the idle list */ | |
| ret = amdgpu_vm_bo_update(adev, bo_va, false); | |
| if (ret) | |
| return ret; | |
| spin_lock(&vm->individual_lock); | |
| } | |
| spin_unlock(&vm->individual_lock); | |
| return 0; | |
| } | |
| /* Make sure the whole VM is ready to be used */ | |
| static int | |
| amdgpu_userq_vm_validate(struct amdgpu_userq_mgr *uq_mgr) | |
| { | |
| struct amdgpu_fpriv *fpriv = uq_mgr_to_fpriv(uq_mgr); | |
| bool invalidated = false, new_addition = false; | |
| struct ttm_operation_ctx ctx = { true, false }; | |
| struct amdgpu_device *adev = uq_mgr->adev; | |
| struct amdgpu_hmm_range *range; | |
| struct amdgpu_vm *vm = &fpriv->vm; | |
| unsigned long key, tmp_key; | |
| struct amdgpu_bo_va *bo_va; | |
| struct amdgpu_bo *bo; | |
| struct drm_exec exec; | |
| struct xarray xa; | |
| int ret; | |
| xa_init(&xa); | |
| retry_lock: | |
| drm_exec_init(&exec, DRM_EXEC_IGNORE_DUPLICATES, 0); | |
| drm_exec_until_all_locked(&exec) { | |
| ret = amdgpu_vm_lock_pd(vm, &exec, 1); | |
| drm_exec_retry_on_contention(&exec); | |
| if (unlikely(ret)) | |
| goto unlock_all; | |
| ret = amdgpu_vm_lock_individual(vm, &exec, TTM_NUM_MOVE_FENCES + 1); | |
| drm_exec_retry_on_contention(&exec); | |
| if (unlikely(ret)) | |
| goto unlock_all; | |
| /* This validates PDs, PTs and per VM BOs */ | |
| ret = amdgpu_vm_validate(adev, vm, NULL, | |
| amdgpu_userq_validate_vm, | |
| NULL); | |
| if (unlikely(ret)) | |
| goto unlock_all; | |
| /* This locks and validates the remaining evicted BOs */ | |
| ret = amdgpu_userq_bo_validate(adev, &exec, vm); | |
| drm_exec_retry_on_contention(&exec); | |
| if (unlikely(ret)) | |
| goto unlock_all; | |
| } | |
| if (invalidated) { | |
| xa_for_each(&xa, tmp_key, range) { | |
| bo = range->bo; | |
| amdgpu_bo_placement_from_domain(bo, AMDGPU_GEM_DOMAIN_CPU); | |
| ret = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx); | |
| if (ret) | |
| goto unlock_all; | |
| amdgpu_ttm_tt_set_user_pages(bo->tbo.ttm, range); | |
| amdgpu_bo_placement_from_domain(bo, AMDGPU_GEM_DOMAIN_GTT); | |
| ret = ttm_bo_validate(&bo->tbo, &bo->placement, &ctx); | |
| if (ret) | |
| goto unlock_all; | |
| } | |
| invalidated = false; | |
| } | |
| ret = amdgpu_vm_handle_moved(adev, vm, NULL); | |
| if (ret) | |
| goto unlock_all; | |
| key = 0; | |
| /* Validate User Ptr BOs */ | |
| list_for_each_entry(bo_va, &vm->always_valid.idle, base.vm_status) { | |
| bo = bo_va->base.bo; | |
| if (!bo) | |
| continue; | |
| if (!amdgpu_ttm_tt_is_userptr(bo->tbo.ttm)) | |
| continue; | |
| range = xa_load(&xa, key); | |
| if (range && range->bo != bo) { | |
| xa_erase(&xa, key); | |
| amdgpu_hmm_range_free(range); | |
| range = NULL; | |
| } | |
| if (!range) { | |
| range = amdgpu_hmm_range_alloc(bo); | |
| if (!range) { | |
| ret = -ENOMEM; | |
| goto unlock_all; | |
| } | |
| xa_store(&xa, key, range, GFP_KERNEL); | |
| new_addition = true; | |
| } | |
| key++; | |
| } | |
| if (new_addition) { | |
| drm_exec_fini(&exec); | |
| xa_for_each(&xa, tmp_key, range) { | |
| if (!range) | |
| continue; | |
| bo = range->bo; | |
| ret = amdgpu_ttm_tt_get_user_pages(bo, range); | |
| if (ret) | |
| goto free_ranges; | |
| } | |
| invalidated = true; | |
| new_addition = false; | |
| goto retry_lock; | |
| } | |
| ret = amdgpu_vm_update_pdes(adev, vm, false); | |
| if (ret) | |
| goto unlock_all; | |
| /* | |
| * We need to wait for all VM updates to finish before restarting the | |
| * queues. Using the idle list like that is now ok since everything is | |
| * locked in place. | |
| */ | |
| list_for_each_entry(bo_va, &vm->always_valid.idle, base.vm_status) | |
| dma_fence_wait(bo_va->last_pt_update, false); | |
| dma_fence_wait(vm->last_update, false); | |
| ret = amdgpu_evf_mgr_rearm(&fpriv->evf_mgr, &exec); | |
| if (ret) | |
| drm_file_err(uq_mgr->file, "Failed to replace eviction fence\n"); | |
| unlock_all: | |
| drm_exec_fini(&exec); | |
| free_ranges: | |
| xa_for_each(&xa, tmp_key, range) { | |
| if (!range) | |
| continue; | |
| bo = range->bo; | |
| amdgpu_hmm_range_free(range); | |
| } | |
| xa_destroy(&xa); | |
| return ret; | |
| } | |
| static void amdgpu_userq_restore_worker(struct work_struct *work) | |
| { | |
| struct amdgpu_userq_mgr *uq_mgr = work_to_uq_mgr(work, resume_work.work); | |
| struct amdgpu_fpriv *fpriv = uq_mgr_to_fpriv(uq_mgr); | |
| struct dma_fence *ev_fence; | |
| int ret; | |
| ev_fence = amdgpu_evf_mgr_get_fence(&fpriv->evf_mgr); | |
| if (!dma_fence_is_signaled(ev_fence)) | |
| goto put_fence; | |
| ret = amdgpu_userq_vm_validate(uq_mgr); | |
| if (ret) { | |
| drm_file_err(uq_mgr->file, "Failed to validate BOs to restore ret=%d\n", ret); | |
| goto put_fence; | |
| } | |
| amdgpu_userq_restore_all(uq_mgr); | |
| put_fence: | |
| dma_fence_put(ev_fence); | |
| } | |
| static int | |
| amdgpu_userq_evict_all(struct amdgpu_userq_mgr *uq_mgr) | |
| { | |
| struct amdgpu_usermode_queue *queue; | |
| unsigned long queue_id; | |
| int ret = 0, r; | |
| /* Try to unmap all the queues in this process ctx */ | |
| xa_for_each(&uq_mgr->userq_xa, queue_id, queue) { | |
| r = amdgpu_userq_unmap_helper(queue); | |
| if (r) | |
| ret = r; | |
| } | |
| if (ret) { | |
| drm_file_err(uq_mgr->file, | |
| "Couldn't unmap all the queues, eviction failed ret=%d\n", ret); | |
| amdgpu_reset_domain_schedule(uq_mgr->adev->reset_domain, | |
| &uq_mgr->reset_work); | |
| flush_work(&uq_mgr->reset_work); | |
| } | |
| return ret; | |
| } | |
| static void | |
| amdgpu_userq_wait_for_signal(struct amdgpu_userq_mgr *uq_mgr) | |
| { | |
| struct amdgpu_usermode_queue *queue; | |
| unsigned long queue_id; | |
| xa_for_each(&uq_mgr->userq_xa, queue_id, queue) { | |
| struct dma_fence *f = queue->last_fence; | |
| if (!f) | |
| continue; | |
| dma_fence_wait(f, false); | |
| } | |
| } | |
| void | |
| amdgpu_userq_evict(struct amdgpu_userq_mgr *uq_mgr) | |
| { | |
| /* Wait for any pending userqueue fence work to finish */ | |
| amdgpu_userq_wait_for_signal(uq_mgr); | |
| amdgpu_userq_evict_all(uq_mgr); | |
| } | |
| int amdgpu_userq_mgr_init(struct amdgpu_userq_mgr *userq_mgr, struct drm_file *file_priv, | |
| struct amdgpu_device *adev) | |
| { | |
| mutex_init(&userq_mgr->userq_mutex); | |
| xa_init_flags(&userq_mgr->userq_xa, XA_FLAGS_ALLOC); | |
| userq_mgr->adev = adev; | |
| userq_mgr->file = file_priv; | |
| INIT_DELAYED_WORK(&userq_mgr->resume_work, amdgpu_userq_restore_worker); | |
| INIT_WORK(&userq_mgr->reset_work, amdgpu_userq_mgr_reset_work); | |
| return 0; | |
| } | |
| void amdgpu_userq_mgr_cancel_reset_work(struct amdgpu_device *adev) | |
| { | |
| struct xarray *xa = &adev->userq_doorbell_xa; | |
| struct amdgpu_usermode_queue *queue; | |
| unsigned long flags, queue_id; | |
| xa_lock_irqsave(xa, flags); | |
| xa_for_each(xa, queue_id, queue) { | |
| cancel_delayed_work(&queue->hang_detect_work); | |
| cancel_work(&queue->userq_mgr->reset_work); | |
| } | |
| xa_unlock_irqrestore(xa, flags); | |
| } | |
| void amdgpu_userq_mgr_cancel_resume(struct amdgpu_userq_mgr *userq_mgr) | |
| { | |
| cancel_delayed_work_sync(&userq_mgr->resume_work); | |
| } | |
| void amdgpu_userq_mgr_fini(struct amdgpu_userq_mgr *userq_mgr) | |
| { | |
| struct amdgpu_usermode_queue *queue; | |
| unsigned long queue_id = 0; | |
| for (;;) { | |
| xa_lock(&userq_mgr->userq_xa); | |
| queue = xa_find(&userq_mgr->userq_xa, &queue_id, ULONG_MAX, | |
| XA_PRESENT); | |
| if (queue) | |
| __xa_erase(&userq_mgr->userq_xa, queue_id); | |
| xa_unlock(&userq_mgr->userq_xa); | |
| if (!queue) | |
| break; | |
| amdgpu_userq_put(queue); | |
| } | |
| xa_destroy(&userq_mgr->userq_xa); | |
| /* | |
| * Drain any in-flight reset_work. By this point all queues are freed | |
| * and userq_count is 0, so if reset_work starts now it exits early. | |
| * We still need to wait in case it was already executing gpu_recover. | |
| */ | |
| cancel_work_sync(&userq_mgr->reset_work); | |
| mutex_destroy(&userq_mgr->userq_mutex); | |
| } | |
| int amdgpu_userq_suspend(struct amdgpu_device *adev) | |
| { | |
| u32 ip_mask = amdgpu_userq_get_supported_ip_mask(adev); | |
| struct amdgpu_usermode_queue *queue; | |
| struct amdgpu_userq_mgr *uqm; | |
| unsigned long queue_id; | |
| int r; | |
| if (!ip_mask) | |
| return 0; | |
| xa_for_each(&adev->userq_doorbell_xa, queue_id, queue) { | |
| uqm = queue->userq_mgr; | |
| cancel_delayed_work_sync(&uqm->resume_work); | |
| guard(mutex)(&uqm->userq_mutex); | |
| if (adev->in_s0ix) | |
| r = amdgpu_userq_preempt_helper(queue); | |
| else | |
| r = amdgpu_userq_unmap_helper(queue); | |
| if (r) | |
| return r; | |
| } | |
| return 0; | |
| } | |
| int amdgpu_userq_resume(struct amdgpu_device *adev) | |
| { | |
| u32 ip_mask = amdgpu_userq_get_supported_ip_mask(adev); | |
| struct amdgpu_usermode_queue *queue; | |
| struct amdgpu_userq_mgr *uqm; | |
| unsigned long queue_id; | |
| int r; | |
| if (!ip_mask) | |
| return 0; | |
| xa_for_each(&adev->userq_doorbell_xa, queue_id, queue) { | |
| uqm = queue->userq_mgr; | |
| guard(mutex)(&uqm->userq_mutex); | |
| if (adev->in_s0ix) | |
| r = amdgpu_userq_restore_helper(queue); | |
| else | |
| r = amdgpu_userq_map_helper(queue); | |
| if (r) | |
| return r; | |
| } | |
| return 0; | |
| } | |
| int amdgpu_userq_stop_sched_for_enforce_isolation(struct amdgpu_device *adev, | |
| u32 idx) | |
| { | |
| u32 ip_mask = amdgpu_userq_get_supported_ip_mask(adev); | |
| struct amdgpu_usermode_queue *queue; | |
| struct amdgpu_userq_mgr *uqm; | |
| unsigned long queue_id; | |
| int ret = 0, r; | |
| /* only need to stop gfx/compute */ | |
| if (!(ip_mask & ((1 << AMDGPU_HW_IP_GFX) | (1 << AMDGPU_HW_IP_COMPUTE)))) | |
| return 0; | |
| if (adev->userq_halt_for_enforce_isolation) | |
| dev_warn(adev->dev, "userq scheduling already stopped!\n"); | |
| adev->userq_halt_for_enforce_isolation = true; | |
| xa_for_each(&adev->userq_doorbell_xa, queue_id, queue) { | |
| uqm = queue->userq_mgr; | |
| cancel_delayed_work_sync(&uqm->resume_work); | |
| mutex_lock(&uqm->userq_mutex); | |
| if (((queue->queue_type == AMDGPU_HW_IP_GFX) || | |
| (queue->queue_type == AMDGPU_HW_IP_COMPUTE)) && | |
| (queue->xcp_id == idx)) { | |
| r = amdgpu_userq_preempt_helper(queue); | |
| if (r) | |
| ret = r; | |
| } | |
| mutex_unlock(&uqm->userq_mutex); | |
| } | |
| return ret; | |
| } | |
| int amdgpu_userq_start_sched_for_enforce_isolation(struct amdgpu_device *adev, | |
| u32 idx) | |
| { | |
| u32 ip_mask = amdgpu_userq_get_supported_ip_mask(adev); | |
| struct amdgpu_usermode_queue *queue; | |
| struct amdgpu_userq_mgr *uqm; | |
| unsigned long queue_id; | |
| int ret = 0, r; | |
| /* only need to stop gfx/compute */ | |
| if (!(ip_mask & ((1 << AMDGPU_HW_IP_GFX) | (1 << AMDGPU_HW_IP_COMPUTE)))) | |
| return 0; | |
| if (!adev->userq_halt_for_enforce_isolation) | |
| dev_warn(adev->dev, "userq scheduling already started!\n"); | |
| adev->userq_halt_for_enforce_isolation = false; | |
| xa_for_each(&adev->userq_doorbell_xa, queue_id, queue) { | |
| uqm = queue->userq_mgr; | |
| mutex_lock(&uqm->userq_mutex); | |
| if (((queue->queue_type == AMDGPU_HW_IP_GFX) || | |
| (queue->queue_type == AMDGPU_HW_IP_COMPUTE)) && | |
| (queue->xcp_id == idx)) { | |
| r = amdgpu_userq_restore_helper(queue); | |
| if (r) | |
| ret = r; | |
| } | |
| mutex_unlock(&uqm->userq_mutex); | |
| } | |
| return ret; | |
| } | |
| void amdgpu_userq_gem_va_unmap_validate(struct amdgpu_device *adev, | |
| struct amdgpu_bo_va_mapping *mapping) | |
| { | |
| u32 ip_mask = amdgpu_userq_get_supported_ip_mask(adev); | |
| struct amdgpu_bo_va *bo_va = mapping->bo_va; | |
| struct dma_resv *resv = bo_va->base.bo->tbo.base.resv; | |
| if (!ip_mask) | |
| return; | |
| /** | |
| * The userq VA mapping reservation should include the eviction fence. | |
| * Note: The eviction fence may be attached to different BOs and this | |
| * unmap is only for one kind of userq VAs, so at this point suppose | |
| * the eviction fence is always unsignaled. | |
| */ | |
| dma_resv_wait_timeout(resv, DMA_RESV_USAGE_BOOKKEEP, | |
| false, MAX_SCHEDULE_TIMEOUT); | |
| } | |
| void amdgpu_userq_pre_reset(struct amdgpu_device *adev) | |
| { | |
| const struct amdgpu_userq_funcs *userq_funcs; | |
| struct amdgpu_usermode_queue *queue; | |
| unsigned long queue_id; | |
| /* TODO: We probably need a new lock for the queue state */ | |
| xa_for_each(&adev->userq_doorbell_xa, queue_id, queue) { | |
| if (queue->state == AMDGPU_USERQ_STATE_MAPPED) { | |
| userq_funcs = adev->userq_funcs[queue->queue_type]; | |
| userq_funcs->unmap(queue); | |
| /* just mark all queues as hung at this point. | |
| * if unmap succeeds, we could map again | |
| * in amdgpu_userq_post_reset() if vram is not lost | |
| */ | |
| queue->state = AMDGPU_USERQ_STATE_HUNG; | |
| } | |
| /* Force-complete any pending fence regardless of queue state so | |
| * that eviction/suspend and queue teardown waiters don't block | |
| * forever on a fence that will never signal after the reset. | |
| */ | |
| amdgpu_userq_fence_driver_force_completion(queue); | |
| } | |
| } | |
| int amdgpu_userq_post_reset(struct amdgpu_device *adev, bool vram_lost) | |
| { | |
| /* if any queue state is AMDGPU_USERQ_STATE_UNMAPPED | |
| * at this point, we should be able to map it again | |
| * and continue if vram is not lost. | |
| */ | |
| struct amdgpu_usermode_queue *queue; | |
| const struct amdgpu_userq_funcs *userq_funcs; | |
| unsigned long queue_id; | |
| int r = 0; | |
| xa_for_each(&adev->userq_doorbell_xa, queue_id, queue) { | |
| if (queue->state == AMDGPU_USERQ_STATE_HUNG && !vram_lost) { | |
| userq_funcs = adev->userq_funcs[queue->queue_type]; | |
| /* Re-map queue */ | |
| r = userq_funcs->map(queue); | |
| if (r) { | |
| dev_err(adev->dev, "Failed to remap queue %ld\n", queue_id); | |
| continue; | |
| } | |
| queue->state = AMDGPU_USERQ_STATE_MAPPED; | |
| } | |
| } | |
| return r; | |
| } |