#define DMC_MAX_CHANNELS       2
 
+#define HIWORD_UPDATE(val, mask)       ((val) | (mask) << 16)
+
 /* DDRMON_CTRL */
 #define DDRMON_CTRL    0x04
-#define CLR_DDRMON_CTRL        (0x1f0000 << 0)
-#define LPDDR4_EN      (0x10001 << 4)
-#define HARDWARE_EN    (0x10001 << 3)
-#define LPDDR3_EN      (0x10001 << 2)
-#define SOFTWARE_EN    (0x10001 << 1)
-#define SOFTWARE_DIS   (0x10000 << 1)
-#define TIME_CNT_EN    (0x10001 << 0)
+#define DDRMON_CTRL_DDR4               BIT(5)
+#define DDRMON_CTRL_LPDDR4             BIT(4)
+#define DDRMON_CTRL_HARDWARE_EN                BIT(3)
+#define DDRMON_CTRL_LPDDR23            BIT(2)
+#define DDRMON_CTRL_SOFTWARE_EN                BIT(1)
+#define DDRMON_CTRL_TIMER_CNT_EN       BIT(0)
+#define DDRMON_CTRL_DDR_TYPE_MASK      (DDRMON_CTRL_DDR4 | \
+                                        DDRMON_CTRL_LPDDR4 | \
+                                        DDRMON_CTRL_LPDDR23)
 
 #define DDRMON_CH0_COUNT_NUM           0x28
 #define DDRMON_CH0_DFI_ACCESS_NUM      0x2c
        void __iomem *dfi_regs = dfi->regs;
 
        /* clear DDRMON_CTRL setting */
-       writel_relaxed(CLR_DDRMON_CTRL, dfi_regs + DDRMON_CTRL);
+       writel_relaxed(HIWORD_UPDATE(0, DDRMON_CTRL_TIMER_CNT_EN | DDRMON_CTRL_SOFTWARE_EN |
+                      DDRMON_CTRL_HARDWARE_EN), dfi_regs + DDRMON_CTRL);
 
        /* set ddr type to dfi */
        if (dfi->ddr_type == ROCKCHIP_DDRTYPE_LPDDR3)
-               writel_relaxed(LPDDR3_EN, dfi_regs + DDRMON_CTRL);
+               writel_relaxed(HIWORD_UPDATE(DDRMON_CTRL_LPDDR23, DDRMON_CTRL_DDR_TYPE_MASK),
+                              dfi_regs + DDRMON_CTRL);
        else if (dfi->ddr_type == ROCKCHIP_DDRTYPE_LPDDR4)
-               writel_relaxed(LPDDR4_EN, dfi_regs + DDRMON_CTRL);
+               writel_relaxed(HIWORD_UPDATE(DDRMON_CTRL_LPDDR4, DDRMON_CTRL_DDR_TYPE_MASK),
+                              dfi_regs + DDRMON_CTRL);
 
        /* enable count, use software mode */
-       writel_relaxed(SOFTWARE_EN, dfi_regs + DDRMON_CTRL);
+       writel_relaxed(HIWORD_UPDATE(DDRMON_CTRL_SOFTWARE_EN, DDRMON_CTRL_SOFTWARE_EN),
+                      dfi_regs + DDRMON_CTRL);
 }
 
 static void rockchip_dfi_stop_hardware_counter(struct devfreq_event_dev *edev)
        struct rockchip_dfi *dfi = devfreq_event_get_drvdata(edev);
        void __iomem *dfi_regs = dfi->regs;
 
-       writel_relaxed(SOFTWARE_DIS, dfi_regs + DDRMON_CTRL);
+       writel_relaxed(HIWORD_UPDATE(0, DDRMON_CTRL_SOFTWARE_EN),
+                      dfi_regs + DDRMON_CTRL);
 }
 
 static void rockchip_dfi_read_counters(struct devfreq_event_dev *edev, struct dmc_count *count)