drm/xe: Do not wake device during a GT reset

commit b3fbda1a630a9439c885b2a5dc5230cc49a87e9e upstream.

Waking the device during a GT reset can lead to unintended memory
allocation, which is not allowed since GT resets occur in the reclaim
path. Prevent this by holding a PM reference while a reset is in flight.

Fixes: dd08ebf6c352 ("drm/xe: Introduce a new DRM driver for Intel GPUs")
Cc: stable@vger.kernel.org
Signed-off-by: Matthew Brost <matthew.brost@intel.com>
Reviewed-by: Matthew Auld <matthew.auld@intel.com>
Link: https://lore.kernel.org/r/20251022005538.828980-3-matthew.brost@intel.com
(cherry picked from commit 480b358e7d8ef69fd8f1b0cad6e07c7d70a36ee4)
Signed-off-by: Lucas De Marchi <lucas.demarchi@intel.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

diff --git a/drivers/gpu/drm/xe/xe_gt.c b/drivers/gpu/drm/xe/xe_gt.c
index 3b53d46aad54a72544492b58104d0d62c271e0df..de011f5629fdb54d1281e434ba9cdb7b9cd53352 100644 (file)
--- a/drivers/gpu/drm/xe/xe_gt.c
+++ b/drivers/gpu/drm/xe/xe_gt.c
@@ -746,17 +746,19 @@ static int gt_reset(struct xe_gt *gt)
 {
        int err;
 
-       if (xe_device_wedged(gt_to_xe(gt)))
-               return -ECANCELED;
+       if (xe_device_wedged(gt_to_xe(gt))) {
+               err = -ECANCELED;
+               goto err_pm_put;
+       }
 
        /* We only support GT resets with GuC submission */
-       if (!xe_device_uc_enabled(gt_to_xe(gt)))
-               return -ENODEV;
+       if (!xe_device_uc_enabled(gt_to_xe(gt))) {
+               err = -ENODEV;
+               goto err_pm_put;
+       }
 
        xe_gt_info(gt, "reset started\n");
 
-       xe_pm_runtime_get(gt_to_xe(gt));
-
        if (xe_fault_inject_gt_reset()) {
                err = -ECANCELED;
                goto err_fail;
@@ -803,6 +805,7 @@ err_fail:
        xe_gt_err(gt, "reset failed (%pe)\n", ERR_PTR(err));
 
        xe_device_declare_wedged(gt_to_xe(gt));
+err_pm_put:
        xe_pm_runtime_put(gt_to_xe(gt));
 
        return err;
@@ -824,7 +827,9 @@ void xe_gt_reset_async(struct xe_gt *gt)
                return;
 
        xe_gt_info(gt, "reset queued\n");
-       queue_work(gt->ordered_wq, &gt->reset.worker);
+       xe_pm_runtime_get_noresume(gt_to_xe(gt));
+       if (!queue_work(gt->ordered_wq, &gt->reset.worker))
+               xe_pm_runtime_put(gt_to_xe(gt));
 }
 
 void xe_gt_suspend_prepare(struct xe_gt *gt)