hba->desc_size.hlth_desc = QUERY_DESC_HEALTH_DEF_SIZE;
 }
 
+static int ufshcd_device_geo_params_init(struct ufs_hba *hba)
+{
+       int err;
+       size_t buff_len;
+       u8 *desc_buf;
+
+       buff_len = hba->desc_size.geom_desc;
+       desc_buf = kmalloc(buff_len, GFP_KERNEL);
+       if (!desc_buf) {
+               err = -ENOMEM;
+               goto out;
+       }
+
+       err = ufshcd_read_desc(hba, QUERY_DESC_IDN_GEOMETRY, 0,
+                       desc_buf, buff_len);
+       if (err) {
+               dev_err(hba->dev, "%s: Failed reading Geometry Desc. err = %d\n",
+                               __func__, err);
+               goto out;
+       }
+
+       if (desc_buf[GEOMETRY_DESC_PARAM_MAX_NUM_LUN] == 1)
+               hba->dev_info.max_lu_supported = 32;
+       else if (desc_buf[GEOMETRY_DESC_PARAM_MAX_NUM_LUN] == 0)
+               hba->dev_info.max_lu_supported = 8;
+
+out:
+       kfree(desc_buf);
+       return err;
+}
+
 static struct ufs_ref_clk ufs_ref_clk_freqs[] = {
        {19200000, REF_CLK_FREQ_19_2_MHZ},
        {26000000, REF_CLK_FREQ_26_MHZ},
        bool flag;
        int ret;
 
+       /* Clear any previous UFS device information */
+       memset(&hba->dev_info, 0, sizeof(hba->dev_info));
+
        /* Init check for device descriptor sizes */
        ufshcd_init_desc_sizes(hba);
 
+       /* Init UFS geometry descriptor related parameters */
+       ret = ufshcd_device_geo_params_init(hba);
+       if (ret)
+               goto out;
+
        /* Check and apply UFS device quirks */
        ret = ufs_get_device_desc(hba);
        if (ret) {
 
        ufs_fixup_device_setup(hba);
 
-       /* Clear any previous UFS device information */
-       memset(&hba->dev_info, 0, sizeof(hba->dev_info));
        if (!ufshcd_query_flag_retry(hba, UPIU_QUERY_OPCODE_READ_FLAG,
                        QUERY_FLAG_IDN_PWR_ON_WPE, &flag))
                hba->dev_info.f_power_on_wp_en = flag;