drivers/clk/imx/clk-imx8-acm.c - linux - Git at Google

 // SPDX-License-Identifier: GPL-2.0+
 //
 // Copyright 2023 NXP
 //

 #include <dt-bindings/clock/imx8-clock.h>
 #include <linux/clk-provider.h>
 #include <linux/device.h>
 #include <linux/err.h>
 #include <linux/io.h>
 #include <linux/module.h>
 #include <linux/of.h>
 #include <linux/of_device.h>
 #include <linux/platform_device.h>
 #include <linux/pm_domain.h>
 #include <linux/pm_runtime.h>
 #include <linux/slab.h>

 #include "clk.h"

 /**
  * struct clk_imx_acm_pm_domains - structure for multi power domain
  * @pd_dev: power domain device
  * @pd_dev_link: power domain device link
  * @num_domains: power domain nummber
  */
 struct clk_imx_acm_pm_domains {
 	struct device **pd_dev;
 	struct device_link **pd_dev_link;
 	int    num_domains;
 };

 /**
  * struct clk_imx8_acm_sel - for clock mux
  * @name: clock name
  * @clkid: clock id
  * @parents: clock parents
  * @num_parents: clock parents number
  * @reg: register offset
  * @shift: bit shift in register
  * @width: bits width
  */
 struct clk_imx8_acm_sel {
 	const char			*name;
 	int				clkid;
 	const struct clk_parent_data	*parents;	/* For mux */
 	int				num_parents;
 	u32				reg;
 	u8				shift;
 	u8				width;
 };

 /**
  * struct imx8_acm_soc_data - soc specific data
  * @sels: pointer to struct clk_imx8_acm_sel
  * @num_sels: numbers of items
  * @mclk_sels: pointer to imx8qm/qxp/dxl_mclk_sels
  */
 struct imx8_acm_soc_data {
 	struct clk_imx8_acm_sel *sels;
 	unsigned int num_sels;
 	struct clk_parent_data *mclk_sels;
 };

 /**
  * struct imx8_acm_priv - private structure
  * @dev_pm: multi power domain
  * @soc_data: pointer to soc data
  * @reg: base address of registers
  * @regs: save registers for suspend
  */
 struct imx8_acm_priv {
 	struct clk_imx_acm_pm_domains dev_pm;
 	const struct imx8_acm_soc_data *soc_data;
 	void __iomem *reg;
 	u32 regs[IMX_ADMA_ACM_CLK_END];
 };

 static const struct clk_parent_data imx8qm_aud_clk_sels[] = {
 	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
 	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
 	{ .fw_name = "dummy" },
 	{ .fw_name = "hdmi_rx_mclk" },
 	{ .fw_name = "ext_aud_mclk0" },
 	{ .fw_name = "ext_aud_mclk1" },
 	{ .fw_name = "esai0_rx_clk" },
 	{ .fw_name = "esai0_rx_hf_clk" },
 	{ .fw_name = "esai0_tx_clk" },
 	{ .fw_name = "esai0_tx_hf_clk" },
 	{ .fw_name = "esai1_rx_clk" },
 	{ .fw_name = "esai1_rx_hf_clk" },
 	{ .fw_name = "esai1_tx_clk" },
 	{ .fw_name = "esai1_tx_hf_clk" },
 	{ .fw_name = "spdif0_rx" },
 	{ .fw_name = "spdif1_rx" },
 	{ .fw_name = "sai0_rx_bclk" },
 	{ .fw_name = "sai0_tx_bclk" },
 	{ .fw_name = "sai1_rx_bclk" },
 	{ .fw_name = "sai1_tx_bclk" },
 	{ .fw_name = "sai2_rx_bclk" },
 	{ .fw_name = "sai3_rx_bclk" },
 	{ .fw_name = "sai4_rx_bclk" },
 };

 static const struct clk_parent_data imx8qm_mclk_out_sels[] = {
 	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
 	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
 	{ .fw_name = "dummy" },
 	{ .fw_name = "hdmi_rx_mclk" },
 	{ .fw_name = "spdif0_rx" },
 	{ .fw_name = "spdif1_rx" },
 	{ .fw_name = "sai4_rx_bclk" },
 	{ .fw_name = "sai6_rx_bclk" },
 };

 #define ACM_AUD_CLK0_SEL_INDEX  2
 #define ACM_AUD_CLK1_SEL_INDEX  3

 static struct clk_parent_data imx8qm_mclk_sels[] = {
 	{ .fw_name = "aud_pll_div_clk0_lpcg_clk" },
 	{ .fw_name = "aud_pll_div_clk1_lpcg_clk" },
 	{  }, /* clk_hw pointer of "acm_aud_clk0_sel" */
 	{  }, /* clk_hw pointer of "acm_aud_clk1_sel" */
 };

 static const struct clk_parent_data imx8qm_asrc_mux_clk_sels[] = {
 	{ .fw_name = "sai4_rx_bclk" },
 	{ .fw_name = "sai5_tx_bclk" },
 	{ .index = -1 },
 	{ .fw_name = "dummy" },

 };

 static struct clk_imx8_acm_sel imx8qm_sels[] = {
 	{ "acm_aud_clk0_sel", IMX_ADMA_ACM_AUD_CLK0_SEL, imx8qm_aud_clk_sels, ARRAY_SIZE(imx8qm_aud_clk_sels), 0x000000, 0, 5 },
 	{ "acm_aud_clk1_sel", IMX_ADMA_ACM_AUD_CLK1_SEL, imx8qm_aud_clk_sels, ARRAY_SIZE(imx8qm_aud_clk_sels), 0x010000, 0, 5 },
 	{ "acm_mclkout0_sel", IMX_ADMA_ACM_MCLKOUT0_SEL, imx8qm_mclk_out_sels, ARRAY_SIZE(imx8qm_mclk_out_sels), 0x020000, 0, 3 },
 	{ "acm_mclkout1_sel", IMX_ADMA_ACM_MCLKOUT1_SEL, imx8qm_mclk_out_sels, ARRAY_SIZE(imx8qm_mclk_out_sels), 0x030000, 0, 3 },
 	{ "acm_asrc0_mclk_sel", IMX_ADMA_ACM_ASRC0_MUX_CLK_SEL, imx8qm_asrc_mux_clk_sels, ARRAY_SIZE(imx8qm_asrc_mux_clk_sels), 0x040000, 0, 2 },
 	{ "acm_esai0_mclk_sel", IMX_ADMA_ACM_ESAI0_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x060000, 0, 2 },
 	{ "acm_esai1_mclk_sel", IMX_ADMA_ACM_ESAI1_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x070000, 0, 2 },
 	{ "acm_sai0_mclk_sel", IMX_ADMA_ACM_SAI0_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x0E0000, 0, 2 },
 	{ "acm_sai1_mclk_sel", IMX_ADMA_ACM_SAI1_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x0F0000, 0, 2 },
 	{ "acm_sai2_mclk_sel", IMX_ADMA_ACM_SAI2_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x100000, 0, 2 },
 	{ "acm_sai3_mclk_sel", IMX_ADMA_ACM_SAI3_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x110000, 0, 2 },
 	{ "acm_sai4_mclk_sel", IMX_ADMA_ACM_SAI4_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x120000, 0, 2 },
 	{ "acm_sai5_mclk_sel", IMX_ADMA_ACM_SAI5_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x130000, 0, 2 },
 	{ "acm_sai6_mclk_sel", IMX_ADMA_ACM_SAI6_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x140000, 0, 2 },
 	{ "acm_sai7_mclk_sel", IMX_ADMA_ACM_SAI7_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x150000, 0, 2 },
 	{ "acm_spdif0_mclk_sel", IMX_ADMA_ACM_SPDIF0_TX_CLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x1A0000, 0, 2 },
 	{ "acm_spdif1_mclk_sel", IMX_ADMA_ACM_SPDIF1_TX_CLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x1B0000, 0, 2 },
 	{ "acm_mqs_mclk_sel", IMX_ADMA_ACM_MQS_TX_CLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x1C0000, 0, 2 },
 };

 static const struct clk_parent_data imx8qxp_aud_clk_sels[] = {
 	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
 	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
 	{ .fw_name = "ext_aud_mclk0" },
 	{ .fw_name = "ext_aud_mclk1" },
 	{ .fw_name = "esai0_rx_clk" },
 	{ .fw_name = "esai0_rx_hf_clk" },
 	{ .fw_name = "esai0_tx_clk" },
 	{ .fw_name = "esai0_tx_hf_clk" },
 	{ .fw_name = "spdif0_rx" },
 	{ .fw_name = "sai0_rx_bclk" },
 	{ .fw_name = "sai0_tx_bclk" },
 	{ .fw_name = "sai1_rx_bclk" },
 	{ .fw_name = "sai1_tx_bclk" },
 	{ .fw_name = "sai2_rx_bclk" },
 	{ .fw_name = "sai3_rx_bclk" },
 };

 static const struct clk_parent_data imx8qxp_mclk_out_sels[] = {
 	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
 	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
 	{ .index = -1 },
 	{ .index = -1 },
 	{ .fw_name = "spdif0_rx" },
 	{ .index = -1 },
 	{ .index = -1 },
 	{ .fw_name = "sai4_rx_bclk" },
 };

 static struct clk_parent_data imx8qxp_mclk_sels[] = {
 	{ .fw_name = "aud_pll_div_clk0_lpcg_clk" },
 	{ .fw_name = "aud_pll_div_clk1_lpcg_clk" },
 	{  }, /* clk_hw pointer of "acm_aud_clk0_sel" */
 	{  }, /* clk_hw pointer of "acm_aud_clk1_sel" */
 };

 static struct clk_imx8_acm_sel imx8qxp_sels[] = {
 	{ "acm_aud_clk0_sel", IMX_ADMA_ACM_AUD_CLK0_SEL, imx8qxp_aud_clk_sels, ARRAY_SIZE(imx8qxp_aud_clk_sels), 0x000000, 0, 5 },
 	{ "acm_aud_clk1_sel", IMX_ADMA_ACM_AUD_CLK1_SEL, imx8qxp_aud_clk_sels, ARRAY_SIZE(imx8qxp_aud_clk_sels), 0x010000, 0, 5 },
 	{ "acm_mclkout0_sel", IMX_ADMA_ACM_MCLKOUT0_SEL, imx8qxp_mclk_out_sels, ARRAY_SIZE(imx8qxp_mclk_out_sels), 0x020000, 0, 3 },
 	{ "acm_mclkout1_sel", IMX_ADMA_ACM_MCLKOUT1_SEL, imx8qxp_mclk_out_sels, ARRAY_SIZE(imx8qxp_mclk_out_sels), 0x030000, 0, 3 },
 	{ "acm_esai0_mclk_sel", IMX_ADMA_ACM_ESAI0_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x060000, 0, 2 },
 	{ "acm_sai0_mclk_sel", IMX_ADMA_ACM_SAI0_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x0E0000, 0, 2 },
 	{ "acm_sai1_mclk_sel", IMX_ADMA_ACM_SAI1_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x0F0000, 0, 2 },
 	{ "acm_sai2_mclk_sel", IMX_ADMA_ACM_SAI2_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x100000, 0, 2 },
 	{ "acm_sai3_mclk_sel", IMX_ADMA_ACM_SAI3_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x110000, 0, 2 },
 	{ "acm_sai4_mclk_sel", IMX_ADMA_ACM_SAI4_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x140000, 0, 2 },
 	{ "acm_sai5_mclk_sel", IMX_ADMA_ACM_SAI5_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x150000, 0, 2 },
 	{ "acm_spdif0_mclk_sel", IMX_ADMA_ACM_SPDIF0_TX_CLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x1A0000, 0, 2 },
 	{ "acm_mqs_mclk_sel", IMX_ADMA_ACM_MQS_TX_CLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x1C0000, 0, 2 },
 };

 static const struct clk_parent_data imx8dxl_aud_clk_sels[] = {
 	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
 	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
 	{ .fw_name = "ext_aud_mclk0" },
 	{ .fw_name = "ext_aud_mclk1" },
 	{ .index = -1 },
 	{ .index = -1 },
 	{ .index = -1 },
 	{ .index = -1 },
 	{ .fw_name = "spdif0_rx" },
 	{ .fw_name = "sai0_rx_bclk" },
 	{ .fw_name = "sai0_tx_bclk" },
 	{ .fw_name = "sai1_rx_bclk" },
 	{ .fw_name = "sai1_tx_bclk" },
 	{ .fw_name = "sai2_rx_bclk" },
 	{ .fw_name = "sai3_rx_bclk" },
 };

 static const struct clk_parent_data imx8dxl_mclk_out_sels[] = {
 	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
 	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
 	{ .index = -1 },
 	{ .index = -1 },
 	{ .fw_name = "spdif0_rx" },
 	{ .index = -1 },
 	{ .index = -1 },
 	{ .index = -1 },
 };

 static struct clk_parent_data imx8dxl_mclk_sels[] = {
 	{ .fw_name = "aud_pll_div_clk0_lpcg_clk" },
 	{ .fw_name = "aud_pll_div_clk1_lpcg_clk" },
 	{  }, /* clk_hw pointer of "acm_aud_clk0_sel" */
 	{  }, /* clk_hw pointer of "acm_aud_clk1_sel" */
 };

 static struct clk_imx8_acm_sel imx8dxl_sels[] = {
 	{ "acm_aud_clk0_sel", IMX_ADMA_ACM_AUD_CLK0_SEL, imx8dxl_aud_clk_sels, ARRAY_SIZE(imx8dxl_aud_clk_sels), 0x000000, 0, 5 },
 	{ "acm_aud_clk1_sel", IMX_ADMA_ACM_AUD_CLK1_SEL, imx8dxl_aud_clk_sels, ARRAY_SIZE(imx8dxl_aud_clk_sels), 0x010000, 0, 5 },
 	{ "acm_mclkout0_sel", IMX_ADMA_ACM_MCLKOUT0_SEL, imx8dxl_mclk_out_sels, ARRAY_SIZE(imx8dxl_mclk_out_sels), 0x020000, 0, 3 },
 	{ "acm_mclkout1_sel", IMX_ADMA_ACM_MCLKOUT1_SEL, imx8dxl_mclk_out_sels, ARRAY_SIZE(imx8dxl_mclk_out_sels), 0x030000, 0, 3 },
 	{ "acm_sai0_mclk_sel", IMX_ADMA_ACM_SAI0_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x0E0000, 0, 2 },
 	{ "acm_sai1_mclk_sel", IMX_ADMA_ACM_SAI1_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x0F0000, 0, 2 },
 	{ "acm_sai2_mclk_sel", IMX_ADMA_ACM_SAI2_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x100000, 0, 2 },
 	{ "acm_sai3_mclk_sel", IMX_ADMA_ACM_SAI3_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x110000, 0, 2 },
 	{ "acm_spdif0_mclk_sel", IMX_ADMA_ACM_SPDIF0_TX_CLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x1A0000, 0, 2 },
 	{ "acm_mqs_mclk_sel", IMX_ADMA_ACM_MQS_TX_CLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x1C0000, 0, 2 },
 };

 /**
  * clk_imx_acm_attach_pm_domains: attach multi power domains
  * @dev: device pointer
  * @dev_pm: power domains for device
  */
 static int clk_imx_acm_attach_pm_domains(struct device *dev,
 					 struct clk_imx_acm_pm_domains *dev_pm)
 {
 	int ret;
 	int i;

 	dev_pm->num_domains = of_count_phandle_with_args(dev->of_node, "power-domains",
 							 "#power-domain-cells");
 	if (dev_pm->num_domains <= 1)
 		return 0;

 	dev_pm->pd_dev = devm_kmalloc_array(dev, dev_pm->num_domains,
 					    sizeof(*dev_pm->pd_dev),
 					    GFP_KERNEL);
 	if (!dev_pm->pd_dev)
 		return -ENOMEM;

 	dev_pm->pd_dev_link = devm_kmalloc_array(dev,
 						 dev_pm->num_domains,
 						 sizeof(*dev_pm->pd_dev_link),
 						 GFP_KERNEL);
 	if (!dev_pm->pd_dev_link)
 		return -ENOMEM;

 	for (i = 0; i < dev_pm->num_domains; i++) {
 		dev_pm->pd_dev[i] = dev_pm_domain_attach_by_id(dev, i);
 		if (IS_ERR(dev_pm->pd_dev[i])) {
 			ret = PTR_ERR(dev_pm->pd_dev[i]);
 			goto detach_pm;
 		}

 		dev_pm->pd_dev_link[i] = device_link_add(dev,
 							 dev_pm->pd_dev[i],
 							 DL_FLAG_STATELESS |
 							 DL_FLAG_PM_RUNTIME |
 							 DL_FLAG_RPM_ACTIVE);
 		if (IS_ERR(dev_pm->pd_dev_link[i])) {
 			dev_pm_domain_detach(dev_pm->pd_dev[i], false);
 			ret = PTR_ERR(dev_pm->pd_dev_link[i]);
 			goto detach_pm;
 		}
 	}
 	return 0;

 detach_pm:
 	while (--i >= 0) {
 		device_link_del(dev_pm->pd_dev_link[i]);
 		dev_pm_domain_detach(dev_pm->pd_dev[i], false);
 	}
 	return ret;
 }

 /**
  * clk_imx_acm_detach_pm_domains: detach multi power domains
  * @dev: deivice pointer
  * @dev_pm: multi power domain for device
  */
 static void clk_imx_acm_detach_pm_domains(struct device *dev,
 					  struct clk_imx_acm_pm_domains *dev_pm)
 {
 	int i;

 	if (dev_pm->num_domains <= 1)
 		return;

 	for (i = 0; i < dev_pm->num_domains; i++) {
 		device_link_del(dev_pm->pd_dev_link[i]);
 		dev_pm_domain_detach(dev_pm->pd_dev[i], false);
 	}
 }

 static int imx8_acm_clk_probe(struct platform_device *pdev)
 {
 	struct clk_hw_onecell_data *clk_hw_data;
 	struct device *dev = &pdev->dev;
 	struct clk_imx8_acm_sel *sels;
 	struct imx8_acm_priv *priv;
 	struct clk_hw **hws;
 	void __iomem *base;
 	int ret;
 	int i;

 	base = devm_of_iomap(dev, dev->of_node, 0, NULL);
 	if (WARN_ON(IS_ERR(base)))
 		return PTR_ERR(base);

 	priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
 	if (!priv)
 		return -ENOMEM;

 	priv->reg = base;
 	priv->soc_data = of_device_get_match_data(dev);
 	platform_set_drvdata(pdev, priv);

 	clk_hw_data = devm_kzalloc(&pdev->dev, struct_size(clk_hw_data, hws, IMX_ADMA_ACM_CLK_END),
 				   GFP_KERNEL);
 	if (!clk_hw_data)
 		return -ENOMEM;

 	clk_hw_data->num = IMX_ADMA_ACM_CLK_END;
 	hws = clk_hw_data->hws;

 	ret = clk_imx_acm_attach_pm_domains(&pdev->dev, &priv->dev_pm);
 	if (ret)
 		return ret;

 	pm_runtime_enable(&pdev->dev);
 	pm_runtime_get_sync(&pdev->dev);

 	sels = priv->soc_data->sels;
 	for (i = 0; i < priv->soc_data->num_sels; i++) {
 		hws[sels[i].clkid] = devm_clk_hw_register_mux_parent_data_table(dev,
 										sels[i].name, sels[i].parents,
 										sels[i].num_parents, 0,
 										base + sels[i].reg,
 										sels[i].shift, sels[i].width,
 										0, NULL, NULL);
 		if (IS_ERR(hws[sels[i].clkid])) {
 			ret = PTR_ERR(hws[sels[i].clkid]);
 			imx_check_clk_hws(hws, IMX_ADMA_ACM_CLK_END);
 			goto err_clk_register;
 		}

 		/*
 		 * The IMX_ADMA_ACM_AUD_CLK0_SEL and IMX_ADMA_ACM_AUD_CLK1_SEL are
 		 * registered first. After registration, update the clk_hw pointer
 		 * to imx8qm/qxp/dxl_mclk_sels structures.
 		 */
 		if (sels[i].clkid == IMX_ADMA_ACM_AUD_CLK0_SEL)
 			priv->soc_data->mclk_sels[ACM_AUD_CLK0_SEL_INDEX].hw =
 								hws[IMX_ADMA_ACM_AUD_CLK0_SEL];
 		if (sels[i].clkid == IMX_ADMA_ACM_AUD_CLK1_SEL)
 			priv->soc_data->mclk_sels[ACM_AUD_CLK1_SEL_INDEX].hw =
 								hws[IMX_ADMA_ACM_AUD_CLK1_SEL];
 	}

 	ret = devm_of_clk_add_hw_provider(dev, of_clk_hw_onecell_get, clk_hw_data);
 	if (ret < 0) {
 		dev_err(dev, "failed to register hws for ACM\n");
 		goto err_clk_register;
 	}

 	pm_runtime_put_sync(&pdev->dev);
 	return 0;

 err_clk_register:
 	pm_runtime_put_sync(&pdev->dev);
 	pm_runtime_disable(&pdev->dev);
 	clk_imx_acm_detach_pm_domains(&pdev->dev, &priv->dev_pm);

 	return ret;
 }

 static void imx8_acm_clk_remove(struct platform_device *pdev)
 {
 	struct imx8_acm_priv *priv = dev_get_drvdata(&pdev->dev);

 	pm_runtime_disable(&pdev->dev);

 	clk_imx_acm_detach_pm_domains(&pdev->dev, &priv->dev_pm);
 }

 static const struct imx8_acm_soc_data imx8qm_acm_data = {
 	.sels = imx8qm_sels,
 	.num_sels = ARRAY_SIZE(imx8qm_sels),
 	.mclk_sels = imx8qm_mclk_sels,
 };

 static const struct imx8_acm_soc_data imx8qxp_acm_data = {
 	.sels = imx8qxp_sels,
 	.num_sels = ARRAY_SIZE(imx8qxp_sels),
 	.mclk_sels = imx8qxp_mclk_sels,
 };

 static const struct imx8_acm_soc_data imx8dxl_acm_data = {
 	.sels = imx8dxl_sels,
 	.num_sels = ARRAY_SIZE(imx8dxl_sels),
 	.mclk_sels = imx8dxl_mclk_sels,
 };

 static const struct of_device_id imx8_acm_match[] = {
 	{ .compatible = "fsl,imx8qm-acm", .data = &imx8qm_acm_data },
 	{ .compatible = "fsl,imx8qxp-acm", .data = &imx8qxp_acm_data },
 	{ .compatible = "fsl,imx8dxl-acm", .data = &imx8dxl_acm_data },
 	{ /* sentinel */ }
 };
 MODULE_DEVICE_TABLE(of, imx8_acm_match);

 static int __maybe_unused imx8_acm_runtime_suspend(struct device *dev)
 {
 	struct imx8_acm_priv *priv = dev_get_drvdata(dev);
 	struct clk_imx8_acm_sel *sels;
 	int i;

 	sels = priv->soc_data->sels;

 	for (i = 0; i < priv->soc_data->num_sels; i++)
 		priv->regs[i] = readl_relaxed(priv->reg + sels[i].reg);

 	return 0;
 }

 static int __maybe_unused imx8_acm_runtime_resume(struct device *dev)
 {
 	struct imx8_acm_priv *priv = dev_get_drvdata(dev);
 	struct clk_imx8_acm_sel *sels;
 	int i;

 	sels = priv->soc_data->sels;

 	for (i = 0; i < priv->soc_data->num_sels; i++)
 		writel_relaxed(priv->regs[i], priv->reg + sels[i].reg);

 	return 0;
 }

 static const struct dev_pm_ops imx8_acm_pm_ops = {
 	SET_RUNTIME_PM_OPS(imx8_acm_runtime_suspend,
 			   imx8_acm_runtime_resume, NULL)
 	SET_NOIRQ_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
 				      pm_runtime_force_resume)
 };

 static struct platform_driver imx8_acm_clk_driver = {
 	.driver = {
 		.name = "imx8-acm",
 		.of_match_table = imx8_acm_match,
 		.pm = &imx8_acm_pm_ops,
 	},
 	.probe = imx8_acm_clk_probe,
 	.remove = imx8_acm_clk_remove,
 };
 module_platform_driver(imx8_acm_clk_driver);

 MODULE_AUTHOR("Shengjiu Wang <shengjiu.wang@nxp.com>");
 MODULE_DESCRIPTION("Freescale i.MX8 Audio Clock Mux driver");
 MODULE_LICENSE("GPL");
	// SPDX-License-Identifier: GPL-2.0+
	//
	// Copyright 2023 NXP
	//

	#include <dt-bindings/clock/imx8-clock.h>
	#include <linux/clk-provider.h>
	#include <linux/device.h>
	#include <linux/err.h>
	#include <linux/io.h>
	#include <linux/module.h>
	#include <linux/of.h>
	#include <linux/of_device.h>
	#include <linux/platform_device.h>
	#include <linux/pm_domain.h>
	#include <linux/pm_runtime.h>
	#include <linux/slab.h>

	#include "clk.h"

	/**
	* struct clk_imx_acm_pm_domains - structure for multi power domain
	* @pd_dev: power domain device
	* @pd_dev_link: power domain device link
	* @num_domains: power domain nummber
	*/
	struct clk_imx_acm_pm_domains {
	struct device **pd_dev;
	struct device_link **pd_dev_link;
	int num_domains;
	};

	/**
	* struct clk_imx8_acm_sel - for clock mux
	* @name: clock name
	* @clkid: clock id
	* @parents: clock parents
	* @num_parents: clock parents number
	* @reg: register offset
	* @shift: bit shift in register
	* @width: bits width
	*/
	struct clk_imx8_acm_sel {
	const char *name;
	int clkid;
	const struct clk_parent_data parents; / For mux */
	int num_parents;
	u32 reg;
	u8 shift;
	u8 width;
	};

	/**
	* struct imx8_acm_soc_data - soc specific data
	* @sels: pointer to struct clk_imx8_acm_sel
	* @num_sels: numbers of items
	* @mclk_sels: pointer to imx8qm/qxp/dxl_mclk_sels
	*/
	struct imx8_acm_soc_data {
	struct clk_imx8_acm_sel *sels;
	unsigned int num_sels;
	struct clk_parent_data *mclk_sels;
	};

	/**
	* struct imx8_acm_priv - private structure
	* @dev_pm: multi power domain
	* @soc_data: pointer to soc data
	* @reg: base address of registers
	* @regs: save registers for suspend
	*/
	struct imx8_acm_priv {
	struct clk_imx_acm_pm_domains dev_pm;
	const struct imx8_acm_soc_data *soc_data;
	void __iomem *reg;
	u32 regs[IMX_ADMA_ACM_CLK_END];
	};

	static const struct clk_parent_data imx8qm_aud_clk_sels[] = {
	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
	{ .fw_name = "dummy" },
	{ .fw_name = "hdmi_rx_mclk" },
	{ .fw_name = "ext_aud_mclk0" },
	{ .fw_name = "ext_aud_mclk1" },
	{ .fw_name = "esai0_rx_clk" },
	{ .fw_name = "esai0_rx_hf_clk" },
	{ .fw_name = "esai0_tx_clk" },
	{ .fw_name = "esai0_tx_hf_clk" },
	{ .fw_name = "esai1_rx_clk" },
	{ .fw_name = "esai1_rx_hf_clk" },
	{ .fw_name = "esai1_tx_clk" },
	{ .fw_name = "esai1_tx_hf_clk" },
	{ .fw_name = "spdif0_rx" },
	{ .fw_name = "spdif1_rx" },
	{ .fw_name = "sai0_rx_bclk" },
	{ .fw_name = "sai0_tx_bclk" },
	{ .fw_name = "sai1_rx_bclk" },
	{ .fw_name = "sai1_tx_bclk" },
	{ .fw_name = "sai2_rx_bclk" },
	{ .fw_name = "sai3_rx_bclk" },
	{ .fw_name = "sai4_rx_bclk" },
	};

	static const struct clk_parent_data imx8qm_mclk_out_sels[] = {
	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
	{ .fw_name = "dummy" },
	{ .fw_name = "hdmi_rx_mclk" },
	{ .fw_name = "spdif0_rx" },
	{ .fw_name = "spdif1_rx" },
	{ .fw_name = "sai4_rx_bclk" },
	{ .fw_name = "sai6_rx_bclk" },
	};

	#define ACM_AUD_CLK0_SEL_INDEX 2
	#define ACM_AUD_CLK1_SEL_INDEX 3

	static struct clk_parent_data imx8qm_mclk_sels[] = {
	{ .fw_name = "aud_pll_div_clk0_lpcg_clk" },
	{ .fw_name = "aud_pll_div_clk1_lpcg_clk" },
	{ }, /* clk_hw pointer of "acm_aud_clk0_sel" */
	{ }, /* clk_hw pointer of "acm_aud_clk1_sel" */
	};

	static const struct clk_parent_data imx8qm_asrc_mux_clk_sels[] = {
	{ .fw_name = "sai4_rx_bclk" },
	{ .fw_name = "sai5_tx_bclk" },
	{ .index = -1 },
	{ .fw_name = "dummy" },

	};

	static struct clk_imx8_acm_sel imx8qm_sels[] = {
	{ "acm_aud_clk0_sel", IMX_ADMA_ACM_AUD_CLK0_SEL, imx8qm_aud_clk_sels, ARRAY_SIZE(imx8qm_aud_clk_sels), 0x000000, 0, 5 },
	{ "acm_aud_clk1_sel", IMX_ADMA_ACM_AUD_CLK1_SEL, imx8qm_aud_clk_sels, ARRAY_SIZE(imx8qm_aud_clk_sels), 0x010000, 0, 5 },
	{ "acm_mclkout0_sel", IMX_ADMA_ACM_MCLKOUT0_SEL, imx8qm_mclk_out_sels, ARRAY_SIZE(imx8qm_mclk_out_sels), 0x020000, 0, 3 },
	{ "acm_mclkout1_sel", IMX_ADMA_ACM_MCLKOUT1_SEL, imx8qm_mclk_out_sels, ARRAY_SIZE(imx8qm_mclk_out_sels), 0x030000, 0, 3 },
	{ "acm_asrc0_mclk_sel", IMX_ADMA_ACM_ASRC0_MUX_CLK_SEL, imx8qm_asrc_mux_clk_sels, ARRAY_SIZE(imx8qm_asrc_mux_clk_sels), 0x040000, 0, 2 },
	{ "acm_esai0_mclk_sel", IMX_ADMA_ACM_ESAI0_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x060000, 0, 2 },
	{ "acm_esai1_mclk_sel", IMX_ADMA_ACM_ESAI1_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x070000, 0, 2 },
	{ "acm_sai0_mclk_sel", IMX_ADMA_ACM_SAI0_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x0E0000, 0, 2 },
	{ "acm_sai1_mclk_sel", IMX_ADMA_ACM_SAI1_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x0F0000, 0, 2 },
	{ "acm_sai2_mclk_sel", IMX_ADMA_ACM_SAI2_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x100000, 0, 2 },
	{ "acm_sai3_mclk_sel", IMX_ADMA_ACM_SAI3_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x110000, 0, 2 },
	{ "acm_sai4_mclk_sel", IMX_ADMA_ACM_SAI4_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x120000, 0, 2 },
	{ "acm_sai5_mclk_sel", IMX_ADMA_ACM_SAI5_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x130000, 0, 2 },
	{ "acm_sai6_mclk_sel", IMX_ADMA_ACM_SAI6_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x140000, 0, 2 },
	{ "acm_sai7_mclk_sel", IMX_ADMA_ACM_SAI7_MCLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x150000, 0, 2 },
	{ "acm_spdif0_mclk_sel", IMX_ADMA_ACM_SPDIF0_TX_CLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x1A0000, 0, 2 },
	{ "acm_spdif1_mclk_sel", IMX_ADMA_ACM_SPDIF1_TX_CLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x1B0000, 0, 2 },
	{ "acm_mqs_mclk_sel", IMX_ADMA_ACM_MQS_TX_CLK_SEL, imx8qm_mclk_sels, ARRAY_SIZE(imx8qm_mclk_sels), 0x1C0000, 0, 2 },
	};

	static const struct clk_parent_data imx8qxp_aud_clk_sels[] = {
	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
	{ .fw_name = "ext_aud_mclk0" },
	{ .fw_name = "ext_aud_mclk1" },
	{ .fw_name = "esai0_rx_clk" },
	{ .fw_name = "esai0_rx_hf_clk" },
	{ .fw_name = "esai0_tx_clk" },
	{ .fw_name = "esai0_tx_hf_clk" },
	{ .fw_name = "spdif0_rx" },
	{ .fw_name = "sai0_rx_bclk" },
	{ .fw_name = "sai0_tx_bclk" },
	{ .fw_name = "sai1_rx_bclk" },
	{ .fw_name = "sai1_tx_bclk" },
	{ .fw_name = "sai2_rx_bclk" },
	{ .fw_name = "sai3_rx_bclk" },
	};

	static const struct clk_parent_data imx8qxp_mclk_out_sels[] = {
	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
	{ .index = -1 },
	{ .index = -1 },
	{ .fw_name = "spdif0_rx" },
	{ .index = -1 },
	{ .index = -1 },
	{ .fw_name = "sai4_rx_bclk" },
	};

	static struct clk_parent_data imx8qxp_mclk_sels[] = {
	{ .fw_name = "aud_pll_div_clk0_lpcg_clk" },
	{ .fw_name = "aud_pll_div_clk1_lpcg_clk" },
	{ }, /* clk_hw pointer of "acm_aud_clk0_sel" */
	{ }, /* clk_hw pointer of "acm_aud_clk1_sel" */
	};

	static struct clk_imx8_acm_sel imx8qxp_sels[] = {
	{ "acm_aud_clk0_sel", IMX_ADMA_ACM_AUD_CLK0_SEL, imx8qxp_aud_clk_sels, ARRAY_SIZE(imx8qxp_aud_clk_sels), 0x000000, 0, 5 },
	{ "acm_aud_clk1_sel", IMX_ADMA_ACM_AUD_CLK1_SEL, imx8qxp_aud_clk_sels, ARRAY_SIZE(imx8qxp_aud_clk_sels), 0x010000, 0, 5 },
	{ "acm_mclkout0_sel", IMX_ADMA_ACM_MCLKOUT0_SEL, imx8qxp_mclk_out_sels, ARRAY_SIZE(imx8qxp_mclk_out_sels), 0x020000, 0, 3 },
	{ "acm_mclkout1_sel", IMX_ADMA_ACM_MCLKOUT1_SEL, imx8qxp_mclk_out_sels, ARRAY_SIZE(imx8qxp_mclk_out_sels), 0x030000, 0, 3 },
	{ "acm_esai0_mclk_sel", IMX_ADMA_ACM_ESAI0_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x060000, 0, 2 },
	{ "acm_sai0_mclk_sel", IMX_ADMA_ACM_SAI0_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x0E0000, 0, 2 },
	{ "acm_sai1_mclk_sel", IMX_ADMA_ACM_SAI1_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x0F0000, 0, 2 },
	{ "acm_sai2_mclk_sel", IMX_ADMA_ACM_SAI2_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x100000, 0, 2 },
	{ "acm_sai3_mclk_sel", IMX_ADMA_ACM_SAI3_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x110000, 0, 2 },
	{ "acm_sai4_mclk_sel", IMX_ADMA_ACM_SAI4_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x140000, 0, 2 },
	{ "acm_sai5_mclk_sel", IMX_ADMA_ACM_SAI5_MCLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x150000, 0, 2 },
	{ "acm_spdif0_mclk_sel", IMX_ADMA_ACM_SPDIF0_TX_CLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x1A0000, 0, 2 },
	{ "acm_mqs_mclk_sel", IMX_ADMA_ACM_MQS_TX_CLK_SEL, imx8qxp_mclk_sels, ARRAY_SIZE(imx8qxp_mclk_sels), 0x1C0000, 0, 2 },
	};

	static const struct clk_parent_data imx8dxl_aud_clk_sels[] = {
	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
	{ .fw_name = "ext_aud_mclk0" },
	{ .fw_name = "ext_aud_mclk1" },
	{ .index = -1 },
	{ .index = -1 },
	{ .index = -1 },
	{ .index = -1 },
	{ .fw_name = "spdif0_rx" },
	{ .fw_name = "sai0_rx_bclk" },
	{ .fw_name = "sai0_tx_bclk" },
	{ .fw_name = "sai1_rx_bclk" },
	{ .fw_name = "sai1_tx_bclk" },
	{ .fw_name = "sai2_rx_bclk" },
	{ .fw_name = "sai3_rx_bclk" },
	};

	static const struct clk_parent_data imx8dxl_mclk_out_sels[] = {
	{ .fw_name = "aud_rec_clk0_lpcg_clk" },
	{ .fw_name = "aud_rec_clk1_lpcg_clk" },
	{ .index = -1 },
	{ .index = -1 },
	{ .fw_name = "spdif0_rx" },
	{ .index = -1 },
	{ .index = -1 },
	{ .index = -1 },
	};

	static struct clk_parent_data imx8dxl_mclk_sels[] = {
	{ .fw_name = "aud_pll_div_clk0_lpcg_clk" },
	{ .fw_name = "aud_pll_div_clk1_lpcg_clk" },
	{ }, /* clk_hw pointer of "acm_aud_clk0_sel" */
	{ }, /* clk_hw pointer of "acm_aud_clk1_sel" */
	};

	static struct clk_imx8_acm_sel imx8dxl_sels[] = {
	{ "acm_aud_clk0_sel", IMX_ADMA_ACM_AUD_CLK0_SEL, imx8dxl_aud_clk_sels, ARRAY_SIZE(imx8dxl_aud_clk_sels), 0x000000, 0, 5 },
	{ "acm_aud_clk1_sel", IMX_ADMA_ACM_AUD_CLK1_SEL, imx8dxl_aud_clk_sels, ARRAY_SIZE(imx8dxl_aud_clk_sels), 0x010000, 0, 5 },
	{ "acm_mclkout0_sel", IMX_ADMA_ACM_MCLKOUT0_SEL, imx8dxl_mclk_out_sels, ARRAY_SIZE(imx8dxl_mclk_out_sels), 0x020000, 0, 3 },
	{ "acm_mclkout1_sel", IMX_ADMA_ACM_MCLKOUT1_SEL, imx8dxl_mclk_out_sels, ARRAY_SIZE(imx8dxl_mclk_out_sels), 0x030000, 0, 3 },
	{ "acm_sai0_mclk_sel", IMX_ADMA_ACM_SAI0_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x0E0000, 0, 2 },
	{ "acm_sai1_mclk_sel", IMX_ADMA_ACM_SAI1_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x0F0000, 0, 2 },
	{ "acm_sai2_mclk_sel", IMX_ADMA_ACM_SAI2_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x100000, 0, 2 },
	{ "acm_sai3_mclk_sel", IMX_ADMA_ACM_SAI3_MCLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x110000, 0, 2 },
	{ "acm_spdif0_mclk_sel", IMX_ADMA_ACM_SPDIF0_TX_CLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x1A0000, 0, 2 },
	{ "acm_mqs_mclk_sel", IMX_ADMA_ACM_MQS_TX_CLK_SEL, imx8dxl_mclk_sels, ARRAY_SIZE(imx8dxl_mclk_sels), 0x1C0000, 0, 2 },
	};

	/**
	* clk_imx_acm_attach_pm_domains: attach multi power domains
	* @dev: device pointer
	* @dev_pm: power domains for device
	*/
	static int clk_imx_acm_attach_pm_domains(struct device *dev,
	struct clk_imx_acm_pm_domains *dev_pm)
	{
	int ret;
	int i;

	dev_pm->num_domains = of_count_phandle_with_args(dev->of_node, "power-domains",
	"#power-domain-cells");
	if (dev_pm->num_domains <= 1)
	return 0;

	dev_pm->pd_dev = devm_kmalloc_array(dev, dev_pm->num_domains,
	sizeof(*dev_pm->pd_dev),
	GFP_KERNEL);
	if (!dev_pm->pd_dev)
	return -ENOMEM;

	dev_pm->pd_dev_link = devm_kmalloc_array(dev,
	dev_pm->num_domains,
	sizeof(*dev_pm->pd_dev_link),
	GFP_KERNEL);
	if (!dev_pm->pd_dev_link)
	return -ENOMEM;

	for (i = 0; i < dev_pm->num_domains; i++) {
	dev_pm->pd_dev[i] = dev_pm_domain_attach_by_id(dev, i);
	if (IS_ERR(dev_pm->pd_dev[i])) {
	ret = PTR_ERR(dev_pm->pd_dev[i]);
	goto detach_pm;
	}

	dev_pm->pd_dev_link[i] = device_link_add(dev,
	dev_pm->pd_dev[i],
	DL_FLAG_STATELESS \|
	DL_FLAG_PM_RUNTIME \|
	DL_FLAG_RPM_ACTIVE);
	if (IS_ERR(dev_pm->pd_dev_link[i])) {
	dev_pm_domain_detach(dev_pm->pd_dev[i], false);
	ret = PTR_ERR(dev_pm->pd_dev_link[i]);
	goto detach_pm;
	}
	}
	return 0;

	detach_pm:
	while (--i >= 0) {
	device_link_del(dev_pm->pd_dev_link[i]);
	dev_pm_domain_detach(dev_pm->pd_dev[i], false);
	}
	return ret;
	}

	/**
	* clk_imx_acm_detach_pm_domains: detach multi power domains
	* @dev: deivice pointer
	* @dev_pm: multi power domain for device
	*/
	static void clk_imx_acm_detach_pm_domains(struct device *dev,
	struct clk_imx_acm_pm_domains *dev_pm)
	{
	int i;

	if (dev_pm->num_domains <= 1)
	return;

	for (i = 0; i < dev_pm->num_domains; i++) {
	device_link_del(dev_pm->pd_dev_link[i]);
	dev_pm_domain_detach(dev_pm->pd_dev[i], false);
	}
	}

	static int imx8_acm_clk_probe(struct platform_device *pdev)
	{
	struct clk_hw_onecell_data *clk_hw_data;
	struct device *dev = &pdev->dev;
	struct clk_imx8_acm_sel *sels;
	struct imx8_acm_priv *priv;
	struct clk_hw **hws;
	void __iomem *base;
	int ret;
	int i;

	base = devm_of_iomap(dev, dev->of_node, 0, NULL);
	if (WARN_ON(IS_ERR(base)))
	return PTR_ERR(base);

	priv = devm_kzalloc(&pdev->dev, sizeof(*priv), GFP_KERNEL);
	if (!priv)
	return -ENOMEM;

	priv->reg = base;
	priv->soc_data = of_device_get_match_data(dev);
	platform_set_drvdata(pdev, priv);

	clk_hw_data = devm_kzalloc(&pdev->dev, struct_size(clk_hw_data, hws, IMX_ADMA_ACM_CLK_END),
	GFP_KERNEL);
	if (!clk_hw_data)
	return -ENOMEM;

	clk_hw_data->num = IMX_ADMA_ACM_CLK_END;
	hws = clk_hw_data->hws;

	ret = clk_imx_acm_attach_pm_domains(&pdev->dev, &priv->dev_pm);
	if (ret)
	return ret;

	pm_runtime_enable(&pdev->dev);
	pm_runtime_get_sync(&pdev->dev);

	sels = priv->soc_data->sels;
	for (i = 0; i < priv->soc_data->num_sels; i++) {
	hws[sels[i].clkid] = devm_clk_hw_register_mux_parent_data_table(dev,
	sels[i].name, sels[i].parents,
	sels[i].num_parents, 0,
	base + sels[i].reg,
	sels[i].shift, sels[i].width,
	0, NULL, NULL);
	if (IS_ERR(hws[sels[i].clkid])) {
	ret = PTR_ERR(hws[sels[i].clkid]);
	imx_check_clk_hws(hws, IMX_ADMA_ACM_CLK_END);
	goto err_clk_register;
	}

	/*
	* The IMX_ADMA_ACM_AUD_CLK0_SEL and IMX_ADMA_ACM_AUD_CLK1_SEL are
	* registered first. After registration, update the clk_hw pointer
	* to imx8qm/qxp/dxl_mclk_sels structures.
	*/
	if (sels[i].clkid == IMX_ADMA_ACM_AUD_CLK0_SEL)
	priv->soc_data->mclk_sels[ACM_AUD_CLK0_SEL_INDEX].hw =
	hws[IMX_ADMA_ACM_AUD_CLK0_SEL];
	if (sels[i].clkid == IMX_ADMA_ACM_AUD_CLK1_SEL)
	priv->soc_data->mclk_sels[ACM_AUD_CLK1_SEL_INDEX].hw =
	hws[IMX_ADMA_ACM_AUD_CLK1_SEL];
	}

	ret = devm_of_clk_add_hw_provider(dev, of_clk_hw_onecell_get, clk_hw_data);
	if (ret < 0) {
	dev_err(dev, "failed to register hws for ACM\n");
	goto err_clk_register;
	}

	pm_runtime_put_sync(&pdev->dev);
	return 0;

	err_clk_register:
	pm_runtime_put_sync(&pdev->dev);
	pm_runtime_disable(&pdev->dev);
	clk_imx_acm_detach_pm_domains(&pdev->dev, &priv->dev_pm);

	return ret;
	}

	static void imx8_acm_clk_remove(struct platform_device *pdev)
	{
	struct imx8_acm_priv *priv = dev_get_drvdata(&pdev->dev);

	pm_runtime_disable(&pdev->dev);

	clk_imx_acm_detach_pm_domains(&pdev->dev, &priv->dev_pm);
	}

	static const struct imx8_acm_soc_data imx8qm_acm_data = {
	.sels = imx8qm_sels,
	.num_sels = ARRAY_SIZE(imx8qm_sels),
	.mclk_sels = imx8qm_mclk_sels,
	};

	static const struct imx8_acm_soc_data imx8qxp_acm_data = {
	.sels = imx8qxp_sels,
	.num_sels = ARRAY_SIZE(imx8qxp_sels),
	.mclk_sels = imx8qxp_mclk_sels,
	};

	static const struct imx8_acm_soc_data imx8dxl_acm_data = {
	.sels = imx8dxl_sels,
	.num_sels = ARRAY_SIZE(imx8dxl_sels),
	.mclk_sels = imx8dxl_mclk_sels,
	};

	static const struct of_device_id imx8_acm_match[] = {
	{ .compatible = "fsl,imx8qm-acm", .data = &imx8qm_acm_data },
	{ .compatible = "fsl,imx8qxp-acm", .data = &imx8qxp_acm_data },
	{ .compatible = "fsl,imx8dxl-acm", .data = &imx8dxl_acm_data },
	{ /* sentinel */ }
	};
	MODULE_DEVICE_TABLE(of, imx8_acm_match);

	static int __maybe_unused imx8_acm_runtime_suspend(struct device *dev)
	{
	struct imx8_acm_priv *priv = dev_get_drvdata(dev);
	struct clk_imx8_acm_sel *sels;
	int i;

	sels = priv->soc_data->sels;

	for (i = 0; i < priv->soc_data->num_sels; i++)
	priv->regs[i] = readl_relaxed(priv->reg + sels[i].reg);

	return 0;
	}

	static int __maybe_unused imx8_acm_runtime_resume(struct device *dev)
	{
	struct imx8_acm_priv *priv = dev_get_drvdata(dev);
	struct clk_imx8_acm_sel *sels;
	int i;

	sels = priv->soc_data->sels;

	for (i = 0; i < priv->soc_data->num_sels; i++)
	writel_relaxed(priv->regs[i], priv->reg + sels[i].reg);

	return 0;
	}

	static const struct dev_pm_ops imx8_acm_pm_ops = {
	SET_RUNTIME_PM_OPS(imx8_acm_runtime_suspend,
	imx8_acm_runtime_resume, NULL)
	SET_NOIRQ_SYSTEM_SLEEP_PM_OPS(pm_runtime_force_suspend,
	pm_runtime_force_resume)
	};

	static struct platform_driver imx8_acm_clk_driver = {
	.driver = {
	.name = "imx8-acm",
	.of_match_table = imx8_acm_match,
	.pm = &imx8_acm_pm_ops,
	},
	.probe = imx8_acm_clk_probe,
	.remove = imx8_acm_clk_remove,
	};
	module_platform_driver(imx8_acm_clk_driver);

	MODULE_AUTHOR("Shengjiu Wang <shengjiu.wang@nxp.com>");
	MODULE_DESCRIPTION("Freescale i.MX8 Audio Clock Mux driver");
	MODULE_LICENSE("GPL");