void eeh_set_pe_aux_size(int size);
 int eeh_phb_pe_create(struct pci_controller *phb);
 struct eeh_pe *eeh_phb_pe_get(struct pci_controller *phb);
-struct eeh_pe *eeh_pe_get(struct eeh_dev *edev);
+struct eeh_pe *eeh_pe_get(struct pci_controller *phb,
+                         int pe_no, int config_addr);
 int eeh_add_to_parent_pe(struct eeh_dev *edev);
 int eeh_rmv_from_parent_pe(struct eeh_dev *edev);
 void eeh_pe_update_time_stamp(struct eeh_pe *pe);
 
  * Bus/Device/Function number. The extra data referred by flag
  * indicates which type of address should be used.
  */
+struct eeh_pe_get_flag {
+       int pe_no;
+       int config_addr;
+};
+
 static void *__eeh_pe_get(void *data, void *flag)
 {
        struct eeh_pe *pe = (struct eeh_pe *)data;
-       struct eeh_dev *edev = (struct eeh_dev *)flag;
+       struct eeh_pe_get_flag *tmp = (struct eeh_pe_get_flag *) flag;
 
        /* Unexpected PHB PE */
        if (pe->type & EEH_PE_PHB)
         * have non-zero PE address
         */
        if (eeh_has_flag(EEH_VALID_PE_ZERO)) {
-               if (edev->pe_config_addr == pe->addr)
+               if (tmp->pe_no == pe->addr)
                        return pe;
        } else {
-               if (edev->pe_config_addr &&
-                   (edev->pe_config_addr == pe->addr))
+               if (tmp->pe_no &&
+                   (tmp->pe_no == pe->addr))
                        return pe;
        }
 
        /* Try BDF address */
-       if (edev->config_addr &&
-          (edev->config_addr == pe->config_addr))
+       if (tmp->config_addr &&
+          (tmp->config_addr == pe->config_addr))
                return pe;
 
        return NULL;
 
 /**
  * eeh_pe_get - Search PE based on the given address
- * @edev: EEH device
+ * @phb: PCI controller
+ * @pe_no: PE number
+ * @config_addr: Config address
  *
  * Search the corresponding PE based on the specified address which
  * is included in the eeh device. The function is used to check if
  * which is composed of PCI bus/device/function number, or unified
  * PE address.
  */
-struct eeh_pe *eeh_pe_get(struct eeh_dev *edev)
+struct eeh_pe *eeh_pe_get(struct pci_controller *phb,
+               int pe_no, int config_addr)
 {
-       struct eeh_pe *root = eeh_phb_pe_get(edev->phb);
+       struct eeh_pe *root = eeh_phb_pe_get(phb);
+       struct eeh_pe_get_flag tmp = { pe_no, config_addr };
        struct eeh_pe *pe;
 
-       pe = eeh_pe_traverse(root, __eeh_pe_get, edev);
+       pe = eeh_pe_traverse(root, __eeh_pe_get, &tmp);
 
        return pe;
 }
         * PE should be composed of PCI bus and its subordinate
         * components.
         */
-       pe = eeh_pe_get(edev);
+       pe = eeh_pe_get(edev->pdn->phb, edev->pe_config_addr,
+                       edev->config_addr);
        if (pe && !(pe->type & EEH_PE_INVALID)) {
                /* Mark the PE as type of PCI bus */
                pe->type = EEH_PE_BUS;
 
                                size_t count, loff_t *ppos)
 {
        struct pci_controller *hose = filp->private_data;
-       struct eeh_dev *edev;
        struct eeh_pe *pe;
        int pe_no, type, func;
        unsigned long addr, mask;
                return -EINVAL;
 
        /* Retrieve PE */
-       edev = kzalloc(sizeof(*edev), GFP_KERNEL);
-       if (!edev)
-               return -ENOMEM;
-       edev->phb = hose;
-       edev->pe_config_addr = pe_no;
-       pe = eeh_pe_get(edev);
-       kfree(edev);
+       pe = eeh_pe_get(hose, pe_no, 0);
        if (!pe)
                return -ENODEV;
 
        struct pnv_phb *phb = hose->private_data;
        struct pnv_ioda_pe *pnv_pe;
        struct eeh_pe *dev_pe;
-       struct eeh_dev edev;
 
        /*
         * If PHB supports compound PE, to fetch
        }
 
        /* Find the PE according to PE# */
-       memset(&edev, 0, sizeof(struct eeh_dev));
-       edev.phb = hose;
-       edev.pe_config_addr = pe_no;
-       dev_pe = eeh_pe_get(&edev);
+       dev_pe = eeh_pe_get(hose, pe_no, 0);
        if (!dev_pe)
                return -EEXIST;