spin_unlock(&dq->lock);
 }
 
-static int hisi_sas_task_exec(struct sas_task *task, gfp_t gfp_flags,
-                             struct sas_tmf_task *tmf)
+static int hisi_sas_queue_command(struct sas_task *task, gfp_t gfp_flags)
 {
        int n_elem = 0, n_elem_dif = 0, n_elem_req = 0;
        struct domain_device *device = task->dev;
        slot->task = task;
        slot->port = port;
 
-       slot->tmf = tmf;
-       slot->is_internal = tmf;
+       slot->tmf = task->tmf;
+       slot->is_internal = task->tmf;
 
        /* protect task_prep and start_delivery sequence */
        hisi_sas_task_deliver(hisi_hba, slot, dq, sas_dev, NULL);
        up(&hisi_hba->sem);
 }
 
-static int hisi_sas_queue_command(struct sas_task *task, gfp_t gfp_flags)
-{
-       return hisi_sas_task_exec(task, gfp_flags, NULL);
-}
-
 static int hisi_sas_phy_set_linkrate(struct hisi_hba *hisi_hba, int phy_no,
                        struct sas_phy_linkrates *r)
 {
                task->slow_task->timer.expires = jiffies + TASK_TIMEOUT;
                add_timer(&task->slow_task->timer);
 
-               res = hisi_sas_task_exec(task, GFP_KERNEL, tmf);
+               task->tmf = tmf;
+
+               res = hisi_sas_queue_command(task, GFP_KERNEL);
                if (res) {
                        del_timer_sync(&task->slow_task->timer);
                        dev_err(dev, "abort tmf: executing internal task failed: %d\n",
 
        return rc;
 }
 
-static int mvs_task_exec(struct sas_task *task, gfp_t gfp_flags,
-                               struct completion *completion, int is_tmf,
-                               struct sas_tmf_task *tmf)
+int mvs_queue_command(struct sas_task *task, gfp_t gfp_flags)
 {
        struct mvs_info *mvi = NULL;
        u32 rc = 0;
        u32 pass = 0;
        unsigned long flags = 0;
+       struct sas_tmf_task *tmf = task->tmf;
+       int is_tmf = !!task->tmf;
 
        mvi = ((struct mvs_device *)task->dev->lldd_dev)->mvi_info;
 
        return rc;
 }
 
-int mvs_queue_command(struct sas_task *task, gfp_t gfp_flags)
-{
-       return mvs_task_exec(task, gfp_flags, NULL, 0, NULL);
-}
-
 static void mvs_slot_free(struct mvs_info *mvi, u32 rx_desc)
 {
        u32 slot_idx = rx_desc & RXQ_SLOT_MASK;
                task->slow_task->timer.expires = jiffies + MVS_TASK_TIMEOUT*HZ;
                add_timer(&task->slow_task->timer);
 
-               res = mvs_task_exec(task, GFP_KERNEL, NULL, 1, tmf);
+               task->tmf = tmf;
+
+               res = mvs_queue_command(task, GFP_KERNEL);
 
                if (res) {
                        del_timer(&task->slow_task->timer);
 
 
 #define DEV_IS_GONE(pm8001_dev)        \
        ((!pm8001_dev || (pm8001_dev->dev_type == SAS_PHY_UNUSED)))
+
 /**
-  * pm8001_task_exec - queue the task(ssp, smp && ata) to the hardware.
+  * pm8001_queue_command - register for upper layer used, all IO commands sent
+  * to HBA are from this interface.
   * @task: the task to be execute.
-  * @gfp_flags: gfp_flags.
-  * @is_tmf: if it is task management task.
-  * @tmf: the task management IU
+  * @gfp_flags: gfp_flags
   */
-static int pm8001_task_exec(struct sas_task *task,
-       gfp_t gfp_flags, int is_tmf, struct sas_tmf_task *tmf)
+int pm8001_queue_command(struct sas_task *task, gfp_t gfp_flags)
 {
        struct domain_device *dev = task->dev;
        struct pm8001_hba_info *pm8001_ha;
        u32 tag = 0xdeadbeef, rc = 0, n_elem = 0;
        unsigned long flags = 0;
        enum sas_protocol task_proto = t->task_proto;
+       struct sas_tmf_task *tmf = task->tmf;
+       int is_tmf = !!task->tmf;
 
        if (!dev->port) {
                struct task_status_struct *tsm = &t->task_status;
        return rc;
 }
 
-/**
-  * pm8001_queue_command - register for upper layer used, all IO commands sent
-  * to HBA are from this interface.
-  * @task: the task to be execute.
-  * @gfp_flags: gfp_flags
-  */
-int pm8001_queue_command(struct sas_task *task, gfp_t gfp_flags)
-{
-       return pm8001_task_exec(task, gfp_flags, 0, NULL);
-}
-
 /**
   * pm8001_ccb_task_free - free the sg for ssp and smp command, free the ccb.
   * @pm8001_ha: our hba card information
                task->slow_task->timer.expires = jiffies + PM8001_TASK_TIMEOUT*HZ;
                add_timer(&task->slow_task->timer);
 
-               res = pm8001_task_exec(task, GFP_KERNEL, 1, tmf);
+               task->tmf = tmf;
+
+               res = pm8001_queue_command(task, GFP_KERNEL);
 
                if (res) {
                        del_timer(&task->slow_task->timer);
 
        void   *lldd_task;        /* for use by LLDDs */
        void   *uldd_task;
        struct sas_task_slow *slow_task;
+       struct sas_tmf_task *tmf;
 };
 
 struct sas_task_slow {