struct xe_device *xe = arg;
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_display_driver_remove_nogem(display);
        struct intel_display *display = xe->display;
        int err;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return 0;
 
        /* Fake uncore lock */
        struct intel_display *display = xe->display;
        int err;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return 0;
 
        err = intel_display_driver_probe(display);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_display_driver_register(display);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_power_domains_disable(display);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        if (master_ctl & DISPLAY_IRQ)
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        if (gu_misc_iir & GU_MISC_GSE)
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        gen11_display_irq_reset(display);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        if (gt->info.id == XE_GT0)
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        /*
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_dmc_resume(display);
        struct intel_display *display = xe->display;
        bool s2idle = suspend_to_idle();
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        /*
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_power_domains_disable(display);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        if (xe->d3cold.allowed) {
        struct intel_display *display = xe->display;
        bool s2idle = suspend_to_idle();
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_display_power_suspend_late(display, s2idle);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        if (xe->d3cold.allowed)
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        /*
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_display_power_resume_early(display);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        intel_dmc_resume(display);
 {
        struct intel_display *display = xe->display;
 
-       if (!display)
+       if (!xe->info.probe_display)
                return;
 
        if (xe->d3cold.allowed) {