#include <linux/irqchip/chained_irq.h>
 #include <linux/irqdomain.h>
 #include <linux/mfd/syscon.h>
+#include <linux/module.h>
 #include <linux/of.h>
 #include <linux/pci.h>
 #include <linux/platform_device.h>
 #define cdns_pcie_to_rc(p) container_of(p, struct cdns_pcie_rc, pcie)
 
 #define ENABLE_REG_SYS_2       0x108
+#define ENABLE_CLR_REG_SYS_2   0x308
 #define STATUS_REG_SYS_2       0x508
 #define STATUS_CLR_REG_SYS_2   0x708
 #define LINK_DOWN              BIT(1)
        return IRQ_HANDLED;
 }
 
+static void j721e_pcie_disable_link_irq(struct j721e_pcie *pcie)
+{
+       u32 reg;
+
+       reg = j721e_pcie_intd_readl(pcie, ENABLE_CLR_REG_SYS_2);
+       reg |= pcie->linkdown_irq_regfield;
+       j721e_pcie_intd_writel(pcie, ENABLE_CLR_REG_SYS_2, reg);
+}
+
 static void j721e_pcie_config_link_irq(struct j721e_pcie *pcie)
 {
        u32 reg;
        struct j721e_pcie *pcie = platform_get_drvdata(pdev);
        struct cdns_pcie *cdns_pcie = pcie->cdns_pcie;
        struct device *dev = &pdev->dev;
+       struct cdns_pcie_ep *ep;
+       struct cdns_pcie_rc *rc;
+
+       if (pcie->mode == PCI_MODE_RC) {
+               rc = container_of(cdns_pcie, struct cdns_pcie_rc, pcie);
+               cdns_pcie_host_disable(rc);
+       } else {
+               ep = container_of(cdns_pcie, struct cdns_pcie_ep, pcie);
+               cdns_pcie_ep_disable(ep);
+       }
+
+       gpiod_set_value_cansleep(pcie->reset_gpio, 0);
 
        clk_disable_unprepare(pcie->refclk);
        cdns_pcie_disable_phy(cdns_pcie);
+       j721e_pcie_disable_link_irq(pcie);
        pm_runtime_put(dev);
        pm_runtime_disable(dev);
 }
                .pm     = pm_sleep_ptr(&j721e_pcie_pm_ops),
        },
 };
-builtin_platform_driver(j721e_pcie_driver);
+module_platform_driver(j721e_pcie_driver);
+
+MODULE_LICENSE("GPL");
+MODULE_DESCRIPTION("PCIe controller driver for TI's J721E and related SoCs");
+MODULE_AUTHOR("Kishon Vijay Abraham I <kishon@ti.com>");