#define        LINK_UP                                         BIT(16)
 #define        DRA7XX_CPU_TO_BUS_ADDR                          0x0FFFFFFF
 
-#define EXP_CAP_ID_OFFSET                              0x70
-
 #define        PCIECTRL_TI_CONF_INTX_ASSERT                    0x0124
 #define        PCIECTRL_TI_CONF_INTX_DEASSERT                  0x0128
 
        struct dra7xx_pcie *dra7xx = to_dra7xx_pcie(pci);
        struct device *dev = pci->dev;
        u32 reg;
-       u32 exp_cap_off = EXP_CAP_ID_OFFSET;
+       u32 exp_cap_off = dw_pcie_find_capability(pci, PCI_CAP_ID_EXP);
 
        if (dw_pcie_link_up(pci)) {
                dev_err(dev, "link is already up\n");
 
 #define LEG_EP                         0x1
 #define RC                             0x2
 
-#define EXP_CAP_ID_OFFSET              0x70
-
 #define KS_PCIE_SYSCLOCKOUTEN          BIT(0)
 
 #define AM654_PCIE_DEV_TYPE_MASK       0x3
 static void ks_pcie_set_link_speed(struct dw_pcie *pci, int link_speed)
 {
        u32 val;
+       u32 offset = dw_pcie_find_capability(pci, PCI_CAP_ID_EXP);
 
        dw_pcie_dbi_ro_wr_en(pci);
 
-       val = dw_pcie_readl_dbi(pci, EXP_CAP_ID_OFFSET + PCI_EXP_LNKCAP);
+       val = dw_pcie_readl_dbi(pci, offset + PCI_EXP_LNKCAP);
        if ((val & PCI_EXP_LNKCAP_SLS) != link_speed) {
                val &= ~((u32)PCI_EXP_LNKCAP_SLS);
                val |= link_speed;
-               dw_pcie_writel_dbi(pci, EXP_CAP_ID_OFFSET + PCI_EXP_LNKCAP,
+               dw_pcie_writel_dbi(pci, offset + PCI_EXP_LNKCAP,
                                   val);
        }
 
-       val = dw_pcie_readl_dbi(pci, EXP_CAP_ID_OFFSET + PCI_EXP_LNKCTL2);
+       val = dw_pcie_readl_dbi(pci, offset + PCI_EXP_LNKCTL2);
        if ((val & PCI_EXP_LNKCAP_SLS) != link_speed) {
                val &= ~((u32)PCI_EXP_LNKCAP_SLS);
                val |= link_speed;
-               dw_pcie_writel_dbi(pci, EXP_CAP_ID_OFFSET + PCI_EXP_LNKCTL2,
+               dw_pcie_writel_dbi(pci, offset + PCI_EXP_LNKCTL2,
                                   val);
        }
 
 
 /* CR6 */
 #define MSI_CTRL_INT                           (1 << 26)
 
-#define EXP_CAP_ID_OFFSET                      0x70
-
 #define to_spear13xx_pcie(x)   dev_get_drvdata((x)->dev)
 
 static int spear13xx_pcie_establish_link(struct spear13xx_pcie *spear13xx_pcie)
        struct pcie_port *pp = &pci->pp;
        struct pcie_app_reg *app_reg = spear13xx_pcie->app_base;
        u32 val;
-       u32 exp_cap_off = EXP_CAP_ID_OFFSET;
+       u32 exp_cap_off = dw_pcie_find_capability(pci, PCI_CAP_ID_EXP);
 
        if (dw_pcie_link_up(pci)) {
                dev_err(pci->dev, "link already up\n");