drm/amdgpu:only call flr_work under infinite timeout

Signed-off-by: Monk Liu <Monk.Liu@amd.com>
Reviewed-by: Christian König <christian.koenig@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>

1 files changed, 9 insertions, 6 deletions

diff --git a/drivers/gpu/drm/amd/amdgpu/mxgpu_ai.c b/drivers/gpu/drm/amd/amdgpu/mxgpu_ai.c
index 69da52d7e6e2..bde3ca3c21c1 100644
--- a/drivers/gpu/drm/amd/amdgpu/mxgpu_ai.c
+++ b/drivers/gpu/drm/amd/amdgpu/mxgpu_ai.c
@@ -266,12 +266,15 @@ static int xgpu_ai_mailbox_rcv_irq(struct amdgpu_device *adev,
 {
         int r;
 
-        /* see what event we get */
-        r = xgpu_ai_mailbox_rcv_msg(adev, IDH_FLR_NOTIFICATION);
-
-        /* only handle FLR_NOTIFY now */
-        if (!r)
-                schedule_work(&adev->virt.flr_work);
+        /* trigger gpu-reset by hypervisor only if TDR disbaled */
+        if (amdgpu_lockup_timeout == 0) {
+                /* see what event we get */
+                r = xgpu_ai_mailbox_rcv_msg(adev, IDH_FLR_NOTIFICATION);
+
+                /* only handle FLR_NOTIFY now */
+                if (!r)
+                        schedule_work(&adev->virt.flr_work);
+        }
 
         return 0;
 }