// SPDX-License-Identifier: GPL-2.0+
 /*
- * PCIe host controller driver for Tegra194 SoC
+ * PCIe host controller driver for the following SoCs
+ * Tegra194
+ * Tegra234
  *
- * Copyright (C) 2019 NVIDIA Corporation.
+ * Copyright (C) 2019-2022 NVIDIA Corporation.
  *
  * Author: Vidya Sagar <vidyas@nvidia.com>
  */
 #include <soc/tegra/bpmp-abi.h>
 #include "../../pci.h"
 
+#define TEGRA194_DWC_IP_VER                    0x490A
+#define TEGRA234_DWC_IP_VER                    0x562A
+
 #define APPL_PINMUX                            0x0
 #define APPL_PINMUX_PEX_RST                    BIT(0)
 #define APPL_PINMUX_CLKREQ_OVERRIDE_EN         BIT(2)
 #define APPL_CTRL_HW_HOT_RST_MODE_MASK         GENMASK(1, 0)
 #define APPL_CTRL_HW_HOT_RST_MODE_SHIFT                22
 #define APPL_CTRL_HW_HOT_RST_MODE_IMDT_RST     0x1
+#define APPL_CTRL_HW_HOT_RST_MODE_IMDT_RST_LTSSM_EN    0x2
 
 #define APPL_INTR_EN_L0_0                      0x8
 #define APPL_INTR_EN_L0_0_LINK_STATE_INT_EN    BIT(0)
        GEN4_CORE_CLK_FREQ
 };
 
+struct tegra_pcie_dw_of_data {
+       u32 version;
+       enum dw_pcie_device_mode mode;
+       bool has_msix_doorbell_access_fix;
+       bool has_sbr_reset_fix;
+       bool has_l1ss_exit_fix;
+       bool has_ltr_req_fix;
+       u32 cdm_chk_int_en_bit;
+       u32 gen4_preset_vec;
+       u8 n_fts[2];
+};
+
 struct tegra_pcie_dw {
        struct device *dev;
        struct resource *appl_res;
        struct dw_pcie pci;
        struct tegra_bpmp *bpmp;
 
-       enum dw_pcie_device_mode mode;
+       struct tegra_pcie_dw_of_data *of_data;
 
        bool supports_clkreq;
        bool enable_cdm_check;
+       bool enable_srns;
        bool link_state;
        bool update_fc_fixup;
+       bool enable_ext_refclk;
        u8 init_link_width;
        u32 msi_ctrl_int;
        u32 num_lanes;
        int ep_state;
 };
 
-struct tegra_pcie_dw_of_data {
-       enum dw_pcie_device_mode mode;
-};
-
 static inline struct tegra_pcie_dw *to_tegra_pcie(struct dw_pcie *pci)
 {
        return container_of(pci, struct tegra_pcie_dw, pci);
        if (status_l0 & APPL_INTR_STATUS_L0_LINK_STATE_INT) {
                status_l1 = appl_readl(pcie, APPL_INTR_STATUS_L1_0_0);
                appl_writel(pcie, status_l1, APPL_INTR_STATUS_L1_0_0);
-               if (status_l1 & APPL_INTR_STATUS_L1_0_0_LINK_REQ_RST_NOT_CHGED) {
+               if (!pcie->of_data->has_sbr_reset_fix &&
+                   status_l1 & APPL_INTR_STATUS_L1_0_0_LINK_REQ_RST_NOT_CHGED) {
                        /* SBR & Surprise Link Down WAR */
                        val = appl_readl(pcie, APPL_CAR_RESET_OVRD);
                        val &= ~APPL_CAR_RESET_OVRD_CYA_OVERRIDE_CORE_RST_N;
                PCI_EXP_LNKSTA_CLS;
        clk_set_rate(pcie->core_clk, pcie_gen_freq[speed - 1]);
 
+       if (pcie->of_data->has_ltr_req_fix)
+               return IRQ_HANDLED;
+
        /* If EP doesn't advertise L1SS, just return */
        val = dw_pcie_readl_dbi(pci, pcie->cfg_link_cap_l1sub);
        if (!(val & (PCI_L1SS_CAP_ASPM_L1_1 | PCI_L1SS_CAP_ASPM_L1_2)))
 static int tegra_pcie_dw_rd_own_conf(struct pci_bus *bus, u32 devfn, int where,
                                     int size, u32 *val)
 {
+       struct pcie_port *pp = bus->sysdata;
+       struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
+       struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
+
        /*
         * This is an endpoint mode specific register happen to appear even
         * when controller is operating in root port mode and system hangs
         * when it is accessed with link being in ASPM-L1 state.
         * So skip accessing it altogether
         */
-       if (!PCI_SLOT(devfn) && where == PORT_LOGIC_MSIX_DOORBELL) {
+       if (!pcie->of_data->has_msix_doorbell_access_fix &&
+           !PCI_SLOT(devfn) && where == PORT_LOGIC_MSIX_DOORBELL) {
                *val = 0x00000000;
                return PCIBIOS_SUCCESSFUL;
        }
 static int tegra_pcie_dw_wr_own_conf(struct pci_bus *bus, u32 devfn, int where,
                                     int size, u32 val)
 {
+       struct pcie_port *pp = bus->sysdata;
+       struct dw_pcie *pci = to_dw_pcie_from_pp(pp);
+       struct tegra_pcie_dw *pcie = to_tegra_pcie(pci);
+
        /*
         * This is an endpoint mode specific register happen to appear even
         * when controller is operating in root port mode and system hangs
         * when it is accessed with link being in ASPM-L1 state.
         * So skip accessing it altogether
         */
-       if (!PCI_SLOT(devfn) && where == PORT_LOGIC_MSIX_DOORBELL)
+       if (!pcie->of_data->has_msix_doorbell_access_fix &&
+           !PCI_SLOT(devfn) && where == PORT_LOGIC_MSIX_DOORBELL)
                return PCIBIOS_SUCCESSFUL;
 
        return pci_generic_config_write(bus, devfn, where, size, val);
        val |= APPL_INTR_EN_L0_0_LINK_STATE_INT_EN;
        appl_writel(pcie, val, APPL_INTR_EN_L0_0);
 
-       val = appl_readl(pcie, APPL_INTR_EN_L1_0_0);
-       val |= APPL_INTR_EN_L1_0_0_LINK_REQ_RST_NOT_INT_EN;
-       appl_writel(pcie, val, APPL_INTR_EN_L1_0_0);
+       if (!pcie->of_data->has_sbr_reset_fix) {
+               val = appl_readl(pcie, APPL_INTR_EN_L1_0_0);
+               val |= APPL_INTR_EN_L1_0_0_LINK_REQ_RST_NOT_INT_EN;
+               appl_writel(pcie, val, APPL_INTR_EN_L1_0_0);
+       }
 
        if (pcie->enable_cdm_check) {
                val = appl_readl(pcie, APPL_INTR_EN_L0_0);
-               val |= APPL_INTR_EN_L0_0_CDM_REG_CHK_INT_EN;
+               val |= pcie->of_data->cdm_chk_int_en_bit;
                appl_writel(pcie, val, APPL_INTR_EN_L0_0);
 
                val = appl_readl(pcie, APPL_INTR_EN_L1_18);
 
        val = dw_pcie_readl_dbi(pci, GEN3_EQ_CONTROL_OFF);
        val &= ~GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_MASK;
-       val |= (0x360 << GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_SHIFT);
+       val |= (pcie->of_data->gen4_preset_vec <<
+               GEN3_EQ_CONTROL_OFF_PSET_REQ_VEC_SHIFT);
        val &= ~GEN3_EQ_CONTROL_OFF_FB_MODE_MASK;
        dw_pcie_writel_dbi(pci, GEN3_EQ_CONTROL_OFF, val);
 
        val |= (pcie->num_lanes << PCI_EXP_LNKSTA_NLW_SHIFT);
        dw_pcie_writel_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKCAP, val);
 
+       /* Clear Slot Clock Configuration bit if SRNS configuration */
+       if (pcie->enable_srns) {
+               val_16 = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base +
+                                          PCI_EXP_LNKSTA);
+               val_16 &= ~PCI_EXP_LNKSTA_SLC;
+               dw_pcie_writew_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKSTA,
+                                  val_16);
+       }
+
        config_gen3_gen4_eq_presets(pcie);
 
        init_host_aspm(pcie);
                disable_aspm_l12(pcie);
        }
 
-       val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
-       val &= ~GEN3_RELATED_OFF_GEN3_ZRXDC_NONCOMPL;
-       dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
+       if (!pcie->of_data->has_l1ss_exit_fix) {
+               val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
+               val &= ~GEN3_RELATED_OFF_GEN3_ZRXDC_NONCOMPL;
+               dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
+       }
 
        if (pcie->update_fc_fixup) {
                val = dw_pcie_readl_dbi(pci, CFG_TIMER_CTRL_MAX_FUNC_NUM_OFF);
        struct pcie_port *pp = &pci->pp;
        bool retry = true;
 
-       if (pcie->mode == DW_PCIE_EP_TYPE) {
+       if (pcie->of_data->mode == DW_PCIE_EP_TYPE) {
                enable_irq(pcie->pex_rst_irq);
                return 0;
        }
        if (of_property_read_bool(np, "nvidia,update-fc-fixup"))
                pcie->update_fc_fixup = true;
 
+       /* RP using an external REFCLK is supported only in Tegra234 */
+       if (pcie->of_data->version == TEGRA194_DWC_IP_VER) {
+               if (pcie->of_data->mode == DW_PCIE_EP_TYPE)
+                       pcie->enable_ext_refclk = true;
+       } else {
+               pcie->enable_ext_refclk =
+                       of_property_read_bool(pcie->dev->of_node,
+                                             "nvidia,enable-ext-refclk");
+       }
+
        pcie->supports_clkreq =
                of_property_read_bool(pcie->dev->of_node, "supports-clkreq");
 
        pcie->enable_cdm_check =
                of_property_read_bool(np, "snps,enable-cdm-check");
 
-       if (pcie->mode == DW_PCIE_RC_TYPE)
+       if (pcie->of_data->version == TEGRA234_DWC_IP_VER)
+               pcie->enable_srns =
+                       of_property_read_bool(np, "nvidia,enable-srns");
+
+       if (pcie->of_data->mode == DW_PCIE_RC_TYPE)
                return 0;
 
        /* Endpoint mode specific DT entries */
        struct tegra_bpmp_message msg;
        struct mrq_uphy_request req;
 
-       /* Controller-5 doesn't need to have its state set by BPMP-FW */
-       if (pcie->cid == 5)
+       /*
+        * Controller-5 doesn't need to have its state set by BPMP-FW in
+        * Tegra194
+        */
+       if (pcie->of_data->version == TEGRA194_DWC_IP_VER && pcie->cid == 5)
                return 0;
 
        memset(&req, 0, sizeof(req));
                return ret;
        }
 
+       if (pcie->enable_ext_refclk) {
+               ret = tegra_pcie_bpmp_set_pll_state(pcie, true);
+               if (ret) {
+                       dev_err(pcie->dev, "Failed to init UPHY: %d\n", ret);
+                       goto fail_pll_init;
+               }
+       }
+
        ret = tegra_pcie_enable_slot_regulators(pcie);
        if (ret < 0)
                goto fail_slot_reg_en;
                goto fail_core_apb_rst;
        }
 
-       if (en_hw_hot_rst) {
+       if (en_hw_hot_rst || pcie->of_data->has_sbr_reset_fix) {
                /* Enable HW_HOT_RST mode */
                val = appl_readl(pcie, APPL_CTRL);
                val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
                         APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
+               val |= (APPL_CTRL_HW_HOT_RST_MODE_IMDT_RST_LTSSM_EN <<
+                       APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
                val |= APPL_CTRL_HW_HOT_RST_EN;
                appl_writel(pcie, val, APPL_CTRL);
        }
        val |= (APPL_CFG_MISC_ARCACHE_VAL << APPL_CFG_MISC_ARCACHE_SHIFT);
        appl_writel(pcie, val, APPL_CFG_MISC);
 
+       if (pcie->enable_srns || pcie->enable_ext_refclk) {
+               /*
+                * When Tegra PCIe RP is using external clock, it cannot supply
+                * same clock to its downstream hierarchy. Hence, gate PCIe RP
+                * REFCLK out pads when RP & EP are using separate clocks or RP
+                * is using an external REFCLK.
+                */
+               val = appl_readl(pcie, APPL_PINMUX);
+               val |= APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE_EN;
+               val &= ~APPL_PINMUX_CLK_OUTPUT_IN_OVERRIDE;
+               appl_writel(pcie, val, APPL_PINMUX);
+       }
+
        if (!pcie->supports_clkreq) {
                val = appl_readl(pcie, APPL_PINMUX);
                val |= APPL_PINMUX_CLKREQ_OVERRIDE_EN;
 fail_reg_en:
        tegra_pcie_disable_slot_regulators(pcie);
 fail_slot_reg_en:
+       if (pcie->enable_ext_refclk)
+               tegra_pcie_bpmp_set_pll_state(pcie, false);
+fail_pll_init:
        tegra_pcie_bpmp_set_ctrl_state(pcie, false);
 
        return ret;
 
        tegra_pcie_disable_slot_regulators(pcie);
 
+       if (pcie->enable_ext_refclk) {
+               ret = tegra_pcie_bpmp_set_pll_state(pcie, false);
+               if (ret)
+                       dev_err(pcie->dev, "Failed to deinit UPHY: %d\n", ret);
+       }
+
        ret = tegra_pcie_bpmp_set_ctrl_state(pcie, false);
        if (ret)
                dev_err(pcie->dev, "Failed to disable controller %d: %d\n",
 
        pm_runtime_put_sync(pcie->dev);
 
+       if (pcie->enable_ext_refclk) {
+               ret = tegra_pcie_bpmp_set_pll_state(pcie, false);
+               if (ret)
+                       dev_err(pcie->dev, "Failed to turn off UPHY: %d\n",
+                               ret);
+       }
+
        ret = tegra_pcie_bpmp_set_pll_state(pcie, false);
        if (ret)
                dev_err(pcie->dev, "Failed to turn off UPHY: %d\n", ret);
                goto fail_set_ctrl_state;
        }
 
-       ret = tegra_pcie_bpmp_set_pll_state(pcie, true);
-       if (ret) {
-               dev_err(dev, "Failed to init UPHY for PCIe EP: %d\n", ret);
-               goto fail_pll_init;
+       if (pcie->enable_ext_refclk) {
+               ret = tegra_pcie_bpmp_set_pll_state(pcie, true);
+               if (ret) {
+                       dev_err(dev, "Failed to init UPHY for PCIe EP: %d\n",
+                               ret);
+                       goto fail_pll_init;
+               }
        }
 
        ret = clk_prepare_enable(pcie->core_clk);
                disable_aspm_l12(pcie);
        }
 
-       val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
-       val &= ~GEN3_RELATED_OFF_GEN3_ZRXDC_NONCOMPL;
-       dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
+       if (!pcie->of_data->has_l1ss_exit_fix) {
+               val = dw_pcie_readl_dbi(pci, GEN3_RELATED_OFF);
+               val &= ~GEN3_RELATED_OFF_GEN3_ZRXDC_NONCOMPL;
+               dw_pcie_writel_dbi(pci, GEN3_RELATED_OFF, val);
+       }
 
        pcie->pcie_cap_base = dw_pcie_find_capability(&pcie->pci,
                                                      PCI_CAP_ID_EXP);
        val_16 |= PCI_EXP_DEVCTL_PAYLOAD_256B;
        dw_pcie_writew_dbi(pci, pcie->pcie_cap_base + PCI_EXP_DEVCTL, val_16);
 
+       /* Clear Slot Clock Configuration bit if SRNS configuration */
+       if (pcie->enable_srns) {
+               val_16 = dw_pcie_readw_dbi(pci, pcie->pcie_cap_base +
+                                          PCI_EXP_LNKSTA);
+               val_16 &= ~PCI_EXP_LNKSTA_SLC;
+               dw_pcie_writew_dbi(pci, pcie->pcie_cap_base + PCI_EXP_LNKSTA,
+                                  val_16);
+       }
+
        clk_set_rate(pcie->core_clk, GEN4_CORE_CLK_FREQ);
 
        val = (ep->msi_mem_phys & MSIX_ADDR_MATCH_LOW_OFF_MASK);
 
        dw_pcie_ep_init_notify(ep);
 
+       /* Program the private control to allow sending LTR upstream */
+       if (pcie->of_data->has_ltr_req_fix) {
+               val = appl_readl(pcie, APPL_LTR_MSG_2);
+               val |= APPL_LTR_MSG_2_LTR_MSG_REQ_STATE;
+               appl_writel(pcie, val, APPL_LTR_MSG_2);
+       }
+
        /* Enable LTSSM */
        val = appl_readl(pcie, APPL_CTRL);
        val |= APPL_CTRL_LTSSM_EN;
        pci = &pcie->pci;
        pci->dev = &pdev->dev;
        pci->ops = &tegra_dw_pcie_ops;
-       pci->n_fts[0] = N_FTS_VAL;
-       pci->n_fts[1] = FTS_VAL;
-       pci->version = 0x490A;
-
+       pcie->dev = &pdev->dev;
+       pcie->of_data = (struct tegra_pcie_dw_of_data *)data;
+       pci->n_fts[0] = pcie->of_data->n_fts[0];
+       pci->n_fts[1] = pcie->of_data->n_fts[1];
+       pci->version = pcie->of_data->version;
        pp = &pci->pp;
        pp->num_vectors = MAX_MSI_IRQS;
-       pcie->dev = &pdev->dev;
-       pcie->mode = (enum dw_pcie_device_mode)data->mode;
 
        ret = tegra_pcie_dw_parse_dt(pcie);
        if (ret < 0) {
 
        platform_set_drvdata(pdev, pcie);
 
-       switch (pcie->mode) {
+       switch (pcie->of_data->mode) {
        case DW_PCIE_RC_TYPE:
                ret = devm_request_irq(dev, pp->irq, tegra_pcie_rp_irq_handler,
                                       IRQF_SHARED, "tegra-pcie-intr", pcie);
                break;
 
        default:
-               dev_err(dev, "Invalid PCIe device type %d\n", pcie->mode);
+               dev_err(dev, "Invalid PCIe device type %d\n",
+                       pcie->of_data->mode);
        }
 
 fail:
 {
        struct tegra_pcie_dw *pcie = platform_get_drvdata(pdev);
 
-       if (pcie->mode == DW_PCIE_RC_TYPE) {
+       if (pcie->of_data->mode == DW_PCIE_RC_TYPE) {
                if (!pcie->link_state)
                        return 0;
 
        struct tegra_pcie_dw *pcie = dev_get_drvdata(dev);
        u32 val;
 
-       if (pcie->mode == DW_PCIE_EP_TYPE) {
+       if (pcie->of_data->mode == DW_PCIE_EP_TYPE) {
                dev_err(dev, "Failed to Suspend as Tegra PCIe is in EP mode\n");
                return -EPERM;
        }
                return 0;
 
        /* Enable HW_HOT_RST mode */
-       val = appl_readl(pcie, APPL_CTRL);
-       val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
-                APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
-       val |= APPL_CTRL_HW_HOT_RST_EN;
-       appl_writel(pcie, val, APPL_CTRL);
+       if (!pcie->of_data->has_sbr_reset_fix) {
+               val = appl_readl(pcie, APPL_CTRL);
+               val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
+                        APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
+               val |= APPL_CTRL_HW_HOT_RST_EN;
+               appl_writel(pcie, val, APPL_CTRL);
+       }
 
        return 0;
 }
        struct tegra_pcie_dw *pcie = dev_get_drvdata(dev);
        u32 val;
 
-       if (pcie->mode == DW_PCIE_EP_TYPE) {
+       if (pcie->of_data->mode == DW_PCIE_EP_TYPE) {
                dev_err(dev, "Suspend is not supported in EP mode");
                return -ENOTSUPP;
        }
                return 0;
 
        /* Disable HW_HOT_RST mode */
-       val = appl_readl(pcie, APPL_CTRL);
-       val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
-                APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
-       val |= APPL_CTRL_HW_HOT_RST_MODE_IMDT_RST <<
-              APPL_CTRL_HW_HOT_RST_MODE_SHIFT;
-       val &= ~APPL_CTRL_HW_HOT_RST_EN;
-       appl_writel(pcie, val, APPL_CTRL);
+       if (!pcie->of_data->has_sbr_reset_fix) {
+               val = appl_readl(pcie, APPL_CTRL);
+               val &= ~(APPL_CTRL_HW_HOT_RST_MODE_MASK <<
+                        APPL_CTRL_HW_HOT_RST_MODE_SHIFT);
+               val |= APPL_CTRL_HW_HOT_RST_MODE_IMDT_RST <<
+                      APPL_CTRL_HW_HOT_RST_MODE_SHIFT;
+               val &= ~APPL_CTRL_HW_HOT_RST_EN;
+               appl_writel(pcie, val, APPL_CTRL);
+       }
 
        return 0;
 }
 {
        struct tegra_pcie_dw *pcie = platform_get_drvdata(pdev);
 
-       if (pcie->mode == DW_PCIE_RC_TYPE) {
+       if (pcie->of_data->mode == DW_PCIE_RC_TYPE) {
                if (!pcie->link_state)
                        return;
 
        }
 }
 
-static const struct tegra_pcie_dw_of_data tegra_pcie_dw_rc_of_data = {
+static const struct tegra_pcie_dw_of_data tegra194_pcie_dw_rc_of_data = {
+       .version = TEGRA194_DWC_IP_VER,
        .mode = DW_PCIE_RC_TYPE,
+       .cdm_chk_int_en_bit = BIT(19),
+       /* Gen4 - 5, 6, 8 and 9 presets enabled */
+       .gen4_preset_vec = 0x360,
+       .n_fts = { 52, 52 },
 };
 
-static const struct tegra_pcie_dw_of_data tegra_pcie_dw_ep_of_data = {
+static const struct tegra_pcie_dw_of_data tegra194_pcie_dw_ep_of_data = {
+       .version = TEGRA194_DWC_IP_VER,
        .mode = DW_PCIE_EP_TYPE,
+       .cdm_chk_int_en_bit = BIT(19),
+       /* Gen4 - 5, 6, 8 and 9 presets enabled */
+       .gen4_preset_vec = 0x360,
+       .n_fts = { 52, 52 },
+};
+
+static const struct tegra_pcie_dw_of_data tegra234_pcie_dw_rc_of_data = {
+       .version = TEGRA234_DWC_IP_VER,
+       .mode = DW_PCIE_RC_TYPE,
+       .has_msix_doorbell_access_fix = true,
+       .has_sbr_reset_fix = true,
+       .has_l1ss_exit_fix = true,
+       .cdm_chk_int_en_bit = BIT(18),
+       /* Gen4 - 6, 8 and 9 presets enabled */
+       .gen4_preset_vec = 0x340,
+       .n_fts = { 52, 80 },
+};
+
+static const struct tegra_pcie_dw_of_data tegra234_pcie_dw_ep_of_data = {
+       .version = TEGRA234_DWC_IP_VER,
+       .mode = DW_PCIE_EP_TYPE,
+       .has_l1ss_exit_fix = true,
+       .has_ltr_req_fix = true,
+       .cdm_chk_int_en_bit = BIT(18),
+       /* Gen4 - 6, 8 and 9 presets enabled */
+       .gen4_preset_vec = 0x340,
+       .n_fts = { 52, 80 },
 };
 
 static const struct of_device_id tegra_pcie_dw_of_match[] = {
        {
                .compatible = "nvidia,tegra194-pcie",
-               .data = &tegra_pcie_dw_rc_of_data,
+               .data = &tegra194_pcie_dw_rc_of_data,
        },
        {
                .compatible = "nvidia,tegra194-pcie-ep",
-               .data = &tegra_pcie_dw_ep_of_data,
+               .data = &tegra194_pcie_dw_ep_of_data,
+       },
+       {
+               .compatible = "nvidia,tegra234-pcie",
+               .data = &tegra234_pcie_dw_rc_of_data,
+       },
+       {
+               .compatible = "nvidia,tegra234-pcie-ep",
+               .data = &tegra234_pcie_dw_ep_of_data,
        },
-       {},
+       {}
 };
 
 static const struct dev_pm_ops tegra_pcie_dw_pm_ops = {