}
 
 static void hisi_sas_task_prep_abort(struct hisi_hba *hisi_hba,
-               struct hisi_sas_slot *slot,
-               int device_id, int abort_flag, int tag_to_abort)
+               struct hisi_sas_internal_abort *abort,
+               struct hisi_sas_slot *slot, int device_id)
 {
        hisi_hba->hw->prep_abort(hisi_hba, slot,
-                       device_id, abort_flag, tag_to_abort);
+                       device_id, abort->flag, abort->tag);
 }
 
 static void hisi_sas_dma_unmap(struct hisi_hba *hisi_hba,
 
 static int
 hisi_sas_internal_abort_task_exec(struct hisi_hba *hisi_hba, int device_id,
-                                 struct sas_task *task, int abort_flag,
-                                 int task_tag, struct hisi_sas_dq *dq)
+                                 struct hisi_sas_internal_abort *abort,
+                                 struct sas_task *task,
+                                 struct hisi_sas_dq *dq)
 {
        struct domain_device *device = task->dev;
        struct hisi_sas_device *sas_dev = device->lldd_dev;
        memset(hisi_sas_status_buf_addr_mem(slot), 0,
               sizeof(struct hisi_sas_err_record));
 
-       hisi_sas_task_prep_abort(hisi_hba, slot, device_id,
-                                     abort_flag, task_tag);
+       hisi_sas_task_prep_abort(hisi_hba, abort, slot, device_id);
 
        spin_lock_irqsave(&task->task_state_lock, flags);
        task->task_state_flags |= SAS_TASK_AT_INITIATOR;
 {
        struct sas_task *task;
        struct hisi_sas_device *sas_dev = device->lldd_dev;
+       struct hisi_sas_internal_abort abort = {
+               .flag = abort_flag,
+               .tag = tag,
+       };
        struct device *dev = hisi_hba->dev;
        int res;
-
        /*
         * The interface is not realized means this HW don't support internal
         * abort, or don't need to do internal abort. Then here, we return
        add_timer(&task->slow_task->timer);
 
        res = hisi_sas_internal_abort_task_exec(hisi_hba, sas_dev->device_id,
-                                               task, abort_flag, tag, dq);
+                                               &abort, task, dq);
        if (res) {
                del_timer_sync(&task->slow_task->timer);
                dev_err(dev, "internal task abort: executing internal task failed: %d\n",