}
 EXPORT_SYMBOL_GPL(dw_pcie_ep_init_notify);
 
-static void __dw_pcie_ep_reset_bar(struct dw_pcie *pci, enum pci_barno bar,
-                                  int flags)
+static unsigned int dw_pcie_ep_func_select(struct dw_pcie_ep *ep, u8 func_no)
+{
+       unsigned int func_offset = 0;
+
+       if (ep->ops->func_conf_select)
+               func_offset = ep->ops->func_conf_select(ep, func_no);
+
+       return func_offset;
+}
+
+static void __dw_pcie_ep_reset_bar(struct dw_pcie *pci, u8 func_no,
+                                  enum pci_barno bar, int flags)
 {
        u32 reg;
+       unsigned int func_offset = 0;
+       struct dw_pcie_ep *ep = &pci->ep;
+
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
 
-       reg = PCI_BASE_ADDRESS_0 + (4 * bar);
+       reg = func_offset + PCI_BASE_ADDRESS_0 + (4 * bar);
        dw_pcie_dbi_ro_wr_en(pci);
        dw_pcie_writel_dbi2(pci, reg, 0x0);
        dw_pcie_writel_dbi(pci, reg, 0x0);
 
 void dw_pcie_ep_reset_bar(struct dw_pcie *pci, enum pci_barno bar)
 {
-       __dw_pcie_ep_reset_bar(pci, bar, 0);
+       u8 func_no, funcs;
+
+       funcs = pci->ep.epc->max_functions;
+
+       for (func_no = 0; func_no < funcs; func_no++)
+               __dw_pcie_ep_reset_bar(pci, func_no, bar, 0);
 }
 
 static int dw_pcie_ep_write_header(struct pci_epc *epc, u8 func_no,
 {
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
+       unsigned int func_offset = 0;
+
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
 
        dw_pcie_dbi_ro_wr_en(pci);
-       dw_pcie_writew_dbi(pci, PCI_VENDOR_ID, hdr->vendorid);
-       dw_pcie_writew_dbi(pci, PCI_DEVICE_ID, hdr->deviceid);
-       dw_pcie_writeb_dbi(pci, PCI_REVISION_ID, hdr->revid);
-       dw_pcie_writeb_dbi(pci, PCI_CLASS_PROG, hdr->progif_code);
-       dw_pcie_writew_dbi(pci, PCI_CLASS_DEVICE,
+       dw_pcie_writew_dbi(pci, func_offset + PCI_VENDOR_ID, hdr->vendorid);
+       dw_pcie_writew_dbi(pci, func_offset + PCI_DEVICE_ID, hdr->deviceid);
+       dw_pcie_writeb_dbi(pci, func_offset + PCI_REVISION_ID, hdr->revid);
+       dw_pcie_writeb_dbi(pci, func_offset + PCI_CLASS_PROG, hdr->progif_code);
+       dw_pcie_writew_dbi(pci, func_offset + PCI_CLASS_DEVICE,
                           hdr->subclass_code | hdr->baseclass_code << 8);
-       dw_pcie_writeb_dbi(pci, PCI_CACHE_LINE_SIZE,
+       dw_pcie_writeb_dbi(pci, func_offset + PCI_CACHE_LINE_SIZE,
                           hdr->cache_line_size);
-       dw_pcie_writew_dbi(pci, PCI_SUBSYSTEM_VENDOR_ID,
+       dw_pcie_writew_dbi(pci, func_offset + PCI_SUBSYSTEM_VENDOR_ID,
                           hdr->subsys_vendor_id);
-       dw_pcie_writew_dbi(pci, PCI_SUBSYSTEM_ID, hdr->subsys_id);
-       dw_pcie_writeb_dbi(pci, PCI_INTERRUPT_PIN,
+       dw_pcie_writew_dbi(pci, func_offset + PCI_SUBSYSTEM_ID, hdr->subsys_id);
+       dw_pcie_writeb_dbi(pci, func_offset + PCI_INTERRUPT_PIN,
                           hdr->interrupt_pin);
        dw_pcie_dbi_ro_wr_dis(pci);
 
        return 0;
 }
 
-static int dw_pcie_ep_inbound_atu(struct dw_pcie_ep *ep, enum pci_barno bar,
-                                 dma_addr_t cpu_addr,
+static int dw_pcie_ep_inbound_atu(struct dw_pcie_ep *ep, u8 func_no,
+                                 enum pci_barno bar, dma_addr_t cpu_addr,
                                  enum dw_pcie_as_type as_type)
 {
        int ret;
                return -EINVAL;
        }
 
-       ret = dw_pcie_prog_inbound_atu(pci, free_win, bar, cpu_addr,
+       ret = dw_pcie_prog_inbound_atu(pci, func_no, free_win, bar, cpu_addr,
                                       as_type);
        if (ret < 0) {
                dev_err(pci->dev, "Failed to program IB window\n");
        return 0;
 }
 
-static int dw_pcie_ep_outbound_atu(struct dw_pcie_ep *ep, phys_addr_t phys_addr,
+static int dw_pcie_ep_outbound_atu(struct dw_pcie_ep *ep, u8 func_no,
+                                  phys_addr_t phys_addr,
                                   u64 pci_addr, size_t size)
 {
        u32 free_win;
                return -EINVAL;
        }
 
-       dw_pcie_prog_outbound_atu(pci, free_win, PCIE_ATU_TYPE_MEM,
-                                 phys_addr, pci_addr, size);
+       dw_pcie_prog_ep_outbound_atu(pci, func_no, free_win, PCIE_ATU_TYPE_MEM,
+                                    phys_addr, pci_addr, size);
 
        set_bit(free_win, ep->ob_window_map);
        ep->outbound_addr[free_win] = phys_addr;
        enum pci_barno bar = epf_bar->barno;
        u32 atu_index = ep->bar_to_atu[bar];
 
-       __dw_pcie_ep_reset_bar(pci, bar, epf_bar->flags);
+       __dw_pcie_ep_reset_bar(pci, func_no, bar, epf_bar->flags);
 
        dw_pcie_disable_atu(pci, atu_index, DW_PCIE_REGION_INBOUND);
        clear_bit(atu_index, ep->ib_window_map);
        size_t size = epf_bar->size;
        int flags = epf_bar->flags;
        enum dw_pcie_as_type as_type;
-       u32 reg = PCI_BASE_ADDRESS_0 + (4 * bar);
+       u32 reg;
+       unsigned int func_offset = 0;
+
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
+
+       reg = PCI_BASE_ADDRESS_0 + (4 * bar) + func_offset;
 
        if (!(flags & PCI_BASE_ADDRESS_SPACE))
                as_type = DW_PCIE_AS_MEM;
        else
                as_type = DW_PCIE_AS_IO;
 
-       ret = dw_pcie_ep_inbound_atu(ep, bar, epf_bar->phys_addr, as_type);
+       ret = dw_pcie_ep_inbound_atu(ep, func_no, bar,
+                                    epf_bar->phys_addr, as_type);
        if (ret)
                return ret;
 
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
 
-       ret = dw_pcie_ep_outbound_atu(ep, addr, pci_addr, size);
+       ret = dw_pcie_ep_outbound_atu(ep, func_no, addr, pci_addr, size);
        if (ret) {
                dev_err(pci->dev, "Failed to enable address\n");
                return ret;
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
        u32 val, reg;
+       unsigned int func_offset = 0;
 
        if (!ep->msi_cap)
                return -EINVAL;
 
-       reg = ep->msi_cap + PCI_MSI_FLAGS;
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
+
+       reg = ep->msi_cap + func_offset + PCI_MSI_FLAGS;
        val = dw_pcie_readw_dbi(pci, reg);
        if (!(val & PCI_MSI_FLAGS_ENABLE))
                return -EINVAL;
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
        u32 val, reg;
+       unsigned int func_offset = 0;
 
        if (!ep->msi_cap)
                return -EINVAL;
 
-       reg = ep->msi_cap + PCI_MSI_FLAGS;
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
+
+       reg = ep->msi_cap + func_offset + PCI_MSI_FLAGS;
        val = dw_pcie_readw_dbi(pci, reg);
        val &= ~PCI_MSI_FLAGS_QMASK;
        val |= (interrupts << 1) & PCI_MSI_FLAGS_QMASK;
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
        u32 val, reg;
+       unsigned int func_offset = 0;
 
        if (!ep->msix_cap)
                return -EINVAL;
 
-       reg = ep->msix_cap + PCI_MSIX_FLAGS;
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
+
+       reg = ep->msix_cap + func_offset + PCI_MSIX_FLAGS;
        val = dw_pcie_readw_dbi(pci, reg);
        if (!(val & PCI_MSIX_FLAGS_ENABLE))
                return -EINVAL;
        struct dw_pcie_ep *ep = epc_get_drvdata(epc);
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
        u32 val, reg;
+       unsigned int func_offset = 0;
 
        if (!ep->msix_cap)
                return -EINVAL;
 
        dw_pcie_dbi_ro_wr_en(pci);
 
-       reg = ep->msix_cap + PCI_MSIX_FLAGS;
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
+
+       reg = ep->msix_cap + func_offset + PCI_MSIX_FLAGS;
        val = dw_pcie_readw_dbi(pci, reg);
        val &= ~PCI_MSIX_FLAGS_QSIZE;
        val |= interrupts;
        dw_pcie_writew_dbi(pci, reg, val);
 
-       reg = ep->msix_cap + PCI_MSIX_TABLE;
+       reg = ep->msix_cap + func_offset + PCI_MSIX_TABLE;
        val = offset | bir;
        dw_pcie_writel_dbi(pci, reg, val);
 
-       reg = ep->msix_cap + PCI_MSIX_PBA;
+       reg = ep->msix_cap + func_offset + PCI_MSIX_PBA;
        val = (offset + (interrupts * PCI_MSIX_ENTRY_SIZE)) | bir;
        dw_pcie_writel_dbi(pci, reg, val);
 
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
        struct pci_epc *epc = ep->epc;
        unsigned int aligned_offset;
+       unsigned int func_offset = 0;
        u16 msg_ctrl, msg_data;
        u32 msg_addr_lower, msg_addr_upper, reg;
        u64 msg_addr;
        if (!ep->msi_cap)
                return -EINVAL;
 
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
+
        /* Raise MSI per the PCI Local Bus Specification Revision 3.0, 6.8.1. */
-       reg = ep->msi_cap + PCI_MSI_FLAGS;
+       reg = ep->msi_cap + func_offset + PCI_MSI_FLAGS;
        msg_ctrl = dw_pcie_readw_dbi(pci, reg);
        has_upper = !!(msg_ctrl & PCI_MSI_FLAGS_64BIT);
-       reg = ep->msi_cap + PCI_MSI_ADDRESS_LO;
+       reg = ep->msi_cap + func_offset + PCI_MSI_ADDRESS_LO;
        msg_addr_lower = dw_pcie_readl_dbi(pci, reg);
        if (has_upper) {
-               reg = ep->msi_cap + PCI_MSI_ADDRESS_HI;
+               reg = ep->msi_cap + func_offset + PCI_MSI_ADDRESS_HI;
                msg_addr_upper = dw_pcie_readl_dbi(pci, reg);
-               reg = ep->msi_cap + PCI_MSI_DATA_64;
+               reg = ep->msi_cap + func_offset + PCI_MSI_DATA_64;
                msg_data = dw_pcie_readw_dbi(pci, reg);
        } else {
                msg_addr_upper = 0;
-               reg = ep->msi_cap + PCI_MSI_DATA_32;
+               reg = ep->msi_cap + func_offset + PCI_MSI_DATA_32;
                msg_data = dw_pcie_readw_dbi(pci, reg);
        }
        aligned_offset = msg_addr_lower & (epc->mem->window.page_size - 1);
 }
 
 int dw_pcie_ep_raise_msix_irq(struct dw_pcie_ep *ep, u8 func_no,
-                            u16 interrupt_num)
+                             u16 interrupt_num)
 {
        struct dw_pcie *pci = to_dw_pcie_from_ep(ep);
        struct pci_epf_msix_tbl *msix_tbl;
        struct pci_epc *epc = ep->epc;
+       unsigned int func_offset = 0;
        u32 reg, msg_data, vec_ctrl;
        unsigned int aligned_offset;
        u32 tbl_offset;
        int ret;
        u8 bir;
 
-       reg = ep->msix_cap + PCI_MSIX_TABLE;
+       func_offset = dw_pcie_ep_func_select(ep, func_no);
+
+       reg = ep->msix_cap + func_offset + PCI_MSIX_TABLE;
        tbl_offset = dw_pcie_readl_dbi(pci, reg);
        bir = (tbl_offset & PCI_MSIX_TABLE_BIR);
        tbl_offset &= PCI_MSIX_TABLE_OFFSET;
        ep->epc = epc;
        epc_set_drvdata(epc, ep);
 
-       if (ep->ops->ep_init)
-               ep->ops->ep_init(ep);
-
        ret = of_property_read_u8(np, "max-functions", &epc->max_functions);
        if (ret < 0)
                epc->max_functions = 1;
 
+       if (ep->ops->ep_init)
+               ep->ops->ep_init(ep);
+
        ret = pci_epc_mem_init(epc, ep->phys_base, ep->addr_size,
                               ep->page_size);
        if (ret < 0) {
 
        dw_pcie_writel_atu(pci, offset + reg, val);
 }
 
-static void dw_pcie_prog_outbound_atu_unroll(struct dw_pcie *pci, int index,
-                                            int type, u64 cpu_addr,
-                                            u64 pci_addr, u32 size)
+static void dw_pcie_prog_outbound_atu_unroll(struct dw_pcie *pci, u8 func_no,
+                                            int index, int type,
+                                            u64 cpu_addr, u64 pci_addr,
+                                            u32 size)
 {
        u32 retries, val;
        u64 limit_addr = cpu_addr + size - 1;
        dw_pcie_writel_ob_unroll(pci, index, PCIE_ATU_UNR_UPPER_TARGET,
                                 upper_32_bits(pci_addr));
        dw_pcie_writel_ob_unroll(pci, index, PCIE_ATU_UNR_REGION_CTRL1,
-                                type);
+                                type | PCIE_ATU_FUNC_NUM(func_no));
        dw_pcie_writel_ob_unroll(pci, index, PCIE_ATU_UNR_REGION_CTRL2,
                                 PCIE_ATU_ENABLE);
 
        dev_err(pci->dev, "Outbound iATU is not being enabled\n");
 }
 
-void dw_pcie_prog_outbound_atu(struct dw_pcie *pci, int index, int type,
-                              u64 cpu_addr, u64 pci_addr, u32 size)
+static void __dw_pcie_prog_outbound_atu(struct dw_pcie *pci, u8 func_no,
+                                       int index, int type, u64 cpu_addr,
+                                       u64 pci_addr, u32 size)
 {
        u32 retries, val;
 
                cpu_addr = pci->ops->cpu_addr_fixup(pci, cpu_addr);
 
        if (pci->iatu_unroll_enabled) {
-               dw_pcie_prog_outbound_atu_unroll(pci, index, type, cpu_addr,
-                                                pci_addr, size);
+               dw_pcie_prog_outbound_atu_unroll(pci, func_no, index, type,
+                                                cpu_addr, pci_addr, size);
                return;
        }
 
                           lower_32_bits(pci_addr));
        dw_pcie_writel_dbi(pci, PCIE_ATU_UPPER_TARGET,
                           upper_32_bits(pci_addr));
-       dw_pcie_writel_dbi(pci, PCIE_ATU_CR1, type);
+       dw_pcie_writel_dbi(pci, PCIE_ATU_CR1, type |
+                          PCIE_ATU_FUNC_NUM(func_no));
        dw_pcie_writel_dbi(pci, PCIE_ATU_CR2, PCIE_ATU_ENABLE);
 
        /*
        dev_err(pci->dev, "Outbound iATU is not being enabled\n");
 }
 
+void dw_pcie_prog_outbound_atu(struct dw_pcie *pci, int index, int type,
+                              u64 cpu_addr, u64 pci_addr, u32 size)
+{
+       __dw_pcie_prog_outbound_atu(pci, 0, index, type,
+                                   cpu_addr, pci_addr, size);
+}
+
+void dw_pcie_prog_ep_outbound_atu(struct dw_pcie *pci, u8 func_no, int index,
+                                 int type, u64 cpu_addr, u64 pci_addr,
+                                 u32 size)
+{
+       __dw_pcie_prog_outbound_atu(pci, func_no, index, type,
+                                   cpu_addr, pci_addr, size);
+}
+
 static u32 dw_pcie_readl_ib_unroll(struct dw_pcie *pci, u32 index, u32 reg)
 {
        u32 offset = PCIE_GET_ATU_INB_UNR_REG_OFFSET(index);
        dw_pcie_writel_atu(pci, offset + reg, val);
 }
 
-static int dw_pcie_prog_inbound_atu_unroll(struct dw_pcie *pci, int index,
-                                          int bar, u64 cpu_addr,
+static int dw_pcie_prog_inbound_atu_unroll(struct dw_pcie *pci, u8 func_no,
+                                          int index, int bar, u64 cpu_addr,
                                           enum dw_pcie_as_type as_type)
 {
        int type;
                return -EINVAL;
        }
 
-       dw_pcie_writel_ib_unroll(pci, index, PCIE_ATU_UNR_REGION_CTRL1, type);
+       dw_pcie_writel_ib_unroll(pci, index, PCIE_ATU_UNR_REGION_CTRL1, type |
+                                PCIE_ATU_FUNC_NUM(func_no));
        dw_pcie_writel_ib_unroll(pci, index, PCIE_ATU_UNR_REGION_CTRL2,
+                                PCIE_ATU_FUNC_NUM_MATCH_EN |
                                 PCIE_ATU_ENABLE |
                                 PCIE_ATU_BAR_MODE_ENABLE | (bar << 8));
 
        return -EBUSY;
 }
 
-int dw_pcie_prog_inbound_atu(struct dw_pcie *pci, int index, int bar,
-                            u64 cpu_addr, enum dw_pcie_as_type as_type)
+int dw_pcie_prog_inbound_atu(struct dw_pcie *pci, u8 func_no, int index,
+                            int bar, u64 cpu_addr,
+                            enum dw_pcie_as_type as_type)
 {
        int type;
        u32 retries, val;
 
        if (pci->iatu_unroll_enabled)
-               return dw_pcie_prog_inbound_atu_unroll(pci, index, bar,
+               return dw_pcie_prog_inbound_atu_unroll(pci, func_no, index, bar,
                                                       cpu_addr, as_type);
 
        dw_pcie_writel_dbi(pci, PCIE_ATU_VIEWPORT, PCIE_ATU_REGION_INBOUND |
                return -EINVAL;
        }
 
-       dw_pcie_writel_dbi(pci, PCIE_ATU_CR1, type);
-       dw_pcie_writel_dbi(pci, PCIE_ATU_CR2, PCIE_ATU_ENABLE
-                          | PCIE_ATU_BAR_MODE_ENABLE | (bar << 8));
+       dw_pcie_writel_dbi(pci, PCIE_ATU_CR1, type |
+                          PCIE_ATU_FUNC_NUM(func_no));
+       dw_pcie_writel_dbi(pci, PCIE_ATU_CR2, PCIE_ATU_ENABLE |
+                          PCIE_ATU_FUNC_NUM_MATCH_EN |
+                          PCIE_ATU_BAR_MODE_ENABLE | (bar << 8));
 
        /*
         * Make sure ATU enable takes effect before any subsequent config