};
 builtin_platform_driver(hisi_pcie_driver);
 
-static int hisi_pcie_almost_ecam_probe(struct platform_device *pdev)
-{
-       struct device *dev = &pdev->dev;
-       struct pci_ecam_ops *ops;
-
-       ops = (struct pci_ecam_ops *)of_device_get_match_data(dev);
-       return pci_host_common_probe(pdev, ops);
-}
-
 static int hisi_pcie_platform_init(struct pci_config_window *cfg)
 {
        struct device *dev = cfg->parent;
 };
 
 static struct platform_driver hisi_pcie_almost_ecam_driver = {
-       .probe  = hisi_pcie_almost_ecam_probe,
+       .probe  = pci_host_common_probe,
        .driver = {
                   .name = "hisi-pcie-almost-ecam",
                   .of_match_table = hisi_pcie_almost_ecam_of_match,
 
 #include <linux/kernel.h>
 #include <linux/module.h>
 #include <linux/of_address.h>
+#include <linux/of_device.h>
 #include <linux/of_pci.h>
 #include <linux/pci-ecam.h>
 #include <linux/platform_device.h>
        return ERR_PTR(err);
 }
 
-int pci_host_common_probe(struct platform_device *pdev,
-                         const struct pci_ecam_ops *ops)
+int pci_host_common_probe(struct platform_device *pdev)
 {
        struct device *dev = &pdev->dev;
        struct pci_host_bridge *bridge;
        struct pci_config_window *cfg;
        struct list_head resources;
+       const struct pci_ecam_ops *ops;
        int ret;
 
+       ops = of_device_get_match_data(&pdev->dev);
+       if (!ops)
+               return -ENODEV;
+
        bridge = devm_pci_alloc_host_bridge(dev, 0);
        if (!bridge)
                return -ENOMEM;
 
 #include <linux/kernel.h>
 #include <linux/init.h>
 #include <linux/module.h>
-#include <linux/of_address.h>
-#include <linux/of_pci.h>
 #include <linux/pci-ecam.h>
 #include <linux/platform_device.h>
 
 };
 MODULE_DEVICE_TABLE(of, gen_pci_of_match);
 
-static int gen_pci_probe(struct platform_device *pdev)
-{
-       const struct of_device_id *of_id;
-       struct pci_ecam_ops *ops;
-
-       of_id = of_match_node(gen_pci_of_match, pdev->dev.of_node);
-       ops = (struct pci_ecam_ops *)of_id->data;
-
-       return pci_host_common_probe(pdev, ops);
-}
-
 static struct platform_driver gen_pci_driver = {
        .driver = {
                .name = "pci-host-generic",
                .of_match_table = gen_pci_of_match,
        },
-       .probe = gen_pci_probe,
+       .probe = pci_host_common_probe,
        .remove = pci_host_common_remove,
 };
 module_platform_driver(gen_pci_driver);
 
 #ifdef CONFIG_PCI_HOST_THUNDER_ECAM
 
 static const struct of_device_id thunder_ecam_of_match[] = {
-       { .compatible = "cavium,pci-host-thunder-ecam" },
+       {
+               .compatible = "cavium,pci-host-thunder-ecam",
+               .data = &pci_thunder_ecam_ops,
+       },
        { },
 };
 
-static int thunder_ecam_probe(struct platform_device *pdev)
-{
-       return pci_host_common_probe(pdev, &pci_thunder_ecam_ops);
-}
-
 static struct platform_driver thunder_ecam_driver = {
        .driver = {
                .name = KBUILD_MODNAME,
                .of_match_table = thunder_ecam_of_match,
                .suppress_bind_attrs = true,
        },
-       .probe = thunder_ecam_probe,
+       .probe = pci_host_common_probe,
 };
 builtin_platform_driver(thunder_ecam_driver);
 
 
 };
 
 static const struct of_device_id thunder_pem_of_match[] = {
-       { .compatible = "cavium,pci-host-thunder-pem" },
+       {
+               .compatible = "cavium,pci-host-thunder-pem",
+               .data = &pci_thunder_pem_ops,
+       },
        { },
 };
 
-static int thunder_pem_probe(struct platform_device *pdev)
-{
-       return pci_host_common_probe(pdev, &pci_thunder_pem_ops);
-}
-
 static struct platform_driver thunder_pem_driver = {
        .driver = {
                .name = KBUILD_MODNAME,
                .of_match_table = thunder_pem_of_match,
                .suppress_bind_attrs = true,
        },
-       .probe = thunder_pem_probe,
+       .probe = pci_host_common_probe,
 };
 builtin_platform_driver(thunder_pem_driver);
 
 
        spin_lock_init(&pcie->used_msi_lock);
        irq_set_chained_handler_and_data(virq, tango_msi_isr, pcie);
 
-       return pci_host_common_probe(pdev, &smp8759_ecam_ops);
+       return pci_host_common_probe(pdev);
 }
 
 static const struct of_device_id tango_pcie_ids[] = {
-       { .compatible = "sigma,smp8759-pcie" },
+       {
+               .compatible = "sigma,smp8759-pcie",
+               .data = &smp8759_ecam_ops,
+       },
        { },
 };
 
 
 
 #if IS_ENABLED(CONFIG_PCI_HOST_COMMON)
 /* for DT-based PCI controllers that support ECAM */
-int pci_host_common_probe(struct platform_device *pdev,
-                         const struct pci_ecam_ops *ops);
+int pci_host_common_probe(struct platform_device *pdev);
 int pci_host_common_remove(struct platform_device *pdev);
 #endif
 #endif