mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/stable/linux.git
synced 2026-08-09 06:14:34 +02:00
pre_reset only force-completes fences of MAPPED queues. A queue in any
other state (e.g. mid-eviction) keeps its last_fence pending; after a
GPU reset that fence never signals, so the eviction/suspend worker and
process teardown (amdgpu_evf_mgr_flush_suspend) wait on it forever and
wedge the machine:
INFO: task kworker/6:28 blocked for more than 120 seconds.
Workqueue: events amdgpu_eviction_fence_suspend_worker [amdgpu]
Call Trace:
dma_fence_wait_timeout+0x7e/0x130
amdgpu_userq_evict+0x67/0x140 [amdgpu]
amdgpu_eviction_fence_suspend_worker+0xd8/0x160 [amdgpu]
process_scheduled_works+0xa6/0x420
Force-complete every queue's fence regardless of state. The unmap and
mark-hung step stays gated on MAPPED, since unmapping a queue that is
not mapped is invalid.
Fixes: 290f46cf57 ("drm/amdgpu: Implement user queue reset functionality")
Reviewed-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Jesse Zhang <Jesse.Zhang@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
(cherry picked from commit 9102b39fa924dcc3dc75a3137bfa9633c40b88c0)
Cc: stable@vger.kernel.org
1422 lines
36 KiB
C
1422 lines
36 KiB
C
// 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;
|
|
}
|