*/
 
 #include <linux/clk.h>
+#include <linux/clk-provider.h>
 #include <linux/io.h>
 #include <linux/iopoll.h>
 #include <linux/kernel.h>
 #include <linux/of_device.h>
 #include <linux/platform_device.h>
 #include <linux/pm_wakeirq.h>
+#include <linux/property.h>
 #include <linux/reboot.h>
 #include <linux/rtc.h>
 #include <linux/slab.h>
 #define JZ_REG_RTC_WAKEUP_FILTER       0x24
 #define JZ_REG_RTC_RESET_COUNTER       0x28
 #define JZ_REG_RTC_SCRATCHPAD  0x34
+#define JZ_REG_RTC_CKPCR       0x40
 
 /* The following are present on the jz4780 */
 #define JZ_REG_RTC_WENR        0x3C
 #define JZ_RTC_WAKEUP_FILTER_MASK      0x0000FFE0
 #define JZ_RTC_RESET_COUNTER_MASK      0x00000FE0
 
+#define JZ_RTC_CKPCR_CK32PULL_DIS      BIT(4)
+#define JZ_RTC_CKPCR_CK32CTL_EN                (BIT(2) | BIT(1))
+
 enum jz4740_rtc_type {
        ID_JZ4740,
        ID_JZ4760,
 
        struct rtc_device *rtc;
 
+       struct clk_hw clk32k;
+
        spinlock_t lock;
 };
 
 static const struct of_device_id jz4740_rtc_of_match[] = {
        { .compatible = "ingenic,jz4740-rtc", .data = (void *)ID_JZ4740 },
        { .compatible = "ingenic,jz4760-rtc", .data = (void *)ID_JZ4760 },
+       { .compatible = "ingenic,jz4770-rtc", .data = (void *)ID_JZ4780 },
        { .compatible = "ingenic,jz4780-rtc", .data = (void *)ID_JZ4780 },
        {},
 };
        jz4740_rtc_reg_write(rtc, JZ_REG_RTC_RESET_COUNTER, reset_ticks);
 }
 
+static int jz4740_rtc_clk32k_enable(struct clk_hw *hw)
+{
+       struct jz4740_rtc *rtc = container_of(hw, struct jz4740_rtc, clk32k);
+
+       return jz4740_rtc_reg_write(rtc, JZ_REG_RTC_CKPCR,
+                                   JZ_RTC_CKPCR_CK32PULL_DIS |
+                                   JZ_RTC_CKPCR_CK32CTL_EN);
+}
+
+static void jz4740_rtc_clk32k_disable(struct clk_hw *hw)
+{
+       struct jz4740_rtc *rtc = container_of(hw, struct jz4740_rtc, clk32k);
+
+       jz4740_rtc_reg_write(rtc, JZ_REG_RTC_CKPCR, 0);
+}
+
+static int jz4740_rtc_clk32k_is_enabled(struct clk_hw *hw)
+{
+       struct jz4740_rtc *rtc = container_of(hw, struct jz4740_rtc, clk32k);
+       u32 ckpcr;
+
+       ckpcr = jz4740_rtc_reg_read(rtc, JZ_REG_RTC_CKPCR);
+
+       return !!(ckpcr & JZ_RTC_CKPCR_CK32CTL_EN);
+}
+
+static const struct clk_ops jz4740_rtc_clk32k_ops = {
+       .enable = jz4740_rtc_clk32k_enable,
+       .disable = jz4740_rtc_clk32k_disable,
+       .is_enabled = jz4740_rtc_clk32k_is_enabled,
+};
+
 static int jz4740_rtc_probe(struct platform_device *pdev)
 {
        struct device *dev = &pdev->dev;
                        dev_warn(dev, "Poweroff handler already present!\n");
        }
 
+       if (device_property_present(dev, "#clock-cells")) {
+               rtc->clk32k.init = CLK_HW_INIT_HW("clk32k", __clk_get_hw(clk),
+                                                 &jz4740_rtc_clk32k_ops, 0);
+
+               ret = devm_clk_hw_register(dev, &rtc->clk32k);
+               if (ret)
+                       return dev_err_probe(dev, ret,
+                                            "Unable to register clk32k clock\n");
+
+               ret = of_clk_add_hw_provider(np, of_clk_hw_simple_get, &rtc->clk32k);
+               if (ret)
+                       return dev_err_probe(dev, ret,
+                                            "Unable to register clk32k clock provider\n");
+       }
+
        return 0;
 }