CC: Guan Xuetao <gxt@mprc.pku.edu.cn>
Signed-off-by: Bjorn Helgaas <bhelgaas@google.com>
 #include <linux/slab.h>
 #include <linux/init.h>
 #include <linux/io.h>
+#include <asm-generic/pci-bridge.h>
 
 static int debug_pci;
-static int use_firmware;
 
 #define CONFIG_CMD(bus, devfn, where)  \
        (0x80000000 | (bus->number << 16) | (devfn << 8) | (where & ~3))
 
        pci_fixup_irqs(pci_common_swizzle, pci_puv3_map_irq);
 
-       if (!use_firmware) {
+       if (!pci_has_flag(PCI_PROBE_ONLY)) {
                /*
                 * Size the bridge windows.
                 */
                debug_pci = 1;
                return NULL;
        } else if (!strcmp(str, "firmware")) {
-               use_firmware = 1;
+               pci_add_flags(PCI_PROBE_ONLY);
                return NULL;
        }
        return str;