3 files changed, 9 insertions, 6 deletions

diff --git a/block/blk-mq-sched.c b/block/blk-mq-sched.c
index 72d0d8361175..97fe904f0a04 100644
--- a/block/blk-mq-sched.c
+++ b/block/blk-mq-sched.c
@@ -68,7 +68,9 @@ error:
 EXPORT_SYMBOL_GPL(blk_mq_sched_init_hctx_data);
 
 static void __blk_mq_sched_assign_ioc(struct request_queue *q,
-                                      struct request *rq, struct io_context *ioc)
+                                      struct request *rq,
+                                      struct bio *bio,
+                                      struct io_context *ioc)
 {
         struct io_cq *icq;
 
@@ -83,7 +85,7 @@ static void __blk_mq_sched_assign_ioc(struct request_queue *q,
         }
 
         rq->elv.icq = icq;
-        if (!blk_mq_sched_get_rq_priv(q, rq)) {
+        if (!blk_mq_sched_get_rq_priv(q, rq, bio)) {
                 rq->rq_flags |= RQF_ELVPRIV;
                 get_io_context(icq->ioc);
                 return;
@@ -99,7 +101,7 @@ static void blk_mq_sched_assign_ioc(struct request_queue *q,
 
         ioc = rq_ioc(bio);
         if (ioc)
-                __blk_mq_sched_assign_ioc(q, rq, ioc);
+                __blk_mq_sched_assign_ioc(q, rq, bio, ioc);
 }
 
 struct request *blk_mq_sched_get_request(struct request_queue *q,
diff --git a/block/blk-mq-sched.h b/block/blk-mq-sched.h
index 5954859c8670..7b5f3b95c78e 100644
--- a/block/blk-mq-sched.h
+++ b/block/blk-mq-sched.h
@@ -49,12 +49,13 @@ blk_mq_sched_bio_merge(struct request_queue *q, struct bio *bio)
 }
 
 static inline int blk_mq_sched_get_rq_priv(struct request_queue *q,
-                                           struct request *rq)
+                                           struct request *rq,
+                                           struct bio *bio)
 {
         struct elevator_queue *e = q->elevator;
 
         if (e && e->type->ops.mq.get_rq_priv)
-                return e->type->ops.mq.get_rq_priv(q, rq);
+                return e->type->ops.mq.get_rq_priv(q, rq, bio);
 
         return 0;
 }
diff --git a/include/linux/elevator.h b/include/linux/elevator.h
index 8265b6330cc2..aebecc4ed088 100644
--- a/include/linux/elevator.h
+++ b/include/linux/elevator.h
@@ -109,7 +109,7 @@ struct elevator_mq_ops {
         void (*requeue_request)(struct request *);
         struct request *(*former_request)(struct request_queue *, struct request *);
         struct request *(*next_request)(struct request_queue *, struct request *);
-        int (*get_rq_priv)(struct request_queue *, struct request *);
+        int (*get_rq_priv)(struct request_queue *, struct request *, struct bio *);
         void (*put_rq_priv)(struct request_queue *, struct request *);
         void (*init_icq)(struct io_cq *);
         void (*exit_icq)(struct io_cq *);