#include <linux/irqchip/chained_irq.h>
 #include <linux/irqdomain.h>
 #include <linux/kernel.h>
+#include <linux/mfd/syscon.h>
 #include <linux/msi.h>
 #include <linux/module.h>
 #include <linux/of_address.h>
 #include <linux/phy/phy.h>
 #include <linux/platform_device.h>
 #include <linux/pm_runtime.h>
+#include <linux/regmap.h>
 #include <linux/reset.h>
 
 #include "../pci.h"
  * struct mtk_pcie - PCIe host information
  * @dev: pointer to PCIe device
  * @base: IO mapped register base
+ * @cfg: IO mapped register map for PCIe config
  * @free_ck: free-run reference clock
  * @mem: non-prefetchable memory resource
  * @ports: pointer to PCIe port information
 struct mtk_pcie {
        struct device *dev;
        void __iomem *base;
+       struct regmap *cfg;
        struct clk *free_ck;
 
        struct list_head ports;
                val |= PCIE_CSR_LTSSM_EN(port->slot) |
                       PCIE_CSR_ASPM_L1_EN(port->slot);
                writel(val, pcie->base + PCIE_SYS_CFG_V2);
+       } else if (pcie->cfg) {
+               val = PCIE_CSR_LTSSM_EN(port->slot) |
+                     PCIE_CSR_ASPM_L1_EN(port->slot);
+               regmap_update_bits(pcie->cfg, PCIE_SYS_CFG_V2, val, val);
        }
 
        /* Assert all reset signals */
        struct device *dev = pcie->dev;
        struct platform_device *pdev = to_platform_device(dev);
        struct resource *regs;
+       struct device_node *cfg_node;
        int err;
 
        /* get shared registers, which are optional */
                        return PTR_ERR(pcie->base);
        }
 
+       cfg_node = of_find_compatible_node(NULL, NULL,
+                                          "mediatek,generic-pciecfg");
+       if (cfg_node) {
+               pcie->cfg = syscon_node_to_regmap(cfg_node);
+               if (IS_ERR(pcie->cfg))
+                       return PTR_ERR(pcie->cfg);
+       }
+
        pcie->free_ck = devm_clk_get(dev, "free_ck");
        if (IS_ERR(pcie->free_ck)) {
                if (PTR_ERR(pcie->free_ck) == -EPROBE_DEFER)