#include <linux/delay.h>
 #include <linux/device.h>
 #include <linux/init.h>
+#include <linux/iopoll.h>
 #include <linux/mod_devicetable.h>
 #include <linux/module.h>
 #include <linux/of_address.h>
 #define GET_REG_SAMPLL_CLK1(val)       ((val >> 22) & 0xfff)
 #define GET_REG_SAMPLL_CLK2(val)       ((val >> 12) & 0xfff)
 
+struct sd_hw_data {
+       struct clk_hw hw;
+       u32 conf;
+       struct rzg2l_cpg_priv *priv;
+};
+
+#define to_sd_hw_data(_hw)     container_of(_hw, struct sd_hw_data, hw)
+
 /**
  * struct rzg2l_cpg_priv - Clock Pulse Generator Private Data
  *
        return clk_hw->clk;
 }
 
+static int rzg2l_cpg_sd_clk_mux_determine_rate(struct clk_hw *hw,
+                                              struct clk_rate_request *req)
+{
+       return clk_mux_determine_rate_flags(hw, req, 0);
+}
+
+static int rzg2l_cpg_sd_clk_mux_set_parent(struct clk_hw *hw, u8 index)
+{
+       struct sd_hw_data *hwdata = to_sd_hw_data(hw);
+       struct rzg2l_cpg_priv *priv = hwdata->priv;
+       u32 off = GET_REG_OFFSET(hwdata->conf);
+       u32 shift = GET_SHIFT(hwdata->conf);
+       const u32 clk_src_266 = 2;
+       u32 bitmask;
+
+       /*
+        * As per the HW manual, we should not directly switch from 533 MHz to
+        * 400 MHz and vice versa. To change the setting from 2’b01 (533 MHz)
+        * to 2’b10 (400 MHz) or vice versa, Switch to 2’b11 (266 MHz) first,
+        * and then switch to the target setting (2’b01 (533 MHz) or 2’b10
+        * (400 MHz)).
+        * Setting a value of '0' to the SEL_SDHI0_SET or SEL_SDHI1_SET clock
+        * switching register is prohibited.
+        * The clock mux has 3 input clocks(533 MHz, 400 MHz, and 266 MHz), and
+        * the index to value mapping is done by adding 1 to the index.
+        */
+       bitmask = (GENMASK(GET_WIDTH(hwdata->conf) - 1, 0) << shift) << 16;
+       if (index != clk_src_266) {
+               u32 msk, val;
+               int ret;
+
+               writel(bitmask | ((clk_src_266 + 1) << shift), priv->base + off);
+
+               msk = off ? CPG_CLKSTATUS_SELSDHI1_STS : CPG_CLKSTATUS_SELSDHI0_STS;
+
+               ret = readl_poll_timeout(priv->base + CPG_CLKSTATUS, val,
+                                        !(val & msk), 100,
+                                        CPG_SDHI_CLK_SWITCH_STATUS_TIMEOUT_US);
+               if (ret) {
+                       dev_err(priv->dev, "failed to switch clk source\n");
+                       return ret;
+               }
+       }
+
+       writel(bitmask | ((index + 1) << shift), priv->base + off);
+
+       return 0;
+}
+
+static u8 rzg2l_cpg_sd_clk_mux_get_parent(struct clk_hw *hw)
+{
+       struct sd_hw_data *hwdata = to_sd_hw_data(hw);
+       struct rzg2l_cpg_priv *priv = hwdata->priv;
+       u32 val = readl(priv->base + GET_REG_OFFSET(hwdata->conf));
+
+       val >>= GET_SHIFT(hwdata->conf);
+       val &= GENMASK(GET_WIDTH(hwdata->conf) - 1, 0);
+       if (val) {
+               val--;
+       } else {
+               /* Prohibited clk source, change it to 533 MHz(reset value) */
+               rzg2l_cpg_sd_clk_mux_set_parent(hw, 0);
+       }
+
+       return val;
+}
+
+static const struct clk_ops rzg2l_cpg_sd_clk_mux_ops = {
+       .determine_rate = rzg2l_cpg_sd_clk_mux_determine_rate,
+       .set_parent     = rzg2l_cpg_sd_clk_mux_set_parent,
+       .get_parent     = rzg2l_cpg_sd_clk_mux_get_parent,
+};
+
+static struct clk * __init
+rzg2l_cpg_sd_mux_clk_register(const struct cpg_core_clk *core,
+                             void __iomem *base,
+                             struct rzg2l_cpg_priv *priv)
+{
+       struct sd_hw_data *clk_hw_data;
+       struct clk_init_data init;
+       struct clk_hw *clk_hw;
+       int ret;
+
+       clk_hw_data = devm_kzalloc(priv->dev, sizeof(*clk_hw_data), GFP_KERNEL);
+       if (!clk_hw_data)
+               return ERR_PTR(-ENOMEM);
+
+       clk_hw_data->priv = priv;
+       clk_hw_data->conf = core->conf;
+
+       init.name = GET_SHIFT(core->conf) ? "sd1" : "sd0";
+       init.ops = &rzg2l_cpg_sd_clk_mux_ops;
+       init.flags = 0;
+       init.num_parents = core->num_parents;
+       init.parent_names = core->parent_names;
+
+       clk_hw = &clk_hw_data->hw;
+       clk_hw->init = &init;
+
+       ret = devm_clk_hw_register(priv->dev, clk_hw);
+       if (ret)
+               return ERR_PTR(ret);
+
+       return clk_hw->clk;
+}
+
 struct pll_clk {
        struct clk_hw hw;
        unsigned int conf;
        case CLK_TYPE_MUX:
                clk = rzg2l_cpg_mux_clk_register(core, priv->base, priv);
                break;
+       case CLK_TYPE_SD_MUX:
+               clk = rzg2l_cpg_sd_mux_clk_register(core, priv->base, priv);
+               break;
        default:
                goto fail;
        }
 
 
 #define CPG_PL2_DDIV           (0x204)
 #define CPG_PL3A_DDIV          (0x208)
+#define CPG_CLKSTATUS          (0x280)
 #define CPG_PL3_SSEL           (0x408)
 #define CPG_PL6_ETH_SSEL       (0x418)
 
+#define CPG_CLKSTATUS_SELSDHI0_STS     BIT(28)
+#define CPG_CLKSTATUS_SELSDHI1_STS     BIT(29)
+
+#define CPG_SDHI_CLK_SWITCH_STATUS_TIMEOUT_US  20000
+
 /* n = 0/1/2 for PLL1/4/6 */
 #define CPG_SAMPLL_CLK1(n)     (0x04 + (16 * n))
 #define CPG_SAMPLL_CLK2(n)     (0x08 + (16 * n))
 
        /* Clock with clock source selector */
        CLK_TYPE_MUX,
+
+       /* Clock with SD clock source selector */
+       CLK_TYPE_SD_MUX,
 };
 
 #define DEF_TYPE(_name, _id, _type...) \
        DEF_TYPE(_name, _id, CLK_TYPE_MUX, .conf = _conf, \
                 .parent_names = _parent_names, .num_parents = _num_parents, \
                 .flag = _flag, .mux_flags = _mux_flags)
+#define DEF_SD_MUX(_name, _id, _conf, _parent_names, _num_parents) \
+       DEF_TYPE(_name, _id, CLK_TYPE_SD_MUX, .conf = _conf, \
+                .parent_names = _parent_names, .num_parents = _num_parents)
 
 /**
  * struct rzg2l_mod_clk - Module Clocks definitions