#define eeh_pe_for_each_dev(pe, edev, tmp) \
                list_for_each_entry_safe(edev, tmp, &pe->edevs, list)
 
+#define eeh_for_each_pe(root, pe) \
+       for (pe = root; pe; pe = eeh_pe_next(pe, root))
+
 static inline bool eeh_pe_passed(struct eeh_pe *pe)
 {
        return pe ? !!atomic_read(&pe->pass_dev_cnt) : false;
 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_next(struct eeh_pe *pe, struct eeh_pe *root);
 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);
 
  * The function is used to retrieve the next PE in the
  * hierarchy PE tree.
  */
-static struct eeh_pe *eeh_pe_next(struct eeh_pe *pe,
-                                 struct eeh_pe *root)
+struct eeh_pe *eeh_pe_next(struct eeh_pe *pe, struct eeh_pe *root)
 {
        struct list_head *next = pe->child_list.next;
 
        struct eeh_pe *pe;
        void *ret;
 
-       for (pe = root; pe; pe = eeh_pe_next(pe, root)) {
+       eeh_for_each_pe(root, pe) {
                ret = fn(pe, flag);
                if (ret) return ret;
        }
        }
 
        /* Traverse root PE */
-       for (pe = root; pe; pe = eeh_pe_next(pe, root)) {
+       eeh_for_each_pe(root, pe) {
                eeh_pe_for_each_dev(pe, edev, tmp) {
                        ret = fn(edev, flag);
                        if (ret)