return readl(regs);
 }
 
+/* This function needs to be protected from pre-emption. */
+static int
+slot_index_alloc_quirk_v2_hw(struct hisi_hba *hisi_hba, int *slot_idx,
+                      struct domain_device *device)
+{
+       unsigned int index = 0;
+       void *bitmap = hisi_hba->slot_index_tags;
+       int sata_dev = dev_is_sata(device);
+
+       while (1) {
+               index = find_next_zero_bit(bitmap, hisi_hba->slot_index_count,
+                                          index);
+               if (index >= hisi_hba->slot_index_count)
+                       return -SAS_QUEUE_FULL;
+               /*
+                * SAS IPTT bit0 should be 1
+                */
+               if (sata_dev || (index & 1))
+                       break;
+               index++;
+       }
+
+       set_bit(index, bitmap);
+       *slot_idx = index;
+       return 0;
+}
+
 static void config_phy_opt_mode_v2_hw(struct hisi_hba *hisi_hba, int phy_no)
 {
        u32 cfg = hisi_sas_phy_read32(hisi_hba, phy_no, PHY_CFG);
 static const struct hisi_sas_hw hisi_sas_v2_hw = {
        .hw_init = hisi_sas_v2_init,
        .setup_itct = setup_itct_v2_hw,
+       .slot_index_alloc = slot_index_alloc_quirk_v2_hw,
        .sl_notify = sl_notify_v2_hw,
        .get_wideport_bitmap = get_wideport_bitmap_v2_hw,
        .free_device = free_device_v2_hw,