}
 
 static inline struct scsi_cmnd *scsi_mq_find_tag(struct Scsi_Host *shost,
-               unsigned int hw_ctx, int tag)
+                                                int unique_tag)
 {
-       struct request *req;
+       u16 hwq = blk_mq_unique_tag_to_hwq(unique_tag);
+       struct request *req = NULL;
 
-       req = blk_mq_tag_to_rq(shost->tag_set.tags[hw_ctx], tag);
+       if (hwq < shost->tag_set.nr_hw_queues)
+               req = blk_mq_tag_to_rq(shost->tag_set.tags[hwq],
+                                      blk_mq_unique_tag_to_tag(unique_tag));
        return req ? (struct scsi_cmnd *)req->special : NULL;
 }
 
 /**
  * scsi_find_tag - find a tagged command by device
  * @SDpnt:     pointer to the ScSI device
- * @tag:       the tag number
+ * @tag:       tag generated by blk_mq_unique_tag()
  *
  * Notes:
  *     Only works with tags allocated by the generic blk layer.
 
         if (tag != SCSI_NO_TAG) {
                if (shost_use_blk_mq(sdev->host))
-                       return scsi_mq_find_tag(sdev->host, 0, tag);
+                       return scsi_mq_find_tag(sdev->host, tag);
 
-               req = blk_queue_find_tag(sdev->request_queue, tag);
+               req = blk_queue_find_tag(sdev->request_queue, tag);
                return req ? (struct scsi_cmnd *)req->special : NULL;
        }
 
 /**
  * scsi_host_find_tag - find the tagged command by host
  * @shost:     pointer to scsi_host
- * @tag:       tag of the scsi_cmnd
+ * @tag:       tag generated by blk_mq_unique_tag()
  *
  * Notes:
  *     Only works with tags allocated by the generic blk layer.
 
        if (tag != SCSI_NO_TAG) {
                if (shost_use_blk_mq(shost))
-                       return scsi_mq_find_tag(shost, 0, tag);
+                       return scsi_mq_find_tag(shost, tag);
                req = blk_map_queue_find_tag(shost->bqt, tag);
                return req ? (struct scsi_cmnd *)req->special : NULL;
        }