Change-Id: I963598ba6eb44bc8620d70e026c0175d1a1de120
Signed-off-by: Chunming Zhou <david1.z...@amd.com>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu_device.c | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c
index de782ae..7e63ef9 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_device.c
@@ -2146,18 +2146,44 @@ retry:
                amdgpu_atombios_scratch_regs_restore(adev);
        }
        if (!r) {
+               struct amdgpu_ring *buffer_ring = adev->mman.buffer_funcs_ring;
+
                amdgpu_irq_gpu_reset_resume_helper(adev);
                r = amdgpu_ib_ring_tests(adev);
                if (r) {
                        dev_err(adev->dev, "ib ring test failed (%d).\n", r);
                        r = amdgpu_suspend(adev);
+                       need_full_reset = true;
                        goto retry;
                }
-
+               /**
+                * recovery vm page tables, since we cannot depend on VRAM is 
no problem
+                * after gpu full reset.
+                */
+               if (need_full_reset && !(adev->flags & AMD_IS_APU)) {
+                       struct amdgpu_vm *vm, *tmp;
+
+                       DRM_INFO("recover page table from shadow\n");
+                       amd_sched_rq_block_entity(
+                               
&buffer_ring->sched.sched_rq[AMD_SCHED_PRIORITY_NORMAL], true);
+                       kthread_unpark(buffer_ring->sched.thread);
+                       spin_lock(&adev->vm_list_lock);
+                       list_for_each_entry_safe(vm, tmp, &adev->vm_list, list) 
{
+                               spin_unlock(&adev->vm_list_lock);
+                               amdgpu_vm_recover_page_table_from_shadow(adev, 
vm);
+                               spin_lock(&adev->vm_list_lock);
+                       }
+                       spin_unlock(&adev->vm_list_lock);
+                       amd_sched_rq_block_entity(
+                               
&buffer_ring->sched.sched_rq[AMD_SCHED_PRIORITY_NORMAL], false);
+               }
                for (i = 0; i < AMDGPU_MAX_RINGS; ++i) {
                        struct amdgpu_ring *ring = adev->rings[i];
                        if (!ring)
                                continue;
+
+                       DRM_INFO("ring:%d recover jobs\n", ring->idx);
+                       kthread_park(buffer_ring->sched.thread);
                        amd_sched_job_recovery(&ring->sched);
                        kthread_unpark(ring->sched.thread);
                }
-- 
1.9.1

_______________________________________________
amd-gfx mailing list
amd-gfx@lists.freedesktop.org
https://lists.freedesktop.org/mailman/listinfo/amd-gfx

Reply via email to