static void free_device_v2_hw(struct hisi_hba *hisi_hba,
                              struct hisi_sas_device *sas_dev)
 {
+       DECLARE_COMPLETION_ONSTACK(completion);
        u64 dev_id = sas_dev->device_id;
-       struct device *dev = hisi_hba->dev;
        struct hisi_sas_itct *itct = &hisi_hba->itct[dev_id];
        u32 reg_val = hisi_sas_read32(hisi_hba, ENT_INT_SRC3);
        int i;
 
+       sas_dev->completion = &completion;
+
        /* SoC bug workaround */
        if (dev_is_sata(sas_dev->sas_device))
                clear_bit(sas_dev->sata_idx, hisi_hba->sata_dev_bitmap);
                hisi_sas_write32(hisi_hba, ENT_INT_SRC3,
                                 ENT_INT_SRC3_ITC_INT_MSK);
 
-       /* clear the itct int*/
        for (i = 0; i < 2; i++) {
-               /* clear the itct table*/
-               reg_val = hisi_sas_read32(hisi_hba, ITCT_CLR);
-               reg_val |= ITCT_CLR_EN_MSK | (dev_id & ITCT_DEV_MSK);
+               reg_val = ITCT_CLR_EN_MSK | (dev_id & ITCT_DEV_MSK);
                hisi_sas_write32(hisi_hba, ITCT_CLR, reg_val);
+               wait_for_completion(sas_dev->completion);
 
-               udelay(10);
-               reg_val = hisi_sas_read32(hisi_hba, ENT_INT_SRC3);
-               if (ENT_INT_SRC3_ITC_INT_MSK & reg_val) {
-                       dev_dbg(dev, "got clear ITCT done interrupt\n");
-
-                       /* invalid the itct state*/
-                       memset(itct, 0, sizeof(struct hisi_sas_itct));
-                       hisi_sas_write32(hisi_hba, ENT_INT_SRC3,
-                                        ENT_INT_SRC3_ITC_INT_MSK);
-
-                       /* clear the itct */
-                       hisi_sas_write32(hisi_hba, ITCT_CLR, 0);
-                       dev_dbg(dev, "clear ITCT ok\n");
-                       break;
-               }
+               memset(itct, 0, sizeof(struct hisi_sas_itct));
        }
 }
 
        hisi_sas_write32(hisi_hba, ENT_INT_SRC3, 0xffffffff);
        hisi_sas_write32(hisi_hba, ENT_INT_SRC_MSK1, 0x7efefefe);
        hisi_sas_write32(hisi_hba, ENT_INT_SRC_MSK2, 0x7efefefe);
-       hisi_sas_write32(hisi_hba, ENT_INT_SRC_MSK3, 0x7ffffffe);
+       hisi_sas_write32(hisi_hba, ENT_INT_SRC_MSK3, 0x7ffe20fe);
        hisi_sas_write32(hisi_hba, SAS_ECC_INTR_MSK, 0xfff00c30);
        for (i = 0; i < hisi_hba->queue_count; i++)
                hisi_sas_write32(hisi_hba, OQ0_INT_SRC_MSK+0x4*i, 0);
                              irq_value);
                        queue_work(hisi_hba->wq, &hisi_hba->rst_work);
                }
+
+               if (irq_value & BIT(ENT_INT_SRC3_ITC_INT_OFF)) {
+                       u32 reg_val = hisi_sas_read32(hisi_hba, ITCT_CLR);
+                       u32 dev_id = reg_val & ITCT_DEV_MSK;
+                       struct hisi_sas_device *sas_dev =
+                                       &hisi_hba->devices[dev_id];
+
+                       hisi_sas_write32(hisi_hba, ITCT_CLR, 0);
+                       dev_dbg(dev, "clear ITCT ok\n");
+                       complete(sas_dev->completion);
+               }
        }
 
+       hisi_sas_write32(hisi_hba, ENT_INT_SRC3, irq_value);
        hisi_sas_write32(hisi_hba, ENT_INT_SRC_MSK3, irq_msk);
 
        return IRQ_HANDLED;