return IRQ_HANDLED;
 }
 
-static void ufs_qcom_irq_free(struct ufs_qcom_irq *uqi)
-{
-       for (struct ufs_qcom_irq *q = uqi; q->irq; q++)
-               devm_free_irq(q->hba->dev, q->irq, q->hba);
-
-       platform_device_msi_free_irqs_all(uqi->hba->dev);
-       devm_kfree(uqi->hba->dev, uqi);
-}
-
-DEFINE_FREE(ufs_qcom_irq, struct ufs_qcom_irq *, if (_T) ufs_qcom_irq_free(_T))
-
 static int ufs_qcom_config_esi(struct ufs_hba *hba)
 {
        struct ufs_qcom_host *host = ufshcd_get_variant(hba);
         */
        nr_irqs = hba->nr_hw_queues - hba->nr_queues[HCTX_TYPE_POLL];
 
-       struct ufs_qcom_irq *qi __free(ufs_qcom_irq) =
-               devm_kcalloc(hba->dev, nr_irqs, sizeof(*qi), GFP_KERNEL);
-       if (!qi)
-               return -ENOMEM;
-       /* Preset so __free() has a pointer to hba in all error paths */
-       qi[0].hba = hba;
-
        ret = platform_device_msi_init_and_alloc_irqs(hba->dev, nr_irqs,
                                                      ufs_qcom_write_msi_msg);
        if (ret) {
-               dev_err(hba->dev, "Failed to request Platform MSI %d\n", ret);
-               return ret;
+               dev_warn(hba->dev, "Platform MSI not supported or failed, continuing without ESI\n");
+               return ret; /* Continue without ESI */
+       }
+
+       struct ufs_qcom_irq *qi = devm_kcalloc(hba->dev, nr_irqs, sizeof(*qi), GFP_KERNEL);
+
+       if (!qi) {
+               platform_device_msi_free_irqs_all(hba->dev);
+               return -ENOMEM;
        }
 
        for (int idx = 0; idx < nr_irqs; idx++) {
                ret = devm_request_irq(hba->dev, qi[idx].irq, ufs_qcom_mcq_esi_handler,
                                       IRQF_SHARED, "qcom-mcq-esi", qi + idx);
                if (ret) {
-                       dev_err(hba->dev, "%s: Fail to request IRQ for %d, err = %d\n",
+                       dev_err(hba->dev, "%s: Failed to request IRQ for %d, err = %d\n",
                                __func__, qi[idx].irq, ret);
-                       qi[idx].irq = 0;
+                       /* Free previously allocated IRQs */
+                       for (int j = 0; j < idx; j++)
+                               devm_free_irq(hba->dev, qi[j].irq, qi + j);
+                       platform_device_msi_free_irqs_all(hba->dev);
+                       devm_kfree(hba->dev, qi);
                        return ret;
                }
        }
 
-       retain_and_null_ptr(qi);
-
        if (host->hw_ver.major >= 6) {
                ufshcd_rmwl(hba, ESI_VEC_MASK, FIELD_PREP(ESI_VEC_MASK, MAX_ESI_VEC - 1),
                            REG_UFS_CFG3);