#define FLASH_TYPE_VAL                 0x18
 
 #define FLASH_SUBTYPE_REG              0x05
-#define FLASH_SUBTYPE_3CH_VAL          0x04
+#define FLASH_SUBTYPE_3CH_PM8150_VAL   0x04
+#define FLASH_SUBTYPE_3CH_PMI8998_VAL  0x03
 #define FLASH_SUBTYPE_4CH_VAL          0x07
 
 #define FLASH_STS_3CH_OTST1            BIT(0)
                return rc;
        }
 
-       if (val == FLASH_SUBTYPE_3CH_VAL) {
+       if (val == FLASH_SUBTYPE_3CH_PM8150_VAL || val == FLASH_SUBTYPE_3CH_PMI8998_VAL) {
                flash_data->hw_type = QCOM_MVFLASH_3CH;
                flash_data->max_channels = 3;
                regs = mvflash_3ch_regs;