}
 
 /* returns true if intel_display_reset_finish() needs to be called */
-bool intel_display_reset_prepare(struct intel_display *display)
+bool intel_display_reset_prepare(struct intel_display *display,
+                                modeset_stuck_fn modeset_stuck, void *context)
 {
-       struct drm_i915_private *dev_priv = to_i915(display->drm);
        struct drm_modeset_acquire_ctx *ctx = &display->restore.reset_ctx;
        struct drm_atomic_state *state;
        int ret;
        if (atomic_read(&display->restore.pending_fb_pin)) {
                drm_dbg_kms(display->drm,
                            "Modeset potentially stuck, unbreaking through wedging\n");
-               intel_gt_set_wedged(to_gt(dev_priv));
+               modeset_stuck(context);
        }
 
        /*
 
 
 struct intel_display;
 
+typedef void modeset_stuck_fn(void *context);
+
 bool intel_display_reset_test(struct intel_display *display);
-bool intel_display_reset_prepare(struct intel_display *display);
+bool intel_display_reset_prepare(struct intel_display *display,
+                                modeset_stuck_fn modeset_stuck, void *context);
 void intel_display_reset_finish(struct intel_display *display, bool test_only);
 
 #endif /* __INTEL_RESET_H__ */
 
        return err;
 }
 
+static void display_reset_modeset_stuck(void *gt)
+{
+       intel_gt_set_wedged(gt);
+}
+
 static void intel_gt_reset_global(struct intel_gt *gt,
                                  u32 engine_mask,
                                  const char *reason)
                        need_display_reset;
 
                if (reset_display)
-                       reset_display = intel_display_reset_prepare(display);
+                       reset_display = intel_display_reset_prepare(display,
+                                                                   display_reset_modeset_stuck,
+                                                                   gt);
 
                intel_gt_reset(gt, engine_mask, reason);