kirin_apb_ctrl_writel(kirin_pcie, val, SOC_PCIECTRL_CTRL1_ADDR);
 }
 
-static int kirin_pcie_rd_own_conf(struct pcie_port *pp,
+static int kirin_pcie_rd_own_conf(struct pci_bus *bus, unsigned int devfn,
                                  int where, int size, u32 *val)
 {
-       struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
-       struct kirin_pcie *kirin_pcie = to_kirin_pcie(pci);
-       int ret;
+       struct dw_pcie *pci = to_dw_pcie_from_pp(bus->sysdata);
 
-       kirin_pcie_sideband_dbi_r_mode(kirin_pcie, true);
-       ret = dw_pcie_read(pci->dbi_base + where, size, val);
-       kirin_pcie_sideband_dbi_r_mode(kirin_pcie, false);
+       if (PCI_SLOT(devfn)) {
+               *val = ~0;
+               return PCIBIOS_DEVICE_NOT_FOUND;
+       }
 
-       return ret;
+       *val = dw_pcie_read_dbi(pci, where, size);
+       return PCIBIOS_SUCCESSFUL;
 }
 
-static int kirin_pcie_wr_own_conf(struct pcie_port *pp,
+static int kirin_pcie_wr_own_conf(struct pci_bus *bus, unsigned int devfn,
                                  int where, int size, u32 val)
 {
-       struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
-       struct kirin_pcie *kirin_pcie = to_kirin_pcie(pci);
-       int ret;
+       struct dw_pcie *pci = to_dw_pcie_from_pp(bus->sysdata);
 
-       kirin_pcie_sideband_dbi_w_mode(kirin_pcie, true);
-       ret = dw_pcie_write(pci->dbi_base + where, size, val);
-       kirin_pcie_sideband_dbi_w_mode(kirin_pcie, false);
+       if (PCI_SLOT(devfn))
+               return PCIBIOS_DEVICE_NOT_FOUND;
 
-       return ret;
+       dw_pcie_write_dbi(pci, where, size, val);
+       return PCIBIOS_SUCCESSFUL;
 }
 
+static struct pci_ops kirin_pci_ops = {
+       .read = kirin_pcie_rd_own_conf,
+       .write = kirin_pcie_wr_own_conf,
+};
+
 static u32 kirin_pcie_read_dbi(struct dw_pcie *pci, void __iomem *base,
                               u32 reg, size_t size)
 {
 
 static int kirin_pcie_host_init(struct pcie_port *pp)
 {
+       pp->bridge->ops = &kirin_pci_ops;
+
        kirin_pcie_establish_link(pp);
 
        if (IS_ENABLED(CONFIG_PCI_MSI))
 };
 
 static const struct dw_pcie_host_ops kirin_pcie_host_ops = {
-       .rd_own_conf = kirin_pcie_rd_own_conf,
-       .wr_own_conf = kirin_pcie_wr_own_conf,
        .host_init = kirin_pcie_host_init,
 };