blk-mq: fix completion byte stats with scheduling

Signed-off-by: Jens Axboe <axboe@fb.com>

diff --git a/block/blk-mq.c b/block/blk-mq.c
index cc1c70330962d067906430873ec7626836416c39..d9c3c6622ca5fb59266304f4051ed932e8c6aed7 100644 (file)
--- a/block/blk-mq.c
+++ b/block/blk-mq.c
@@ -877,6 +877,8 @@ static void sched_rq_end_io(struct request *rq, int error)
        /* transfer queue ref to 'rq', blk_mq_free_request() drops it */
        sched_rq->rq_flags &= ~RQF_HAS_Q_REF;
 
+       blk_account_io_completion(sched_rq, blk_rq_bytes(sched_rq));
+
        spin_lock_irqsave(q->queue_lock, flags);
        blk_finish_request(sched_rq, error);
        spin_unlock_irqrestore(q->queue_lock, flags);