530 lines · 14.4 KB · C
RawDownload
/* * Copyright 2015 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 <linux/kthread.h>#include <linux/wait.h>#include <linux/sched.h> #include <drm/drm_drv.h> #include "amdgpu.h"#include "amdgpu_trace.h"#include "amdgpu_reset.h"#include "amdgpu_dev_coredump.h"#include "amdgpu_xgmi.h" static void amdgpu_job_do_core_dump(struct amdgpu_device *adev,				    struct amdgpu_job *job){	int i; 	dev_info(adev->dev, "Dumping IP State\n");	for (i = 0; i < adev->num_ip_blocks; i++)		if (adev->ip_blocks[i].version->funcs->dump_ip_state)			adev->ip_blocks[i].version->funcs				->dump_ip_state((void *)&adev->ip_blocks[i]);	dev_info(adev->dev, "Dumping IP State Completed\n"); 	amdgpu_coredump(adev, true, false, job);} static void amdgpu_job_core_dump(struct amdgpu_device *adev,				 struct amdgpu_job *job){	struct list_head device_list, *device_list_handle =  NULL;	struct amdgpu_device *tmp_adev = NULL;	struct amdgpu_hive_info *hive = NULL; 	if (!amdgpu_sriov_vf(adev))		hive = amdgpu_get_xgmi_hive(adev);	if (hive)		mutex_lock(&hive->hive_lock);	/*	 * Reuse the logic in amdgpu_device_gpu_recover() to build list of	 * devices for code dump	 */	INIT_LIST_HEAD(&device_list);	if (!amdgpu_sriov_vf(adev) && (adev->gmc.xgmi.num_physical_nodes > 1) && hive) {		list_for_each_entry(tmp_adev, &hive->device_list, gmc.xgmi.head)			list_add_tail(&tmp_adev->reset_list, &device_list);		if (!list_is_first(&adev->reset_list, &device_list))			list_rotate_to_front(&adev->reset_list, &device_list);		device_list_handle = &device_list;	} else {		list_add_tail(&adev->reset_list, &device_list);		device_list_handle = &device_list;	} 	/* Do the coredump for each device */	list_for_each_entry(tmp_adev, device_list_handle, reset_list)		amdgpu_job_do_core_dump(tmp_adev, job); 	if (hive) {		mutex_unlock(&hive->hive_lock);		amdgpu_put_xgmi_hive(hive);	}} static enum drm_gpu_sched_stat amdgpu_job_timedout(struct drm_sched_job *s_job){	struct amdgpu_ring *ring = to_amdgpu_ring(s_job->sched);	struct amdgpu_job *job = to_amdgpu_job(s_job);	struct drm_wedge_task_info *info = NULL;	struct amdgpu_task_info *ti = NULL;	struct amdgpu_device *adev = ring->adev;	int idx, r; 	if (!drm_dev_enter(adev_to_drm(adev), &idx)) {		dev_info(adev->dev, "%s - device unplugged skipping recovery on scheduler:%s",			 __func__, s_job->sched->name); 		/* Effectively the job is aborted as the device is gone */		return DRM_GPU_SCHED_STAT_ENODEV;	} 	/*	 * Do the coredump immediately after a job timeout to get a very	 * close dump/snapshot/representation of GPU's current error status	 * Skip it for SRIOV, since VF FLR will be triggered by host driver	 * before job timeout	 */	if (!amdgpu_sriov_vf(adev))		amdgpu_job_core_dump(adev, job); 	if (amdgpu_gpu_recovery &&	    amdgpu_ring_is_reset_type_supported(ring, AMDGPU_RESET_TYPE_SOFT_RECOVERY) &&	    amdgpu_ring_soft_recovery(ring, job->vmid, s_job->s_fence->parent)) {		dev_err(adev->dev, "ring %s timeout, but soft recovered\n",			s_job->sched->name);		goto exit;	} 	dev_err(adev->dev, "ring %s timeout, signaled seq=%u, emitted seq=%u\n",		job->base.sched->name, atomic_read(&ring->fence_drv.last_seq),		ring->fence_drv.sync_seq); 	ti = amdgpu_vm_get_task_info_pasid(ring->adev, job->pasid);	if (ti) {		amdgpu_vm_print_task_info(adev, ti);		info = &ti->task;	} 	/* attempt a per ring reset */	if (amdgpu_gpu_recovery &&	    amdgpu_ring_is_reset_type_supported(ring, AMDGPU_RESET_TYPE_PER_QUEUE) &&	    ring->funcs->reset) {		dev_err(adev->dev, "Starting %s ring reset\n",			s_job->sched->name);		/* Stop the scheduler to prevent anybody else from touching the ring buffer. */		drm_sched_wqueue_stop(&ring->sched);		r = amdgpu_ring_reset(ring, job->vmid, job->hw_fence);		if (!r) {			/* Start the scheduler again */			drm_sched_wqueue_start(&ring->sched);			atomic_inc(&ring->adev->gpu_reset_counter);			dev_err(adev->dev, "Ring %s reset succeeded\n",				ring->sched.name);			drm_dev_wedged_event(adev_to_drm(adev),					     DRM_WEDGE_RECOVERY_NONE, info);			goto exit;		}		dev_err(adev->dev, "Ring %s reset failed\n", ring->sched.name);	} 	/* Attempt an IP block soft reset, if supported. */	if (amdgpu_gpu_recovery &&	    amdgpu_ring_is_reset_type_supported(ring, AMDGPU_RESET_TYPE_IP_BLOCK_SOFT_RESET)) {		r = amdgpu_device_ip_soft_reset(ring, job->hw_fence);		if (!r) {			atomic_inc(&ring->adev->gpu_reset_counter);			drm_dev_wedged_event(adev_to_drm(adev), DRM_WEDGE_RECOVERY_NONE, info);			goto exit;		}	} 	if (dma_fence_get_status(&s_job->s_fence->finished) == 0)		dma_fence_set_error(&s_job->s_fence->finished, -ETIME); 	if (amdgpu_device_should_recover_gpu(ring->adev)) {		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_JOB;		clear_bit(AMDGPU_NEED_FULL_RESET, &reset_context.flags); 		/*		 * To avoid an unnecessary extra coredump, as we have already		 * got the very close representation of GPU's error status		 */		set_bit(AMDGPU_SKIP_COREDUMP, &reset_context.flags); 		r = amdgpu_device_gpu_recover(ring->adev, job, &reset_context);		if (r)			dev_err(adev->dev, "GPU Recovery Failed: %d\n", r);	} else {		drm_sched_suspend_timeout(&ring->sched);		if (amdgpu_sriov_vf(adev))			adev->virt.tdr_debug = true;	} exit:	amdgpu_vm_put_task_info(ti);	drm_dev_exit(idx);	/* This is needed to add the job back to the pending list */	return DRM_GPU_SCHED_STAT_NO_HANG;} int amdgpu_job_alloc(struct amdgpu_device *adev, struct amdgpu_vm *vm,		     struct drm_sched_entity *entity, void *owner,		     unsigned int num_ibs, u64 drm_client_id,		     gfp_t gfp_flags, struct amdgpu_job **job){	struct amdgpu_fence *af;	int r; 	if (num_ibs == 0)		return -EINVAL; 	*job = kzalloc_flex(**job, ibs, num_ibs, gfp_flags);	if (!*job)		return -ENOMEM; 	af = kzalloc_obj(struct amdgpu_fence, gfp_flags);	if (!af) {		r = -ENOMEM;		goto err_job;	}	(*job)->hw_fence = af; 	af = kzalloc_obj(struct amdgpu_fence, gfp_flags);	if (!af) {		r = -ENOMEM;		goto err_fence;	}	(*job)->hw_vm_fence = af; 	(*job)->vm = vm; 	amdgpu_sync_create(&(*job)->explicit_sync);	(*job)->generation = amdgpu_vm_generation(adev, vm);	(*job)->vm_pd_addr = AMDGPU_BO_INVALID_OFFSET; 	if (!entity)		return 0; 	r = drm_sched_job_init(&(*job)->base, entity, 1, owner, drm_client_id);	if (!r)		return 0; 	kfree((*job)->hw_vm_fence); err_fence:	kfree((*job)->hw_fence);err_job:	kfree(*job);	*job = NULL; 	return r;} int amdgpu_job_alloc_with_ib(struct amdgpu_device *adev,			     struct drm_sched_entity *entity, void *owner,			     size_t size, enum amdgpu_ib_pool_type pool_type,			     u64 k_job_id, struct amdgpu_job **job){	int r; 	r = amdgpu_job_alloc(adev, NULL, entity, owner, 1, k_job_id,			     amdgpu_ib_pool_gfp_flags(adev, pool_type),			     job);	if (r)		return r; 	(*job)->num_ibs = 1;	r = amdgpu_ib_get(adev, NULL, size, pool_type, &(*job)->ibs[0]);	if (r) {		if (entity)			drm_sched_job_cleanup(&(*job)->base);		kfree((*job)->hw_vm_fence);		kfree((*job)->hw_fence);		kfree(*job);		*job = NULL;	} 	return r;} void amdgpu_job_set_resources(struct amdgpu_job *job, struct amdgpu_bo *gds,			      struct amdgpu_bo *gws, struct amdgpu_bo *oa){	if (gds) {		job->gds_base = amdgpu_bo_gpu_offset(gds) >> PAGE_SHIFT;		job->gds_size = amdgpu_bo_size(gds) >> PAGE_SHIFT;	}	if (gws) {		job->gws_base = amdgpu_bo_gpu_offset(gws) >> PAGE_SHIFT;		job->gws_size = amdgpu_bo_size(gws) >> PAGE_SHIFT;	}	if (oa) {		job->oa_base = amdgpu_bo_gpu_offset(oa) >> PAGE_SHIFT;		job->oa_size = amdgpu_bo_size(oa) >> PAGE_SHIFT;	}} void amdgpu_job_free_resources(struct amdgpu_job *job){	struct dma_fence *f;	unsigned i; 	/* Check if any fences were initialized */	if (job->base.s_fence &&	    dma_fence_was_initialized(&job->base.s_fence->finished))		f = &job->base.s_fence->finished;	else if (dma_fence_was_initialized(&job->hw_fence->base))		f = &job->hw_fence->base;	else		f = NULL; 	for (i = 0; i < job->num_ibs; ++i)		amdgpu_ib_free(&job->ibs[i], f);} static void amdgpu_job_free_cb(struct drm_sched_job *s_job){	struct amdgpu_job *job = to_amdgpu_job(s_job); 	drm_sched_job_cleanup(s_job); 	amdgpu_sync_free(&job->explicit_sync); 	if (dma_fence_was_initialized(&job->hw_fence->base))		dma_fence_put(&job->hw_fence->base);	else		kfree(job->hw_fence);	if (dma_fence_was_initialized(&job->hw_vm_fence->base))		dma_fence_put(&job->hw_vm_fence->base);	else		kfree(job->hw_vm_fence); 	kfree(job);} void amdgpu_job_set_gang_leader(struct amdgpu_job *job,				struct amdgpu_job *leader){	struct dma_fence *fence = &leader->base.s_fence->scheduled; 	WARN_ON(job->gang_submit); 	/*	 * Don't add a reference when we are the gang leader to avoid circle	 * dependency.	 */	if (job != leader)		dma_fence_get(fence);	job->gang_submit = fence;} void amdgpu_job_free(struct amdgpu_job *job){	if (job->base.entity)		drm_sched_job_cleanup(&job->base); 	amdgpu_job_free_resources(job);	amdgpu_sync_free(&job->explicit_sync);	if (job->gang_submit != &job->base.s_fence->scheduled)		dma_fence_put(job->gang_submit); 	if (dma_fence_was_initialized(&job->hw_fence->base))		dma_fence_put(&job->hw_fence->base);	else		kfree(job->hw_fence);	if (dma_fence_was_initialized(&job->hw_vm_fence->base))		dma_fence_put(&job->hw_vm_fence->base);	else		kfree(job->hw_vm_fence); 	kfree(job);} struct dma_fence *amdgpu_job_submit(struct amdgpu_job *job){	struct dma_fence *f; 	drm_sched_job_arm(&job->base);	f = dma_fence_get(&job->base.s_fence->finished);	amdgpu_job_free_resources(job);	drm_sched_entity_push_job(&job->base); 	return f;} int amdgpu_job_submit_direct(struct amdgpu_job *job, struct amdgpu_ring *ring,			     struct dma_fence **fence){	int r; 	job->base.sched = &ring->sched;	r = amdgpu_ib_schedule(ring, job->num_ibs, job->ibs, job, fence); 	if (r)		return r; 	amdgpu_job_free(job);	return 0;} static struct dma_fence *amdgpu_job_prepare_job(struct drm_sched_job *sched_job,		      struct drm_sched_entity *s_entity){	struct amdgpu_ring *ring = to_amdgpu_ring(s_entity->rq->sched);	struct amdgpu_job *job = to_amdgpu_job(sched_job);	struct dma_fence *fence;	int r; 	r = drm_sched_entity_error(s_entity);	if (r)		goto error; 	if (job->gang_submit) {		fence = amdgpu_device_switch_gang(ring->adev, job->gang_submit);		if (fence)			return fence;	} 	fence = amdgpu_device_enforce_isolation(ring->adev, ring, job);	if (fence)		return fence; 	if (job->vm && !job->vmid) {		r = amdgpu_vmid_grab(job->vm, ring, job, &fence);		if (r) {			dev_err(ring->adev->dev, "Error getting VM ID (%d)\n", r);			goto error;		}		return fence;	} 	return NULL; error:	dma_fence_set_error(&job->base.s_fence->finished, r);	return NULL;} static struct dma_fence *amdgpu_job_run(struct drm_sched_job *sched_job){	struct amdgpu_ring *ring = to_amdgpu_ring(sched_job->sched);	struct amdgpu_device *adev = ring->adev;	struct dma_fence *fence = NULL, *finished;	struct amdgpu_job *job;	int r = 0; 	job = to_amdgpu_job(sched_job);	finished = &job->base.s_fence->finished; 	trace_amdgpu_sched_run_job(job); 	/* Skip job if VRAM is lost and never resubmit gangs */	if (job->generation != amdgpu_vm_generation(adev, job->vm) ||	    (job->job_run_counter && job->gang_submit))		dma_fence_set_error(finished, -ECANCELED); 	if (finished->error < 0) {		dev_dbg(adev->dev, "Skip scheduling IBs in ring(%s)",			ring->name);	} else {		r = amdgpu_ib_schedule(ring, job->num_ibs, job->ibs, job,				       &fence);		if (r)			dev_err(adev->dev,				"Error scheduling IBs (%d) in ring(%s)", r,				ring->name);	} 	job->job_run_counter++;	amdgpu_job_free_resources(job); 	fence = r ? ERR_PTR(r) : fence;	return fence;} /* * This is a duplicate function from DRM scheduler sched_internal.h. * Plan is to remove it when amdgpu_job_stop_all_jobs_on_sched is removed, due * latter being incorrect and racy. * * See https://lore.kernel.org/amd-gfx/44edde63-7181-44fb-a4f7-94e50514f539@amd.com/ */static struct drm_sched_job *drm_sched_entity_queue_pop(struct drm_sched_entity *entity){	struct spsc_node *node; 	node = spsc_queue_pop(&entity->job_queue);	if (!node)		return NULL; 	return container_of(node, struct drm_sched_job, queue_node);} void amdgpu_job_stop_all_jobs_on_sched(struct drm_gpu_scheduler *sched){	struct drm_sched_job *s_job;	struct drm_sched_entity *s_entity = NULL;	int i; 	/* Signal all jobs not yet scheduled */

Lines 1–500 of 530.