diff mbox series

[06/16] phy: ti: j721e-wiz: Add support for WIZ module present in TI J721E SoC

Message ID 20210309122748.31842-7-kishon@ti.com
State New
Headers show
Series TI/Cadence: Add Sierra/Torrent SERDES driver | expand

Commit Message

Kishon Vijay Abraham I March 9, 2021, 12:27 p.m. UTC
From: Jean-Jacques Hiblot <jjhiblot@ti.com>


Add support for WIZ module present in TI's J721E SoC. WIZ is a SERDES
wrapper used to configure some of the input signals to the SERDES. It is
used with both Sierra(16G) and Torrent(10G) SERDES. This driver configures
three clock selects (pll0, pll1, dig) and supports resets for each of the
lanes.

This is an adaptation of the linux driver.

Signed-off-by: Jean-Jacques Hiblot <jjhiblot@ti.com>

Signed-off-by: Vignesh Raghavendra <vigneshr@ti.com>

Signed-off-by: Kishon Vijay Abraham I <kishon@ti.com>

---
 drivers/phy/Kconfig            |   1 +
 drivers/phy/Makefile           |   1 +
 drivers/phy/ti/Kconfig         |   9 +
 drivers/phy/ti/Makefile        |   1 +
 drivers/phy/ti/phy-j721e-wiz.c | 985 +++++++++++++++++++++++++++++++++
 5 files changed, 997 insertions(+)
 create mode 100644 drivers/phy/ti/Kconfig
 create mode 100644 drivers/phy/ti/Makefile
 create mode 100644 drivers/phy/ti/phy-j721e-wiz.c

-- 
2.17.1
diff mbox series

Patch

diff --git a/drivers/phy/Kconfig b/drivers/phy/Kconfig
index 9208e430a6..7821161e3c 100644
--- a/drivers/phy/Kconfig
+++ b/drivers/phy/Kconfig
@@ -270,5 +270,6 @@  config PHY_MTK_TPHY
 
 source "drivers/phy/rockchip/Kconfig"
 source "drivers/phy/cadence/Kconfig"
+source "drivers/phy/ti/Kconfig"
 
 endmenu
diff --git a/drivers/phy/Makefile b/drivers/phy/Makefile
index 4736c5eadb..2efaa8827b 100644
--- a/drivers/phy/Makefile
+++ b/drivers/phy/Makefile
@@ -32,3 +32,4 @@  obj-$(CONFIG_MT76X8_USB_PHY) += mt76x8-usb-phy.o
 obj-$(CONFIG_PHY_DA8XX_USB) += phy-da8xx-usb.o
 obj-$(CONFIG_PHY_MTK_TPHY) += phy-mtk-tphy.o
 obj-y += cadence/
+obj-y += ti/
diff --git a/drivers/phy/ti/Kconfig b/drivers/phy/ti/Kconfig
new file mode 100644
index 0000000000..111085f235
--- /dev/null
+++ b/drivers/phy/ti/Kconfig
@@ -0,0 +1,9 @@ 
+config PHY_J721E_WIZ
+	tristate "TI J721E WIZ (SERDES Wrapper) support"
+	depends on ARCH_K3
+	help
+	  This option enables support for WIZ module present in TI's J721E
+	  SoC. WIZ is a serdes wrapper used to configure some of the input
+	  signals to the SERDES (Sierra/Torrent). This driver configures
+	  three clock selects (pll0, pll1, dig) and resets for each of the
+	  lanes.
diff --git a/drivers/phy/ti/Makefile b/drivers/phy/ti/Makefile
new file mode 100644
index 0000000000..873ddbf036
--- /dev/null
+++ b/drivers/phy/ti/Makefile
@@ -0,0 +1 @@ 
+obj-$(CONFIG_$(SPL_)PHY_J721E_WIZ)	+= phy-j721e-wiz.o
diff --git a/drivers/phy/ti/phy-j721e-wiz.c b/drivers/phy/ti/phy-j721e-wiz.c
new file mode 100644
index 0000000000..39384f87b8
--- /dev/null
+++ b/drivers/phy/ti/phy-j721e-wiz.c
@@ -0,0 +1,985 @@ 
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Copyright (C) 2017-2018 Texas Instruments Incorporated - http://www.ti.com/
+ * Jean-Jacques Hiblot <jjhiblot@ti.com>
+ */
+
+#include <common.h>
+#include <clk-uclass.h>
+#include <dm.h>
+#include <dm/device_compat.h>
+#include <asm/gpio.h>
+#include <dm/lists.h>
+#include <dm/device-internal.h>
+#include <regmap.h>
+#include <reset-uclass.h>
+#include <dt-bindings/phy/phy.h>
+
+#define WIZ_MAX_LANES		4
+#define WIZ_MUX_NUM_CLOCKS	3
+#define WIZ_DIV_NUM_CLOCKS_16G	2
+#define WIZ_DIV_NUM_CLOCKS_10G	1
+
+#define WIZ_SERDES_CTRL		0x404
+#define WIZ_SERDES_TOP_CTRL	0x408
+#define WIZ_SERDES_RST		0x40c
+#define WIZ_SERDES_TYPEC	0x410
+#define WIZ_LANECTL(n)		(0x480 + (0x40 * (n)))
+#define WIZ_LANEDIV(n)		(0x484 + (0x40 * (n)))
+
+#define WIZ_MAX_LANES		4
+#define WIZ_MUX_NUM_CLOCKS	3
+#define WIZ_DIV_NUM_CLOCKS_16G	2
+#define WIZ_DIV_NUM_CLOCKS_10G	1
+
+#define WIZ_SERDES_TYPEC_LN10_SWAP	BIT(30)
+
+enum wiz_lane_standard_mode {
+	LANE_MODE_GEN1,
+	LANE_MODE_GEN2,
+	LANE_MODE_GEN3,
+	LANE_MODE_GEN4,
+};
+
+enum wiz_refclk_mux_sel {
+	PLL0_REFCLK,
+	PLL1_REFCLK,
+	REFCLK_DIG,
+};
+
+enum wiz_refclk_div_sel {
+	CMN_REFCLK,
+	CMN_REFCLK1,
+};
+
+static const struct reg_field por_en = REG_FIELD(WIZ_SERDES_CTRL, 31, 31);
+static const struct reg_field phy_reset_n = REG_FIELD(WIZ_SERDES_RST, 31, 31);
+static const struct reg_field pll1_refclk_mux_sel =
+					REG_FIELD(WIZ_SERDES_RST, 29, 29);
+static const struct reg_field pll0_refclk_mux_sel =
+					REG_FIELD(WIZ_SERDES_RST, 28, 28);
+static const struct reg_field refclk_dig_sel_16g =
+					REG_FIELD(WIZ_SERDES_RST, 24, 25);
+static const struct reg_field refclk_dig_sel_10g =
+					REG_FIELD(WIZ_SERDES_RST, 24, 24);
+static const struct reg_field pma_cmn_refclk_int_mode =
+					REG_FIELD(WIZ_SERDES_TOP_CTRL, 28, 29);
+static const struct reg_field pma_cmn_refclk_mode =
+					REG_FIELD(WIZ_SERDES_TOP_CTRL, 30, 31);
+static const struct reg_field pma_cmn_refclk_dig_div =
+					REG_FIELD(WIZ_SERDES_TOP_CTRL, 26, 27);
+static const struct reg_field pma_cmn_refclk1_dig_div =
+					REG_FIELD(WIZ_SERDES_TOP_CTRL, 24, 25);
+
+static const struct reg_field p_enable[WIZ_MAX_LANES] = {
+	REG_FIELD(WIZ_LANECTL(0), 30, 31),
+	REG_FIELD(WIZ_LANECTL(1), 30, 31),
+	REG_FIELD(WIZ_LANECTL(2), 30, 31),
+	REG_FIELD(WIZ_LANECTL(3), 30, 31),
+};
+
+enum p_enable { P_ENABLE = 2, P_ENABLE_FORCE = 1, P_ENABLE_DISABLE = 0 };
+
+static const struct reg_field p_align[WIZ_MAX_LANES] = {
+	REG_FIELD(WIZ_LANECTL(0), 29, 29),
+	REG_FIELD(WIZ_LANECTL(1), 29, 29),
+	REG_FIELD(WIZ_LANECTL(2), 29, 29),
+	REG_FIELD(WIZ_LANECTL(3), 29, 29),
+};
+
+static const struct reg_field p_raw_auto_start[WIZ_MAX_LANES] = {
+	REG_FIELD(WIZ_LANECTL(0), 28, 28),
+	REG_FIELD(WIZ_LANECTL(1), 28, 28),
+	REG_FIELD(WIZ_LANECTL(2), 28, 28),
+	REG_FIELD(WIZ_LANECTL(3), 28, 28),
+};
+
+static const struct reg_field p_standard_mode[WIZ_MAX_LANES] = {
+	REG_FIELD(WIZ_LANECTL(0), 24, 25),
+	REG_FIELD(WIZ_LANECTL(1), 24, 25),
+	REG_FIELD(WIZ_LANECTL(2), 24, 25),
+	REG_FIELD(WIZ_LANECTL(3), 24, 25),
+};
+
+static const struct reg_field p_mac_div_sel0[WIZ_MAX_LANES] = {
+	REG_FIELD(WIZ_LANEDIV(0), 16, 22),
+	REG_FIELD(WIZ_LANEDIV(1), 16, 22),
+	REG_FIELD(WIZ_LANEDIV(2), 16, 22),
+	REG_FIELD(WIZ_LANEDIV(3), 16, 22),
+};
+
+static const struct reg_field p_mac_div_sel1[WIZ_MAX_LANES] = {
+	REG_FIELD(WIZ_LANEDIV(0), 0, 8),
+	REG_FIELD(WIZ_LANEDIV(1), 0, 8),
+	REG_FIELD(WIZ_LANEDIV(2), 0, 8),
+	REG_FIELD(WIZ_LANEDIV(3), 0, 8),
+};
+
+struct wiz_clk_mux_sel {
+	enum wiz_refclk_mux_sel mux_sel;
+	u32			table[4];
+	const char		*node_name;
+};
+
+struct wiz_clk_div_sel {
+	enum wiz_refclk_div_sel div_sel;
+	const char		*node_name;
+};
+
+static struct wiz_clk_mux_sel clk_mux_sel_16g[] = {
+	{
+		/*
+		 * Mux value to be configured for each of the input clocks
+		 * in the order populated in device tree
+		 */
+		.mux_sel = PLL0_REFCLK,
+		.table = { 1, 0 },
+		.node_name = "pll0-refclk",
+	},
+	{
+		.mux_sel = PLL1_REFCLK,
+		.table = { 1, 0 },
+		.node_name = "pll1-refclk",
+	},
+	{
+		.mux_sel = REFCLK_DIG,
+		.table = { 1, 3, 0, 2 },
+		.node_name = "refclk-dig",
+	},
+};
+
+static struct wiz_clk_mux_sel clk_mux_sel_10g[] = {
+	{
+		/*
+		 * Mux value to be configured for each of the input clocks
+		 * in the order populated in device tree
+		 */
+		.mux_sel = PLL0_REFCLK,
+		.table = { 1, 0 },
+		.node_name = "pll0-refclk",
+	},
+	{
+		.mux_sel = PLL1_REFCLK,
+		.table = { 1, 0 },
+		.node_name = "pll1-refclk",
+	},
+	{
+		.mux_sel = REFCLK_DIG,
+		.table = { 1, 0 },
+		.node_name = "refclk-dig",
+	},
+};
+
+static struct wiz_clk_div_sel clk_div_sel[] = {
+	{
+		.div_sel = CMN_REFCLK,
+		.node_name = "cmn-refclk-dig-div",
+	},
+	{
+		.div_sel = CMN_REFCLK1,
+		.node_name = "cmn-refclk1-dig-div",
+	},
+};
+
+enum wiz_type {
+	J721E_WIZ_16G,
+	J721E_WIZ_10G,
+	AM64_WIZ_10G,
+};
+
+#define WIZ_TYPEC_DIR_DEBOUNCE_MIN	100	/* ms */
+#define WIZ_TYPEC_DIR_DEBOUNCE_MAX	1000
+
+struct wiz {
+	struct regmap		*regmap;
+	enum wiz_type		type;
+	struct regmap_field	*por_en;
+	struct regmap_field	*phy_reset_n;
+	struct regmap_field	*phy_en_refclk;
+	struct regmap_field	*p_enable[WIZ_MAX_LANES];
+	struct regmap_field	*p_align[WIZ_MAX_LANES];
+	struct regmap_field	*p_raw_auto_start[WIZ_MAX_LANES];
+	struct regmap_field	*p_standard_mode[WIZ_MAX_LANES];
+	struct regmap_field	*p_mac_div_sel0[WIZ_MAX_LANES];
+	struct regmap_field	*p_mac_div_sel1[WIZ_MAX_LANES];
+	struct regmap_field	*pma_cmn_refclk_int_mode;
+	struct regmap_field	*pma_cmn_refclk_mode;
+	struct regmap_field	*pma_cmn_refclk_dig_div;
+	struct regmap_field	*pma_cmn_refclk1_dig_div;
+	struct regmap_field	*clk_div_sel[WIZ_DIV_NUM_CLOCKS_16G];
+	struct regmap_field	*clk_mux_sel[WIZ_MUX_NUM_CLOCKS];
+
+	struct udevice		*dev;
+	u32			num_lanes;
+	struct gpio_desc	*gpio_typec_dir;
+	u32			lane_phy_type[WIZ_MAX_LANES];
+};
+
+struct wiz_div_clk {
+	struct clk parent_clk;
+	struct wiz *wiz;
+};
+
+struct wiz_mux_clk {
+	struct clk parent_clks[4];
+	struct wiz *wiz;
+};
+
+struct wiz_reset {
+	struct wiz *wiz;
+};
+
+static ulong wiz_div_clk_get_rate(struct clk *clk)
+{
+	struct udevice *dev = clk->dev;
+	struct wiz_div_clk *priv = dev_get_priv(dev);
+	struct wiz_clk_div_sel *data = dev_get_plat(dev);
+	struct wiz *wiz = priv->wiz;
+	ulong parent_rate = clk_get_rate(&priv->parent_clk);
+	u32 val;
+
+	regmap_field_read(wiz->clk_div_sel[data->div_sel], &val);
+
+	return parent_rate >> val;
+}
+
+static ulong wiz_div_clk_set_rate(struct clk *clk, ulong rate)
+{
+	struct udevice *dev = clk->dev;
+	struct wiz_div_clk *priv = dev_get_priv(dev);
+	struct wiz_clk_div_sel *data = dev_get_plat(dev);
+	struct wiz *wiz = priv->wiz;
+	ulong parent_rate = clk_get_rate(&priv->parent_clk);
+	u32 div = parent_rate / rate;
+
+	div = __ffs(div);
+	regmap_field_write(wiz->clk_div_sel[data->div_sel], div);
+
+	return parent_rate >> div;
+}
+
+const struct clk_ops wiz_div_clk_ops = {
+	.get_rate = wiz_div_clk_get_rate,
+	.set_rate = wiz_div_clk_set_rate,
+};
+
+int wiz_div_clk_probe(struct udevice *dev)
+{
+	struct wiz_div_clk *priv = dev_get_priv(dev);
+	struct clk parent_clk;
+	int rc;
+
+	rc = clk_get_by_index(dev, 0, &parent_clk);
+	if (rc) {
+		dev_err(dev, "unable to get parent clock. ret %d\n", rc);
+		return rc;
+	}
+	priv->parent_clk = parent_clk;
+	priv->wiz = dev_get_priv(dev->parent);
+	return 0;
+}
+
+U_BOOT_DRIVER(wiz_div_clk) = {
+	.name		= "wiz_div_clk",
+	.id		= UCLASS_CLK,
+	.priv_auto	= sizeof(struct wiz_div_clk),
+	.ops		= &wiz_div_clk_ops,
+	.probe		= wiz_div_clk_probe,
+};
+
+static int clk_mux_val_to_index(u32 table[4], unsigned int val)
+{
+	int i;
+
+	for (i = 0; i < 4; i++)
+		if (table[i] == val)
+			return i;
+	return -EINVAL;
+}
+
+static ulong wiz_clk_mux_get_rate(struct clk *clk)
+{
+	struct udevice *dev = clk->dev;
+	struct wiz_mux_clk *priv = dev_get_priv(dev);
+	struct wiz_clk_mux_sel *data = dev_get_plat(dev);
+	struct wiz *wiz = priv->wiz;
+	unsigned int val, idx;
+
+	regmap_field_read(wiz->clk_mux_sel[data->mux_sel], &val);
+	idx = clk_mux_val_to_index(data->table, val);
+	if (priv->parent_clks[idx].dev)
+		return clk_get_rate(&priv->parent_clks[idx]);
+
+	return 0;
+}
+
+static int wiz_clk_mux_set_parent(struct clk *clk,  struct clk *parent)
+{
+	struct udevice *dev = clk->dev;
+	struct wiz_mux_clk *priv = dev_get_priv(dev);
+	struct wiz_clk_mux_sel *data = dev_get_plat(dev);
+	struct wiz *wiz = priv->wiz;
+	int i;
+
+	for (i = 0; i < ARRAY_SIZE(priv->parent_clks); i++)
+		if (parent->dev == priv->parent_clks[i].dev)
+			break;
+
+	if (i == ARRAY_SIZE(priv->parent_clks))
+		return -EINVAL;
+
+	regmap_field_write(wiz->clk_mux_sel[data->mux_sel], data->table[i]);
+	return clk_get_rate(parent);
+}
+
+static const struct clk_ops wiz_clk_mux_ops = {
+	.set_parent = wiz_clk_mux_set_parent,
+	.get_rate = wiz_clk_mux_get_rate,
+};
+
+int wiz_mux_clk_probe(struct udevice *dev)
+{
+	struct wiz_mux_clk *priv = dev_get_priv(dev);
+	int rc;
+	int i;
+
+	for (i = 0; i < ARRAY_SIZE(priv->parent_clks); i++) {
+		rc = clk_get_by_index(dev, i, &priv->parent_clks[i]);
+		if (rc)
+			priv->parent_clks[i].dev = NULL;
+	}
+	priv->wiz = dev_get_priv(dev->parent);
+	return 0;
+}
+
+U_BOOT_DRIVER(wiz_mux_clk) = {
+	.name		= "wiz_mux_clk",
+	.id		= UCLASS_CLK,
+	.priv_auto	= sizeof(struct wiz_mux_clk),
+	.ops		= &wiz_clk_mux_ops,
+	.probe		= wiz_mux_clk_probe,
+};
+
+static int wiz_reset_request(struct reset_ctl *reset_ctl)
+{
+	return 0;
+}
+
+static int wiz_reset_free(struct reset_ctl *reset_ctl)
+{
+	return 0;
+}
+
+static int wiz_reset_assert(struct reset_ctl *reset_ctl)
+{
+	struct wiz_reset *priv = dev_get_priv(reset_ctl->dev);
+	struct wiz *wiz = priv->wiz;
+	int ret;
+	int id = reset_ctl->id;
+
+	if (id == 0) {
+		ret = regmap_field_write(wiz->phy_reset_n, false);
+		return ret;
+	}
+
+	ret = regmap_field_write(wiz->p_enable[id - 1], P_ENABLE_DISABLE);
+	return ret;
+}
+
+static int wiz_reset_deassert(struct reset_ctl *reset_ctl)
+{
+	struct wiz_reset *priv = dev_get_priv(reset_ctl->dev);
+	struct wiz *wiz = priv->wiz;
+	int ret;
+	int id = reset_ctl->id;
+
+	/* if typec-dir gpio was specified, set LN10 SWAP bit based on that */
+	if (id == 0 && wiz->gpio_typec_dir) {
+		if (dm_gpio_get_value(wiz->gpio_typec_dir)) {
+			regmap_update_bits(wiz->regmap, WIZ_SERDES_TYPEC,
+					   WIZ_SERDES_TYPEC_LN10_SWAP,
+					   WIZ_SERDES_TYPEC_LN10_SWAP);
+		} else {
+			regmap_update_bits(wiz->regmap, WIZ_SERDES_TYPEC,
+					   WIZ_SERDES_TYPEC_LN10_SWAP, 0);
+		}
+	}
+
+	if (id == 0) {
+		ret = regmap_field_write(wiz->phy_reset_n, true);
+		return ret;
+	}
+
+	if (wiz->lane_phy_type[id - 1] == PHY_TYPE_PCIE)
+		ret = regmap_field_write(wiz->p_enable[id - 1], P_ENABLE);
+	else
+		ret = regmap_field_write(wiz->p_enable[id - 1], P_ENABLE_FORCE);
+
+	return ret;
+}
+
+static struct reset_ops wiz_reset_ops = {
+	.request = wiz_reset_request,
+	.rfree = wiz_reset_free,
+	.rst_assert = wiz_reset_assert,
+	.rst_deassert = wiz_reset_deassert,
+};
+
+int wiz_reset_probe(struct udevice *dev)
+{
+	struct wiz_reset *priv = dev_get_priv(dev);
+
+	priv->wiz = dev_get_priv(dev->parent);
+
+	return 0;
+}
+
+U_BOOT_DRIVER(wiz_reset) = {
+	.name = "wiz-reset",
+	.id = UCLASS_RESET,
+	.probe = wiz_reset_probe,
+	.ops = &wiz_reset_ops,
+};
+
+static int wiz_reset(struct wiz *wiz)
+{
+	int ret;
+
+	ret = regmap_field_write(wiz->por_en, 0x1);
+	if (ret)
+		return ret;
+
+	mdelay(1);
+
+	ret = regmap_field_write(wiz->por_en, 0x0);
+	if (ret)
+		return ret;
+
+	return 0;
+}
+
+static int wiz_p_mac_div_sel(struct wiz *wiz)
+{
+	u32 num_lanes = wiz->num_lanes;
+	int ret;
+	int i;
+
+	for (i = 0; i < num_lanes; i++) {
+		if (wiz->lane_phy_type[i] == PHY_TYPE_QSGMII) {
+			ret = regmap_field_write(wiz->p_mac_div_sel0[i], 1);
+			if (ret)
+				return ret;
+
+			ret = regmap_field_write(wiz->p_mac_div_sel1[i], 2);
+			if (ret)
+				return ret;
+		}
+	}
+
+	return 0;
+}
+
+static int wiz_mode_select(struct wiz *wiz)
+{
+	u32 num_lanes = wiz->num_lanes;
+	int ret;
+	int i;
+
+	for (i = 0; i < num_lanes; i++) {
+		if (wiz->lane_phy_type[i] == PHY_TYPE_QSGMII) {
+			ret = regmap_field_write(wiz->p_standard_mode[i],
+						 LANE_MODE_GEN2);
+			if (ret)
+				return ret;
+		}
+	}
+
+	return 0;
+}
+
+static int wiz_init_raw_interface(struct wiz *wiz, bool enable)
+{
+	u32 num_lanes = wiz->num_lanes;
+	int i;
+	int ret;
+
+	for (i = 0; i < num_lanes; i++) {
+		ret = regmap_field_write(wiz->p_align[i], enable);
+		if (ret)
+			return ret;
+
+		ret = regmap_field_write(wiz->p_raw_auto_start[i], enable);
+		if (ret)
+			return ret;
+	}
+
+	return 0;
+}
+
+static int wiz_init(struct wiz *wiz)
+{
+	struct udevice *dev = wiz->dev;
+	int ret;
+
+	ret = wiz_reset(wiz);
+	if (ret) {
+		dev_err(dev, "WIZ reset failed\n");
+		return ret;
+	}
+
+	ret = wiz_mode_select(wiz);
+	if (ret) {
+		dev_err(dev, "WIZ mode select failed\n");
+		return ret;
+	}
+
+	ret = wiz_p_mac_div_sel(wiz);
+	if (ret) {
+		dev_err(dev, "Configuring P0 MAC DIV SEL failed\n");
+		return ret;
+	}
+
+	ret = wiz_init_raw_interface(wiz, true);
+	if (ret) {
+		dev_err(dev, "WIZ interface initialization failed\n");
+		return ret;
+	}
+
+	return 0;
+}
+
+static int wiz_regfield_init(struct wiz *wiz)
+{
+	struct regmap *regmap = wiz->regmap;
+	int num_lanes = wiz->num_lanes;
+	struct udevice *dev = wiz->dev;
+	enum wiz_type type;
+	int i;
+
+	wiz->por_en = devm_regmap_field_alloc(dev, regmap, por_en);
+	if (IS_ERR(wiz->por_en)) {
+		dev_err(dev, "POR_EN reg field init failed\n");
+		return PTR_ERR(wiz->por_en);
+	}
+
+	wiz->phy_reset_n = devm_regmap_field_alloc(dev, regmap,
+						   phy_reset_n);
+	if (IS_ERR(wiz->phy_reset_n)) {
+		dev_err(dev, "PHY_RESET_N reg field init failed\n");
+		return PTR_ERR(wiz->phy_reset_n);
+	}
+
+	wiz->pma_cmn_refclk_int_mode =
+		devm_regmap_field_alloc(dev, regmap, pma_cmn_refclk_int_mode);
+	if (IS_ERR(wiz->pma_cmn_refclk_int_mode)) {
+		dev_err(dev, "PMA_CMN_REFCLK_INT_MODE reg field init failed\n");
+		return PTR_ERR(wiz->pma_cmn_refclk_int_mode);
+	}
+
+	wiz->pma_cmn_refclk_mode =
+		devm_regmap_field_alloc(dev, regmap, pma_cmn_refclk_mode);
+	if (IS_ERR(wiz->pma_cmn_refclk_mode)) {
+		dev_err(dev, "PMA_CMN_REFCLK_MODE reg field init failed\n");
+		return PTR_ERR(wiz->pma_cmn_refclk_mode);
+	}
+
+	wiz->clk_div_sel[CMN_REFCLK] =
+		devm_regmap_field_alloc(dev, regmap, pma_cmn_refclk_dig_div);
+	if (IS_ERR(wiz->clk_div_sel[CMN_REFCLK])) {
+		dev_err(dev, "PMA_CMN_REFCLK_DIG_DIV reg field init failed\n");
+		return PTR_ERR(wiz->clk_div_sel[CMN_REFCLK]);
+	}
+
+	wiz->clk_div_sel[CMN_REFCLK1] =
+		devm_regmap_field_alloc(dev, regmap, pma_cmn_refclk1_dig_div);
+	if (IS_ERR(wiz->clk_div_sel[CMN_REFCLK1])) {
+		dev_err(dev, "PMA_CMN_REFCLK1_DIG_DIV reg field init failed\n");
+		return PTR_ERR(wiz->clk_div_sel[CMN_REFCLK1]);
+	}
+
+	wiz->clk_mux_sel[PLL0_REFCLK] =
+		devm_regmap_field_alloc(dev, regmap, pll0_refclk_mux_sel);
+	if (IS_ERR(wiz->clk_mux_sel[PLL0_REFCLK])) {
+		dev_err(dev, "PLL0_REFCLK_SEL reg field init failed\n");
+		return PTR_ERR(wiz->clk_mux_sel[PLL0_REFCLK]);
+	}
+
+	wiz->clk_mux_sel[PLL1_REFCLK] =
+		devm_regmap_field_alloc(dev, regmap, pll1_refclk_mux_sel);
+	if (IS_ERR(wiz->clk_mux_sel[PLL1_REFCLK])) {
+		dev_err(dev, "PLL1_REFCLK_SEL reg field init failed\n");
+		return PTR_ERR(wiz->clk_mux_sel[PLL1_REFCLK]);
+	}
+
+	type = dev_get_driver_data(dev);
+	if (type == J721E_WIZ_10G || type == AM64_WIZ_10G)
+		wiz->clk_mux_sel[REFCLK_DIG] =
+			devm_regmap_field_alloc(dev, regmap,
+						refclk_dig_sel_10g);
+	else
+		wiz->clk_mux_sel[REFCLK_DIG] =
+			devm_regmap_field_alloc(dev, regmap,
+						refclk_dig_sel_16g);
+	if (IS_ERR(wiz->clk_mux_sel[REFCLK_DIG])) {
+		dev_err(dev, "REFCLK_DIG_SEL reg field init failed\n");
+		return PTR_ERR(wiz->clk_mux_sel[REFCLK_DIG]);
+	}
+
+	for (i = 0; i < num_lanes; i++) {
+		wiz->p_enable[i] = devm_regmap_field_alloc(dev, regmap,
+							   p_enable[i]);
+		if (IS_ERR(wiz->p_enable[i])) {
+			dev_err(dev, "P%d_ENABLE reg field init failed\n", i);
+			return PTR_ERR(wiz->p_enable[i]);
+		}
+
+		wiz->p_align[i] = devm_regmap_field_alloc(dev, regmap,
+							  p_align[i]);
+		if (IS_ERR(wiz->p_align[i])) {
+			dev_err(dev, "P%d_ALIGN reg field init failed\n", i);
+			return PTR_ERR(wiz->p_align[i]);
+		}
+
+		wiz->p_raw_auto_start[i] =
+		  devm_regmap_field_alloc(dev, regmap, p_raw_auto_start[i]);
+		if (IS_ERR(wiz->p_raw_auto_start[i])) {
+			dev_err(dev, "P%d_RAW_AUTO_START reg field init fail\n",
+				i);
+			return PTR_ERR(wiz->p_raw_auto_start[i]);
+		}
+
+		wiz->p_standard_mode[i] =
+		  devm_regmap_field_alloc(dev, regmap, p_standard_mode[i]);
+		if (IS_ERR(wiz->p_standard_mode[i])) {
+			dev_err(dev, "P%d_STANDARD_MODE reg field init fail\n",
+				i);
+			return PTR_ERR(wiz->p_standard_mode[i]);
+		}
+
+		wiz->p_mac_div_sel0[i] =
+		  devm_regmap_field_alloc(dev, regmap, p_mac_div_sel0[i]);
+		if (IS_ERR(wiz->p_mac_div_sel0[i])) {
+			dev_err(dev, "P%d_MAC_DIV_SEL0 reg field init fail\n",
+				i);
+			return PTR_ERR(wiz->p_mac_div_sel0[i]);
+		}
+
+		wiz->p_mac_div_sel1[i] =
+		  devm_regmap_field_alloc(dev, regmap, p_mac_div_sel1[i]);
+		if (IS_ERR(wiz->p_mac_div_sel1[i])) {
+			dev_err(dev, "P%d_MAC_DIV_SEL1 reg field init fail\n",
+				i);
+			return PTR_ERR(wiz->p_mac_div_sel1[i]);
+		}
+	}
+
+	return 0;
+}
+
+static int wiz_clock_init(struct wiz *wiz)
+{
+	struct udevice *dev = wiz->dev;
+	unsigned long rate;
+	struct clk *clk;
+	int ret;
+
+	clk = devm_clk_get(dev, "core_ref_clk");
+	if (IS_ERR(clk)) {
+		dev_err(dev, "core_ref_clk clock not found\n");
+		ret = PTR_ERR(clk);
+		return ret;
+	}
+
+	rate = clk_get_rate(clk);
+	if (rate >= 100000000)
+		regmap_field_write(wiz->pma_cmn_refclk_int_mode, 0x1);
+	else
+		regmap_field_write(wiz->pma_cmn_refclk_int_mode, 0x3);
+
+	clk = devm_clk_get(dev, "ext_ref_clk");
+	if (IS_ERR(clk)) {
+		dev_err(dev, "ext_ref_clk clock not found\n");
+		ret = PTR_ERR(clk);
+		return ret;
+	}
+
+	rate = clk_get_rate(clk);
+	if (rate >= 100000000)
+		regmap_field_write(wiz->pma_cmn_refclk_mode, 0x0);
+	else
+		regmap_field_write(wiz->pma_cmn_refclk_mode, 0x2);
+
+	return 0;
+}
+
+static ofnode get_child_by_name(struct udevice *dev, const char *name)
+{
+	int l = strlen(name);
+	ofnode node = dev_read_first_subnode(dev);
+
+	while (ofnode_valid(node)) {
+		const char *child_name = ofnode_get_name(node);
+
+		if (!strncmp(child_name, name, l)) {
+			if (child_name[l] == '\0' || child_name[l] == '@')
+				return node;
+		}
+		node = dev_read_next_subnode(node);
+	}
+	return node;
+}
+
+static int j721e_wiz_bind_clocks(struct udevice *dev)
+{
+	int i;
+	int rc;
+	ofnode node;
+	struct driver *div_clk_drv;
+	struct driver *mux_clk_drv;
+	struct wiz_clk_mux_sel	*clk_mux_sel;
+	unsigned int clk_div_sel_num;
+	enum wiz_type type;
+
+	div_clk_drv = lists_driver_lookup_name("wiz_div_clk");
+	if (!div_clk_drv) {
+		dev_err(dev, "Cannot find driver 'wiz_div_clk'\n");
+		return -ENOENT;
+	}
+
+	mux_clk_drv = lists_driver_lookup_name("wiz_mux_clk");
+	if (!mux_clk_drv) {
+		dev_err(dev, "Cannot find driver 'wiz_mux_clk'\n");
+		return -ENOENT;
+	}
+
+	type = dev_get_driver_data(dev);
+	if (type == J721E_WIZ_10G || type == AM64_WIZ_10G) {
+		clk_mux_sel = clk_mux_sel_10g;
+		clk_div_sel_num = WIZ_DIV_NUM_CLOCKS_10G;
+	} else {
+		clk_mux_sel = clk_mux_sel_16g;
+		clk_div_sel_num = WIZ_DIV_NUM_CLOCKS_16G;
+	}
+
+	for (i = 0; i < clk_div_sel_num; i++) {
+		node = get_child_by_name(dev, clk_div_sel[i].node_name);
+		if (!ofnode_valid(node)) {
+			dev_err(dev, "cannot find node for clock %s\n",
+				clk_div_sel[i].node_name);
+			continue;
+		}
+		rc = device_bind(dev, div_clk_drv, clk_div_sel[i].node_name,
+				 &clk_div_sel[i], node, NULL);
+		if (rc) {
+			dev_err(dev, "cannot bind driver for clock %s\n",
+				clk_div_sel[i].node_name);
+		}
+	}
+
+	for (i = 0; i < WIZ_MUX_NUM_CLOCKS; i++) {
+		node = get_child_by_name(dev, clk_mux_sel[i].node_name);
+		if (!ofnode_valid(node)) {
+			dev_err(dev, "cannot find node for clock %s\n",
+				clk_mux_sel[i].node_name);
+			continue;
+		}
+		rc = device_bind(dev, mux_clk_drv, clk_mux_sel[i].node_name,
+				 &clk_mux_sel[i], node, NULL);
+		if (rc) {
+			dev_err(dev, "cannot bind driver for clock %s\n",
+				clk_mux_sel[i].node_name);
+		}
+	}
+
+	return 0;
+}
+
+static int j721e_wiz_bind_reset(struct udevice *dev)
+{
+	int rc;
+	struct driver *drv;
+
+	drv = lists_driver_lookup_name("wiz-reset");
+	if (!drv) {
+		dev_err(dev, "Cannot find driver 'wiz-reset'\n");
+		return -ENOENT;
+	}
+
+	rc = device_bind(dev, drv, "wiz-reset", NULL, dev_ofnode(dev), NULL);
+	if (rc) {
+		dev_err(dev, "cannot bind driver for wiz-reset\n");
+		return rc;
+	}
+
+	return 0;
+}
+
+static int j721e_wiz_bind(struct udevice *dev)
+{
+	dm_scan_fdt_dev(dev);
+
+	return 0;
+}
+
+static int wiz_get_lane_phy_types(struct udevice *dev, struct wiz *wiz)
+{
+	ofnode child, serdes;
+
+	serdes = get_child_by_name(dev, "serdes");
+	if (!ofnode_valid(serdes)) {
+		dev_err(dev, "%s: Getting \"serdes\"-node failed\n", __func__);
+		return -EINVAL;
+	}
+
+	ofnode_for_each_subnode(child, serdes) {
+		u32 reg, num_lanes = 1, phy_type = PHY_NONE;
+		int ret, i;
+
+		ret = ofnode_read_u32(child, "reg", &reg);
+		if (ret) {
+			dev_err(dev, "%s: Reading \"reg\" from failed: %d\n",
+				__func__, ret);
+			return ret;
+		}
+		ofnode_read_u32(child, "cdns,num-lanes", &num_lanes);
+		ofnode_read_u32(child, "cdns,phy-type", &phy_type);
+
+		dev_dbg(dev, "%s: Lanes %u-%u have phy-type %u\n", __func__,
+			reg, reg + num_lanes - 1, phy_type);
+
+		for (i = reg; i < reg + num_lanes; i++)
+			wiz->lane_phy_type[i] = phy_type;
+	}
+
+	return 0;
+}
+
+static int j721e_wiz_probe(struct udevice *dev)
+{
+	struct wiz *wiz = dev_get_priv(dev);
+	unsigned int val;
+	int rc, i;
+	ofnode node;
+	struct regmap *regmap;
+	u32 num_lanes;
+
+	node = get_child_by_name(dev, "serdes");
+
+	if (!ofnode_valid(node)) {
+		dev_err(dev, "Failed to get SERDES child DT node\n");
+		return -ENODEV;
+	}
+
+	rc = regmap_init_mem(node, &regmap);
+	if (rc)  {
+		dev_err(dev, "Failed to get memory resource\n");
+		return rc;
+	}
+	rc = dev_read_u32(dev, "num-lanes", &num_lanes);
+	if (rc) {
+		dev_err(dev, "Failed to read num-lanes property\n");
+		goto err_addr_to_resource;
+	}
+
+	if (num_lanes > WIZ_MAX_LANES) {
+		dev_err(dev, "Cannot support %d lanes\n", num_lanes);
+		goto err_addr_to_resource;
+	}
+
+	wiz->gpio_typec_dir = devm_gpiod_get_optional(dev, "typec-dir",
+						      GPIOD_IS_IN);
+	if (IS_ERR(wiz->gpio_typec_dir)) {
+		rc = PTR_ERR(wiz->gpio_typec_dir);
+		dev_err(dev, "Failed to request typec-dir gpio: %d\n", rc);
+		goto err_addr_to_resource;
+	}
+
+	wiz->regmap = regmap;
+	wiz->num_lanes = num_lanes;
+	wiz->dev = dev;
+
+	rc = wiz_get_lane_phy_types(dev, wiz);
+	if (rc) {
+		dev_err(dev, "Failed to get lane PHY types\n");
+		goto err_addr_to_resource;
+	}
+
+	rc = wiz_regfield_init(wiz);
+	if (rc) {
+		dev_err(dev, "Failed to initialize regfields\n");
+		goto err_addr_to_resource;
+	}
+
+	for (i = 0; i < wiz->num_lanes; i++) {
+		regmap_field_read(wiz->p_enable[i], &val);
+		if (val & (P_ENABLE | P_ENABLE_FORCE)) {
+			dev_err(dev, "SERDES already configured\n");
+			rc = -EBUSY;
+			goto err_addr_to_resource;
+		}
+	}
+
+	rc = j721e_wiz_bind_clocks(dev);
+	if (rc) {
+		dev_err(dev, "Failed to bind clocks\n");
+		goto err_addr_to_resource;
+	}
+
+	rc = j721e_wiz_bind_reset(dev);
+	if (rc) {
+		dev_err(dev, "Failed to bind reset\n");
+		goto err_addr_to_resource;
+	}
+
+	rc = wiz_clock_init(wiz);
+	if (rc) {
+		dev_warn(dev, "Failed to initialize clocks\n");
+		goto err_addr_to_resource;
+	}
+
+	rc = wiz_init(wiz);
+	if (rc) {
+		dev_err(dev, "WIZ initialization failed\n");
+		goto err_addr_to_resource;
+	}
+
+	return 0;
+
+err_addr_to_resource:
+	free(regmap);
+
+	return rc;
+}
+
+static int j721e_wiz_remove(struct udevice *dev)
+{
+	struct wiz *wiz = dev_get_priv(dev);
+
+	if (wiz->regmap)
+		free(wiz->regmap);
+
+	return 0;
+}
+
+static const struct udevice_id j721e_wiz_ids[] = {
+	{
+		.compatible = "ti,j721e-wiz-16g", .data = J721E_WIZ_16G,
+	},
+	{
+		.compatible = "ti,j721e-wiz-10g", .data = J721E_WIZ_10G,
+	},
+	{
+		.compatible = "ti,am64-wiz-10g", .data = AM64_WIZ_10G,
+	},
+	{}
+};
+
+U_BOOT_DRIVER(phy_j721e_wiz) = {
+	.name		= "phy-j721e-wiz",
+	.id		= UCLASS_NOP,
+	.of_match	= j721e_wiz_ids,
+	.bind		= j721e_wiz_bind,
+	.probe		= j721e_wiz_probe,
+	.remove		= j721e_wiz_remove,
+	.priv_auto	= sizeof(struct wiz),
+};