powerpc/eeh: Probe after unbalanced kref check

[linux-2.6-microblaze.git] / arch / powerpc / kernel / eeh.c

diff --git a/arch/powerpc/kernel/eeh.c b/arch/powerpc/kernel/eeh.c
index af9b597..8e61d71 100644 (file)
--- a/arch/powerpc/kernel/eeh.c
+++ b/arch/powerpc/kernel/eeh.c
@@ -1116,9 +1116,6 @@ void eeh_add_device_late(struct pci_dev *dev)
                return;
        }
 
-       if (eeh_has_flag(EEH_PROBE_MODE_DEV))
-               eeh_ops->probe(pdn, NULL);
-
        /*
         * The EEH cache might not be removed correctly because of
         * unbalanced kref to the device during unplug time, which
@@ -1142,6 +1139,9 @@ void eeh_add_device_late(struct pci_dev *dev)
                dev->dev.archdata.edev = NULL;
        }
 
+       if (eeh_has_flag(EEH_PROBE_MODE_DEV))
+               eeh_ops->probe(pdn, NULL);
+
        edev->pdev = dev;
        dev->dev.archdata.edev = edev;