#define MTK_KPD_DEBOUNCE       0x0018
 #define MTK_KPD_DEBOUNCE_MASK  GENMASK(13, 0)
 #define MTK_KPD_DEBOUNCE_MAX_MS        256
+#define MTK_KPD_SEL            0x0020
+#define MTK_KPD_SEL_COL        GENMASK(15, 10)
+#define MTK_KPD_SEL_ROW        GENMASK(9, 4)
+#define MTK_KPD_SEL_COLMASK(c) GENMASK((c) + 9, 10)
+#define MTK_KPD_SEL_ROWMASK(r) GENMASK((r) + 3, 4)
 #define MTK_KPD_NUM_MEMS       5
 #define MTK_KPD_NUM_BITS       136     /* 4*32+8 MEM5 only use 8 BITS */
 
        regmap_write(keypad->regmap, MTK_KPD_DEBOUNCE,
                     (debounce * (1 << 5)) & MTK_KPD_DEBOUNCE_MASK);
 
+       regmap_update_bits(keypad->regmap, MTK_KPD_SEL, MTK_KPD_SEL_ROW,
+                          MTK_KPD_SEL_ROWMASK(keypad->n_rows));
+       regmap_update_bits(keypad->regmap, MTK_KPD_SEL, MTK_KPD_SEL_COL,
+                          MTK_KPD_SEL_COLMASK(keypad->n_cols));
+
        keypad->clk = devm_clk_get(&pdev->dev, "kpd");
        if (IS_ERR(keypad->clk))
                return PTR_ERR(keypad->clk);