diff mbox series

blk-mq: Use request queue-wide tags for tagset-wide sbitmap

Message ID 1620037333-2495-1-git-send-email-john.garry@huawei.com
State New
Headers show
Series blk-mq: Use request queue-wide tags for tagset-wide sbitmap | expand

Commit Message

John Garry May 3, 2021, 10:22 a.m. UTC
The tags used for an IO scheduler are currently per hctx.

As such, when q->nr_hw_queues grows, so does the request queue total IO
scheduler tag depth.

This may cause problems for SCSI MQ HBAs whose total driver depth is
fixed.

Ming and Yanhui report higher CPU usage and lower throughput in scenarios
where the fixed total driver tag depth is appreciably lower than the total
scheduler tag depth:
https://lore.kernel.org/linux-block/440dfcfc-1a2c-bd98-1161-cec4d78c6dfc@huawei.com/T/#mc0d6d4f95275a2743d1c8c3e4dc9ff6c9aa3a76b

In that scenario, since the scheduler tag is got first, much contention
is introduced since a driver tag may not be available after we have got
the sched tag.

Improve this scenario by introducing request queue-wide tags for when
a tagset-wide sbitmap is used. The static sched requests are still
allocated per hctx, as requests are initialised per hctx, as in
blk_mq_init_request(..., hctx_idx, ...) ->
set->ops->init_request(.., hctx_idx, ...).

For simplicity of resizing the request queue sbitmap when updating the
request queue depth, just init at the max possible size, so we don't need
to deal with the possibly with swapping out a new sbitmap for old if
we need to grow.

Signed-off-by: John Garry <john.garry@huawei.com>


-- 
2.26.2

Comments

Ming Lei May 6, 2021, 8:32 a.m. UTC | #1
On Mon, May 03, 2021 at 06:22:13PM +0800, John Garry wrote:
> The tags used for an IO scheduler are currently per hctx.

> 

> As such, when q->nr_hw_queues grows, so does the request queue total IO

> scheduler tag depth.

> 

> This may cause problems for SCSI MQ HBAs whose total driver depth is

> fixed.

> 

> Ming and Yanhui report higher CPU usage and lower throughput in scenarios

> where the fixed total driver tag depth is appreciably lower than the total

> scheduler tag depth:

> https://lore.kernel.org/linux-block/440dfcfc-1a2c-bd98-1161-cec4d78c6dfc@huawei.com/T/#mc0d6d4f95275a2743d1c8c3e4dc9ff6c9aa3a76b

> 

> In that scenario, since the scheduler tag is got first, much contention

> is introduced since a driver tag may not be available after we have got

> the sched tag.

> 

> Improve this scenario by introducing request queue-wide tags for when

> a tagset-wide sbitmap is used. The static sched requests are still

> allocated per hctx, as requests are initialised per hctx, as in

> blk_mq_init_request(..., hctx_idx, ...) ->

> set->ops->init_request(.., hctx_idx, ...).

> 

> For simplicity of resizing the request queue sbitmap when updating the

> request queue depth, just init at the max possible size, so we don't need

> to deal with the possibly with swapping out a new sbitmap for old if

> we need to grow.

> 

> Signed-off-by: John Garry <john.garry@huawei.com>

> 

> diff --git a/block/blk-mq-sched.c b/block/blk-mq-sched.c

> index e1e997af89a0..121207abc026 100644

> --- a/block/blk-mq-sched.c

> +++ b/block/blk-mq-sched.c

> @@ -497,11 +497,9 @@ static void blk_mq_sched_free_tags(struct blk_mq_tag_set *set,

>  				   struct blk_mq_hw_ctx *hctx,

>  				   unsigned int hctx_idx)

>  {

> -	unsigned int flags = set->flags & ~BLK_MQ_F_TAG_HCTX_SHARED;

> -

>  	if (hctx->sched_tags) {

>  		blk_mq_free_rqs(set, hctx->sched_tags, hctx_idx);

> -		blk_mq_free_rq_map(hctx->sched_tags, flags);

> +		blk_mq_free_rq_map(hctx->sched_tags, set->flags);

>  		hctx->sched_tags = NULL;

>  	}

>  }

> @@ -511,12 +509,10 @@ static int blk_mq_sched_alloc_tags(struct request_queue *q,

>  				   unsigned int hctx_idx)

>  {

>  	struct blk_mq_tag_set *set = q->tag_set;

> -	/* Clear HCTX_SHARED so tags are init'ed */

> -	unsigned int flags = set->flags & ~BLK_MQ_F_TAG_HCTX_SHARED;

>  	int ret;

>  

>  	hctx->sched_tags = blk_mq_alloc_rq_map(set, hctx_idx, q->nr_requests,

> -					       set->reserved_tags, flags);

> +					       set->reserved_tags, set->flags);

>  	if (!hctx->sched_tags)

>  		return -ENOMEM;

>  

> @@ -534,11 +530,8 @@ static void blk_mq_sched_tags_teardown(struct request_queue *q)

>  	int i;

>  

>  	queue_for_each_hw_ctx(q, hctx, i) {

> -		/* Clear HCTX_SHARED so tags are freed */

> -		unsigned int flags = hctx->flags & ~BLK_MQ_F_TAG_HCTX_SHARED;

> -

>  		if (hctx->sched_tags) {

> -			blk_mq_free_rq_map(hctx->sched_tags, flags);

> +			blk_mq_free_rq_map(hctx->sched_tags, hctx->flags);

>  			hctx->sched_tags = NULL;

>  		}

>  	}

> @@ -568,12 +561,25 @@ int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)

>  	queue_for_each_hw_ctx(q, hctx, i) {

>  		ret = blk_mq_sched_alloc_tags(q, hctx, i);

>  		if (ret)

> -			goto err;

> +			goto err_free_tags;

> +	}

> +

> +	if (blk_mq_is_sbitmap_shared(q->tag_set->flags)) {

> +		ret = blk_mq_init_sched_shared_sbitmap(q);

> +		if (ret)

> +			goto err_free_tags;

> +

> +		queue_for_each_hw_ctx(q, hctx, i) {

> +			hctx->sched_tags->bitmap_tags =

> +					q->sched_bitmap_tags;

> +			hctx->sched_tags->breserved_tags =

> +					q->sched_breserved_tags;

> +		}

>  	}

>  

>  	ret = e->ops.init_sched(q, e);

>  	if (ret)

> -		goto err;

> +		goto err_free_sbitmap;

>  

>  	blk_mq_debugfs_register_sched(q);

>  

> @@ -584,6 +590,7 @@ int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)

>  				eq = q->elevator;

>  				blk_mq_sched_free_requests(q);

>  				blk_mq_exit_sched(q, eq);

> +				blk_mq_exit_sched_shared_sbitmap(q);


blk_mq_exit_sched_shared_sbitmap() has been called in blk_mq_exit_sched() already.

>  				kobject_put(&eq->kobj);

>  				return ret;

>  			}

> @@ -593,7 +600,10 @@ int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)

>  

>  	return 0;

>  

> -err:

> +err_free_sbitmap:

> +	if (blk_mq_is_sbitmap_shared(q->tag_set->flags))

> +		blk_mq_exit_sched_shared_sbitmap(q);

> +err_free_tags:

>  	blk_mq_sched_free_requests(q);

>  	blk_mq_sched_tags_teardown(q);

>  	q->elevator = NULL;

> @@ -631,5 +641,7 @@ void blk_mq_exit_sched(struct request_queue *q, struct elevator_queue *e)

>  	if (e->type->ops.exit_sched)

>  		e->type->ops.exit_sched(e);

>  	blk_mq_sched_tags_teardown(q);

> +	if (blk_mq_is_sbitmap_shared(q->tag_set->flags))

> +		blk_mq_exit_sched_shared_sbitmap(q);

>  	q->elevator = NULL;

>  }

> diff --git a/block/blk-mq-tag.c b/block/blk-mq-tag.c

> index 2a37731e8244..734fedceca7d 100644

> --- a/block/blk-mq-tag.c

> +++ b/block/blk-mq-tag.c

> @@ -466,19 +466,40 @@ static int blk_mq_init_bitmap_tags(struct blk_mq_tags *tags,

>  	return -ENOMEM;

>  }

>  

> -int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set, unsigned int flags)

> +static int __blk_mq_init_bitmaps(struct sbitmap_queue *bitmap_tags,

> +				 struct sbitmap_queue *breserved_tags,

> +				 struct blk_mq_tag_set *set,

> +				 unsigned int queue_depth,

> +				 unsigned int reserved)

>  {

> -	unsigned int depth = set->queue_depth - set->reserved_tags;

> +	unsigned int depth = queue_depth - reserved;

>  	int alloc_policy = BLK_MQ_FLAG_TO_ALLOC_POLICY(set->flags);

>  	bool round_robin = alloc_policy == BLK_TAG_ALLOC_RR;

> -	int i, node = set->numa_node;

>  

> -	if (bt_alloc(&set->__bitmap_tags, depth, round_robin, node))

> +	if (bt_alloc(bitmap_tags, depth, round_robin, set->numa_node))

>  		return -ENOMEM;

> -	if (bt_alloc(&set->__breserved_tags, set->reserved_tags,

> -		     round_robin, node))

> +	if (bt_alloc(breserved_tags, set->reserved_tags,

> +		     round_robin, set->numa_node))

>  		goto free_bitmap_tags;

>  

> +	return 0;

> +

> +free_bitmap_tags:

> +	sbitmap_queue_free(bitmap_tags);

> +	return -ENOMEM;

> +}

> +

> +int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set)


IMO, this function should be named as blk_mq_init_shared_tags
and moved to blk-mq-sched.c

> +{

> +	int i, ret;

> +

> +	ret = __blk_mq_init_bitmaps(&set->__bitmap_tags,

> +				    &set->__breserved_tags,

> +				    set, set->queue_depth,

> +				    set->reserved_tags);

> +	if (ret)

> +		return ret;

> +

>  	for (i = 0; i < set->nr_hw_queues; i++) {

>  		struct blk_mq_tags *tags = set->tags[i];

>  

> @@ -487,9 +508,6 @@ int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set, unsigned int flags)

>  	}

>  

>  	return 0;

> -free_bitmap_tags:

> -	sbitmap_queue_free(&set->__bitmap_tags);

> -	return -ENOMEM;

>  }

>  

>  void blk_mq_exit_shared_sbitmap(struct blk_mq_tag_set *set)

> @@ -498,6 +516,52 @@ void blk_mq_exit_shared_sbitmap(struct blk_mq_tag_set *set)

>  	sbitmap_queue_free(&set->__breserved_tags);

>  }

>  

> +#define MAX_SCHED_RQ (16 * BLKDEV_MAX_RQ)

> +

> +int blk_mq_init_sched_shared_sbitmap(struct request_queue *queue)

> +{

> +	struct blk_mq_tag_set *set = queue->tag_set;

> +	int ret;

> +

> +	queue->sched_bitmap_tags =

> +		kmalloc(sizeof(*queue->sched_bitmap_tags), GFP_KERNEL);

> +	queue->sched_breserved_tags =

> +		kmalloc(sizeof(*queue->sched_breserved_tags), GFP_KERNEL);

> +	if (!queue->sched_bitmap_tags || !queue->sched_breserved_tags)

> +		goto err;


The two sbitmap queues can be embedded into 'request queue', so that
we can avoid to re-allocation in every elevator switch.

I will ask Yanhui to test the patch and see if it can make a difference.


Thanks,
Ming
John Garry May 7, 2021, 10:15 a.m. UTC | #2
On 06/05/2021 09:32, Ming Lei wrote:
>> +	if (blk_mq_is_sbitmap_shared(q->tag_set->flags)) {

>> +		ret = blk_mq_init_sched_shared_sbitmap(q);

>> +		if (ret)

>> +			goto err_free_tags;

>> +

>> +		queue_for_each_hw_ctx(q, hctx, i) {

>> +			hctx->sched_tags->bitmap_tags =

>> +					q->sched_bitmap_tags;

>> +			hctx->sched_tags->breserved_tags =

>> +					q->sched_breserved_tags;

>> +		}

>>   	}

>>   

>>   	ret = e->ops.init_sched(q, e);

>>   	if (ret)

>> -		goto err;

>> +		goto err_free_sbitmap;

>>   

>>   	blk_mq_debugfs_register_sched(q);

>>   

>> @@ -584,6 +590,7 @@ int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)

>>   				eq = q->elevator;

>>   				blk_mq_sched_free_requests(q);

>>   				blk_mq_exit_sched(q, eq);

>> +				blk_mq_exit_sched_shared_sbitmap(q);

> blk_mq_exit_sched_shared_sbitmap() has been called in blk_mq_exit_sched() already.


ah, yes

> 

>>   				kobject_put(&eq->kobj);

>>   				return ret;

>>   			}

>> @@ -593,7 +600,10 @@ int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)

>>   

>>   	return 0;

>>   

>> -err:

>> +err_free_sbitmap:

>> +	if (blk_mq_is_sbitmap_shared(q->tag_set->flags))

>> +		blk_mq_exit_sched_shared_sbitmap(q);

>> +err_free_tags:

>>   	blk_mq_sched_free_requests(q);

>>   	blk_mq_sched_tags_teardown(q);

>>   	q->elevator = NULL;

>> @@ -631,5 +641,7 @@ void blk_mq_exit_sched(struct request_queue *q, struct elevator_queue *e)

>>   	if (e->type->ops.exit_sched)

>>   		e->type->ops.exit_sched(e);

>>   	blk_mq_sched_tags_teardown(q);

>> +	if (blk_mq_is_sbitmap_shared(q->tag_set->flags))

>> +		blk_mq_exit_sched_shared_sbitmap(q);

>>   	q->elevator = NULL;

>>   }

>> diff --git a/block/blk-mq-tag.c b/block/blk-mq-tag.c

>> index 2a37731e8244..734fedceca7d 100644

>> --- a/block/blk-mq-tag.c

>> +++ b/block/blk-mq-tag.c

>> @@ -466,19 +466,40 @@ static int blk_mq_init_bitmap_tags(struct blk_mq_tags *tags,

>>   	return -ENOMEM;

>>   }

>>   

>> -int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set, unsigned int flags)

>> +static int __blk_mq_init_bitmaps(struct sbitmap_queue *bitmap_tags,

>> +				 struct sbitmap_queue *breserved_tags,

>> +				 struct blk_mq_tag_set *set,

>> +				 unsigned int queue_depth,

>> +				 unsigned int reserved)

>>   {

>> -	unsigned int depth = set->queue_depth - set->reserved_tags;

>> +	unsigned int depth = queue_depth - reserved;

>>   	int alloc_policy = BLK_MQ_FLAG_TO_ALLOC_POLICY(set->flags);

>>   	bool round_robin = alloc_policy == BLK_TAG_ALLOC_RR;

>> -	int i, node = set->numa_node;

>>   

>> -	if (bt_alloc(&set->__bitmap_tags, depth, round_robin, node))

>> +	if (bt_alloc(bitmap_tags, depth, round_robin, set->numa_node))

>>   		return -ENOMEM;

>> -	if (bt_alloc(&set->__breserved_tags, set->reserved_tags,

>> -		     round_robin, node))

>> +	if (bt_alloc(breserved_tags, set->reserved_tags,

>> +		     round_robin, set->numa_node))

>>   		goto free_bitmap_tags;

>>   

>> +	return 0;

>> +

>> +free_bitmap_tags:

>> +	sbitmap_queue_free(bitmap_tags);

>> +	return -ENOMEM;

>> +}

>> +

>> +int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set)

> IMO, this function should be named as blk_mq_init_shared_tags

> and moved to blk-mq-sched.c


But this is for regular tags.

I assume you mean blk_mq_init_sched_shared_sbitmap(), below.

If so, I can relocate it.

As for "sbitmap" vs "tags" in the name, I'm just being consistent 
between preexisting blk_mq_init_shared_sbitmap() and 
blk_mq_sbitmap_shared(),  and new blk_mq_init_sched_shared_sbitmap()

> 

>> +{

>> +	int i, ret;

>> +

>> +	ret = __blk_mq_init_bitmaps(&set->__bitmap_tags,

>> +				    &set->__breserved_tags,

>> +				    set, set->queue_depth,

>> +				    set->reserved_tags);

>> +	if (ret)

>> +		return ret;

>> +

>>   	for (i = 0; i < set->nr_hw_queues; i++) {

>>   		struct blk_mq_tags *tags = set->tags[i];

>>   

>> @@ -487,9 +508,6 @@ int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set, unsigned int flags)

>>   	}

>>   

>>   	return 0;

>> -free_bitmap_tags:

>> -	sbitmap_queue_free(&set->__bitmap_tags);

>> -	return -ENOMEM;

>>   }

>>   

>>   void blk_mq_exit_shared_sbitmap(struct blk_mq_tag_set *set)

>> @@ -498,6 +516,52 @@ void blk_mq_exit_shared_sbitmap(struct blk_mq_tag_set *set)

>>   	sbitmap_queue_free(&set->__breserved_tags);

>>   }

>>   

>> +#define MAX_SCHED_RQ (16 * BLKDEV_MAX_RQ)

>> +

>> +int blk_mq_init_sched_shared_sbitmap(struct request_queue *queue)

>> +{

>> +	struct blk_mq_tag_set *set = queue->tag_set;

>> +	int ret;

>> +

>> +	queue->sched_bitmap_tags =

>> +		kmalloc(sizeof(*queue->sched_bitmap_tags), GFP_KERNEL);

>> +	queue->sched_breserved_tags =

>> +		kmalloc(sizeof(*queue->sched_breserved_tags), GFP_KERNEL);

>> +	if (!queue->sched_bitmap_tags || !queue->sched_breserved_tags)

>> +		goto err;

> The two sbitmap queues can be embedded into 'request queue', so that

> we can avoid to re-allocation in every elevator switch.


ok

> 

> I will ask Yanhui to test the patch and see if it can make a difference.


Thanks
Ming Lei May 11, 2021, 12:52 a.m. UTC | #3
On Mon, May 03, 2021 at 06:22:13PM +0800, John Garry wrote:
> The tags used for an IO scheduler are currently per hctx.

> 

> As such, when q->nr_hw_queues grows, so does the request queue total IO

> scheduler tag depth.

> 

> This may cause problems for SCSI MQ HBAs whose total driver depth is

> fixed.

> 

> Ming and Yanhui report higher CPU usage and lower throughput in scenarios

> where the fixed total driver tag depth is appreciably lower than the total

> scheduler tag depth:

> https://lore.kernel.org/linux-block/440dfcfc-1a2c-bd98-1161-cec4d78c6dfc@huawei.com/T/#mc0d6d4f95275a2743d1c8c3e4dc9ff6c9aa3a76b

> 


No difference any more wrt. fio running on scsi_debug with this patch in
Yanhui's test machine:

	modprobe scsi_debug host_max_queue=128 submit_queues=32 virtual_gb=256 delay=1
vs.
	modprobe scsi_debug max_queue=128 submit_queues=1 virtual_gb=256 delay=1

Without this patch, the latter's result is 30% higher than the former's.

note: scsi_debug's queue depth needs to be updated to 128 for avoiding io hang,
which is another scsi issue.


Thanks, 
Ming
Douglas Gilbert May 11, 2021, 1:35 a.m. UTC | #4
On 2021-05-10 8:52 p.m., Ming Lei wrote:
> On Mon, May 03, 2021 at 06:22:13PM +0800, John Garry wrote:

>> The tags used for an IO scheduler are currently per hctx.

>>

>> As such, when q->nr_hw_queues grows, so does the request queue total IO

>> scheduler tag depth.

>>

>> This may cause problems for SCSI MQ HBAs whose total driver depth is

>> fixed.

>>

>> Ming and Yanhui report higher CPU usage and lower throughput in scenarios

>> where the fixed total driver tag depth is appreciably lower than the total

>> scheduler tag depth:

>> https://lore.kernel.org/linux-block/440dfcfc-1a2c-bd98-1161-cec4d78c6dfc@huawei.com/T/#mc0d6d4f95275a2743d1c8c3e4dc9ff6c9aa3a76b

>>

> 

> No difference any more wrt. fio running on scsi_debug with this patch in

> Yanhui's test machine:

> 

> 	modprobe scsi_debug host_max_queue=128 submit_queues=32 virtual_gb=256 delay=1

> vs.

> 	modprobe scsi_debug max_queue=128 submit_queues=1 virtual_gb=256 delay=1

> 

> Without this patch, the latter's result is 30% higher than the former's.

> 

> note: scsi_debug's queue depth needs to be updated to 128 for avoiding io hang,

> which is another scsi issue.


"scsi_debug: Fix cmd_per_lun, set to max_queue" made it into lk 5.13.0-rc1 as
commit fc09acb7de31badb2ea9e85d21e071be1a5736e4 . Is this the issue you are
referring to, or is there a separate issue in the wider scsi stack?

BTW Martin's 5.14/scsi-queue is up and running with lk 5.13.0-rc1 .

Doug Gilbert
Ming Lei May 11, 2021, 1:47 a.m. UTC | #5
On Mon, May 10, 2021 at 09:35:01PM -0400, Douglas Gilbert wrote:
> On 2021-05-10 8:52 p.m., Ming Lei wrote:

> > On Mon, May 03, 2021 at 06:22:13PM +0800, John Garry wrote:

> > > The tags used for an IO scheduler are currently per hctx.

> > > 

> > > As such, when q->nr_hw_queues grows, so does the request queue total IO

> > > scheduler tag depth.

> > > 

> > > This may cause problems for SCSI MQ HBAs whose total driver depth is

> > > fixed.

> > > 

> > > Ming and Yanhui report higher CPU usage and lower throughput in scenarios

> > > where the fixed total driver tag depth is appreciably lower than the total

> > > scheduler tag depth:

> > > https://lore.kernel.org/linux-block/440dfcfc-1a2c-bd98-1161-cec4d78c6dfc@huawei.com/T/#mc0d6d4f95275a2743d1c8c3e4dc9ff6c9aa3a76b

> > > 

> > 

> > No difference any more wrt. fio running on scsi_debug with this patch in

> > Yanhui's test machine:

> > 

> > 	modprobe scsi_debug host_max_queue=128 submit_queues=32 virtual_gb=256 delay=1

> > vs.

> > 	modprobe scsi_debug max_queue=128 submit_queues=1 virtual_gb=256 delay=1

> > 

> > Without this patch, the latter's result is 30% higher than the former's.

> > 

> > note: scsi_debug's queue depth needs to be updated to 128 for avoiding io hang,

> > which is another scsi issue.

> 

> "scsi_debug: Fix cmd_per_lun, set to max_queue" made it into lk 5.13.0-rc1 as

> commit fc09acb7de31badb2ea9e85d21e071be1a5736e4 . Is this the issue you are

> referring to, or is there a separate issue in the wider scsi stack?


OK, that is it, then it isn't necessary to update scsi_debug's queue
depth for the test.


Thanks,
Ming
John Garry May 11, 2021, 7:33 a.m. UTC | #6
On 11/05/2021 01:52, Ming Lei wrote:
>> fixed.

>>

>> Ming and Yanhui report higher CPU usage and lower throughput in scenarios

>> where the fixed total driver tag depth is appreciably lower than the total

>> scheduler tag depth:

>> https://lore.kernel.org/linux-block/440dfcfc-1a2c-bd98-1161-cec4d78c6dfc@huawei.com/T/#mc0d6d4f95275a2743d1c8c3e4dc9ff6c9aa3a76b

>>

> No difference any more wrt. fio running on scsi_debug with this patch in

> Yanhui's test machine:

> 

> 	modprobe scsi_debug host_max_queue=128 submit_queues=32 virtual_gb=256 delay=1

> vs.

> 	modprobe scsi_debug max_queue=128 submit_queues=1 virtual_gb=256 delay=1

> 

> Without this patch, the latter's result is 30% higher than the former's.

> 


ok, good. I'll post a v2 with comments addressed.

> note: scsi_debug's queue depth needs to be updated to 128 for avoiding io hang,

> which is another scsi issue.

> 

I was just carrying Doug's patch to test.

Thanks,
John
diff mbox series

Patch

diff --git a/block/blk-mq-sched.c b/block/blk-mq-sched.c
index e1e997af89a0..121207abc026 100644
--- a/block/blk-mq-sched.c
+++ b/block/blk-mq-sched.c
@@ -497,11 +497,9 @@  static void blk_mq_sched_free_tags(struct blk_mq_tag_set *set,
 				   struct blk_mq_hw_ctx *hctx,
 				   unsigned int hctx_idx)
 {
-	unsigned int flags = set->flags & ~BLK_MQ_F_TAG_HCTX_SHARED;
-
 	if (hctx->sched_tags) {
 		blk_mq_free_rqs(set, hctx->sched_tags, hctx_idx);
-		blk_mq_free_rq_map(hctx->sched_tags, flags);
+		blk_mq_free_rq_map(hctx->sched_tags, set->flags);
 		hctx->sched_tags = NULL;
 	}
 }
@@ -511,12 +509,10 @@  static int blk_mq_sched_alloc_tags(struct request_queue *q,
 				   unsigned int hctx_idx)
 {
 	struct blk_mq_tag_set *set = q->tag_set;
-	/* Clear HCTX_SHARED so tags are init'ed */
-	unsigned int flags = set->flags & ~BLK_MQ_F_TAG_HCTX_SHARED;
 	int ret;
 
 	hctx->sched_tags = blk_mq_alloc_rq_map(set, hctx_idx, q->nr_requests,
-					       set->reserved_tags, flags);
+					       set->reserved_tags, set->flags);
 	if (!hctx->sched_tags)
 		return -ENOMEM;
 
@@ -534,11 +530,8 @@  static void blk_mq_sched_tags_teardown(struct request_queue *q)
 	int i;
 
 	queue_for_each_hw_ctx(q, hctx, i) {
-		/* Clear HCTX_SHARED so tags are freed */
-		unsigned int flags = hctx->flags & ~BLK_MQ_F_TAG_HCTX_SHARED;
-
 		if (hctx->sched_tags) {
-			blk_mq_free_rq_map(hctx->sched_tags, flags);
+			blk_mq_free_rq_map(hctx->sched_tags, hctx->flags);
 			hctx->sched_tags = NULL;
 		}
 	}
@@ -568,12 +561,25 @@  int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)
 	queue_for_each_hw_ctx(q, hctx, i) {
 		ret = blk_mq_sched_alloc_tags(q, hctx, i);
 		if (ret)
-			goto err;
+			goto err_free_tags;
+	}
+
+	if (blk_mq_is_sbitmap_shared(q->tag_set->flags)) {
+		ret = blk_mq_init_sched_shared_sbitmap(q);
+		if (ret)
+			goto err_free_tags;
+
+		queue_for_each_hw_ctx(q, hctx, i) {
+			hctx->sched_tags->bitmap_tags =
+					q->sched_bitmap_tags;
+			hctx->sched_tags->breserved_tags =
+					q->sched_breserved_tags;
+		}
 	}
 
 	ret = e->ops.init_sched(q, e);
 	if (ret)
-		goto err;
+		goto err_free_sbitmap;
 
 	blk_mq_debugfs_register_sched(q);
 
@@ -584,6 +590,7 @@  int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)
 				eq = q->elevator;
 				blk_mq_sched_free_requests(q);
 				blk_mq_exit_sched(q, eq);
+				blk_mq_exit_sched_shared_sbitmap(q);
 				kobject_put(&eq->kobj);
 				return ret;
 			}
@@ -593,7 +600,10 @@  int blk_mq_init_sched(struct request_queue *q, struct elevator_type *e)
 
 	return 0;
 
-err:
+err_free_sbitmap:
+	if (blk_mq_is_sbitmap_shared(q->tag_set->flags))
+		blk_mq_exit_sched_shared_sbitmap(q);
+err_free_tags:
 	blk_mq_sched_free_requests(q);
 	blk_mq_sched_tags_teardown(q);
 	q->elevator = NULL;
@@ -631,5 +641,7 @@  void blk_mq_exit_sched(struct request_queue *q, struct elevator_queue *e)
 	if (e->type->ops.exit_sched)
 		e->type->ops.exit_sched(e);
 	blk_mq_sched_tags_teardown(q);
+	if (blk_mq_is_sbitmap_shared(q->tag_set->flags))
+		blk_mq_exit_sched_shared_sbitmap(q);
 	q->elevator = NULL;
 }
diff --git a/block/blk-mq-tag.c b/block/blk-mq-tag.c
index 2a37731e8244..734fedceca7d 100644
--- a/block/blk-mq-tag.c
+++ b/block/blk-mq-tag.c
@@ -466,19 +466,40 @@  static int blk_mq_init_bitmap_tags(struct blk_mq_tags *tags,
 	return -ENOMEM;
 }
 
-int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set, unsigned int flags)
+static int __blk_mq_init_bitmaps(struct sbitmap_queue *bitmap_tags,
+				 struct sbitmap_queue *breserved_tags,
+				 struct blk_mq_tag_set *set,
+				 unsigned int queue_depth,
+				 unsigned int reserved)
 {
-	unsigned int depth = set->queue_depth - set->reserved_tags;
+	unsigned int depth = queue_depth - reserved;
 	int alloc_policy = BLK_MQ_FLAG_TO_ALLOC_POLICY(set->flags);
 	bool round_robin = alloc_policy == BLK_TAG_ALLOC_RR;
-	int i, node = set->numa_node;
 
-	if (bt_alloc(&set->__bitmap_tags, depth, round_robin, node))
+	if (bt_alloc(bitmap_tags, depth, round_robin, set->numa_node))
 		return -ENOMEM;
-	if (bt_alloc(&set->__breserved_tags, set->reserved_tags,
-		     round_robin, node))
+	if (bt_alloc(breserved_tags, set->reserved_tags,
+		     round_robin, set->numa_node))
 		goto free_bitmap_tags;
 
+	return 0;
+
+free_bitmap_tags:
+	sbitmap_queue_free(bitmap_tags);
+	return -ENOMEM;
+}
+
+int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set)
+{
+	int i, ret;
+
+	ret = __blk_mq_init_bitmaps(&set->__bitmap_tags,
+				    &set->__breserved_tags,
+				    set, set->queue_depth,
+				    set->reserved_tags);
+	if (ret)
+		return ret;
+
 	for (i = 0; i < set->nr_hw_queues; i++) {
 		struct blk_mq_tags *tags = set->tags[i];
 
@@ -487,9 +508,6 @@  int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set, unsigned int flags)
 	}
 
 	return 0;
-free_bitmap_tags:
-	sbitmap_queue_free(&set->__bitmap_tags);
-	return -ENOMEM;
 }
 
 void blk_mq_exit_shared_sbitmap(struct blk_mq_tag_set *set)
@@ -498,6 +516,52 @@  void blk_mq_exit_shared_sbitmap(struct blk_mq_tag_set *set)
 	sbitmap_queue_free(&set->__breserved_tags);
 }
 
+#define MAX_SCHED_RQ (16 * BLKDEV_MAX_RQ)
+
+int blk_mq_init_sched_shared_sbitmap(struct request_queue *queue)
+{
+	struct blk_mq_tag_set *set = queue->tag_set;
+	int ret;
+
+	queue->sched_bitmap_tags =
+		kmalloc(sizeof(*queue->sched_bitmap_tags), GFP_KERNEL);
+	queue->sched_breserved_tags =
+		kmalloc(sizeof(*queue->sched_breserved_tags), GFP_KERNEL);
+	if (!queue->sched_bitmap_tags || !queue->sched_breserved_tags)
+		goto err;
+
+	/*
+	 * Set initial depth at max so that we don't need to reallocate for
+	 * updating nr_requests.
+	 */
+	ret = __blk_mq_init_bitmaps(queue->sched_bitmap_tags,
+				    queue->sched_breserved_tags,
+				    set, MAX_SCHED_RQ, set->reserved_tags);
+	if (ret)
+		goto err;
+
+	sbitmap_queue_resize(queue->sched_bitmap_tags,
+			     queue->nr_requests - set->reserved_tags);
+
+	return 0;
+
+err:
+	kfree(queue->sched_bitmap_tags);
+	kfree(queue->sched_breserved_tags);
+	return -ENOMEM;
+}
+
+void blk_mq_exit_sched_shared_sbitmap(struct request_queue *queue)
+{
+	sbitmap_queue_free(queue->sched_bitmap_tags);
+	kfree(queue->sched_bitmap_tags);
+	queue->sched_bitmap_tags = NULL;
+
+	sbitmap_queue_free(queue->sched_breserved_tags);
+	kfree(queue->sched_breserved_tags);
+	queue->sched_breserved_tags = NULL;
+}
+
 struct blk_mq_tags *blk_mq_init_tags(unsigned int total_tags,
 				     unsigned int reserved_tags,
 				     int node, unsigned int flags)
@@ -551,8 +615,6 @@  int blk_mq_tag_update_depth(struct blk_mq_hw_ctx *hctx,
 	 */
 	if (tdepth > tags->nr_tags) {
 		struct blk_mq_tag_set *set = hctx->queue->tag_set;
-		/* Only sched tags can grow, so clear HCTX_SHARED flag  */
-		unsigned int flags = set->flags & ~BLK_MQ_F_TAG_HCTX_SHARED;
 		struct blk_mq_tags *new;
 		bool ret;
 
@@ -563,21 +625,21 @@  int blk_mq_tag_update_depth(struct blk_mq_hw_ctx *hctx,
 		 * We need some sort of upper limit, set it high enough that
 		 * no valid use cases should require more.
 		 */
-		if (tdepth > 16 * BLKDEV_MAX_RQ)
+		if (tdepth > MAX_SCHED_RQ)
 			return -EINVAL;
 
 		new = blk_mq_alloc_rq_map(set, hctx->queue_num, tdepth,
-				tags->nr_reserved_tags, flags);
+				tags->nr_reserved_tags, set->flags);
 		if (!new)
 			return -ENOMEM;
 		ret = blk_mq_alloc_rqs(set, new, hctx->queue_num, tdepth);
 		if (ret) {
-			blk_mq_free_rq_map(new, flags);
+			blk_mq_free_rq_map(new, set->flags);
 			return -ENOMEM;
 		}
 
 		blk_mq_free_rqs(set, *tagsptr, hctx->queue_num);
-		blk_mq_free_rq_map(*tagsptr, flags);
+		blk_mq_free_rq_map(*tagsptr, set->flags);
 		*tagsptr = new;
 	} else {
 		/*
diff --git a/block/blk-mq-tag.h b/block/blk-mq-tag.h
index 7d3e6b333a4a..553fa71efd42 100644
--- a/block/blk-mq-tag.h
+++ b/block/blk-mq-tag.h
@@ -27,10 +27,10 @@  extern struct blk_mq_tags *blk_mq_init_tags(unsigned int nr_tags,
 					int node, unsigned int flags);
 extern void blk_mq_free_tags(struct blk_mq_tags *tags, unsigned int flags);
 
-extern int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set,
-				      unsigned int flags);
+extern int blk_mq_init_shared_sbitmap(struct blk_mq_tag_set *set);
 extern void blk_mq_exit_shared_sbitmap(struct blk_mq_tag_set *set);
-
+extern int blk_mq_init_sched_shared_sbitmap(struct request_queue *queue);
+extern void blk_mq_exit_sched_shared_sbitmap(struct request_queue *queue);
 extern unsigned int blk_mq_get_tag(struct blk_mq_alloc_data *data);
 extern void blk_mq_put_tag(struct blk_mq_tags *tags, struct blk_mq_ctx *ctx,
 			   unsigned int tag);
diff --git a/block/blk-mq.c b/block/blk-mq.c
index 927189a55575..f6e22b32a07f 100644
--- a/block/blk-mq.c
+++ b/block/blk-mq.c
@@ -3473,7 +3473,7 @@  int blk_mq_alloc_tag_set(struct blk_mq_tag_set *set)
 	if (blk_mq_is_sbitmap_shared(set->flags)) {
 		atomic_set(&set->active_queues_shared_sbitmap, 0);
 
-		if (blk_mq_init_shared_sbitmap(set, set->flags)) {
+		if (blk_mq_init_shared_sbitmap(set)) {
 			ret = -ENOMEM;
 			goto out_free_mq_rq_maps;
 		}
@@ -3549,15 +3549,24 @@  int blk_mq_update_nr_requests(struct request_queue *q, unsigned int nr)
 		} else {
 			ret = blk_mq_tag_update_depth(hctx, &hctx->sched_tags,
 							nr, true);
+			if (blk_mq_is_sbitmap_shared(set->flags)) {
+				hctx->sched_tags->bitmap_tags =
+					q->sched_bitmap_tags;
+				hctx->sched_tags->breserved_tags =
+					q->sched_breserved_tags;
+			}
 		}
 		if (ret)
 			break;
 		if (q->elevator && q->elevator->type->ops.depth_updated)
 			q->elevator->type->ops.depth_updated(hctx);
 	}
-
-	if (!ret)
+	if (!ret) {
 		q->nr_requests = nr;
+		if (q->elevator && blk_mq_is_sbitmap_shared(set->flags))
+			sbitmap_queue_resize(q->sched_bitmap_tags,
+					     nr - set->reserved_tags);
+	}
 
 	blk_mq_unquiesce_queue(q);
 	blk_mq_unfreeze_queue(q);
diff --git a/include/linux/blkdev.h b/include/linux/blkdev.h
index f2e77ba97550..8055ebd9f285 100644
--- a/include/linux/blkdev.h
+++ b/include/linux/blkdev.h
@@ -26,6 +26,7 @@ 
 #include <linux/scatterlist.h>
 #include <linux/blkzoned.h>
 #include <linux/pm.h>
+#include <linux/sbitmap.h>
 
 struct module;
 struct scsi_ioctl_command;
@@ -496,6 +497,9 @@  struct request_queue {
 
 	atomic_t		nr_active_requests_shared_sbitmap;
 
+	struct sbitmap_queue *sched_bitmap_tags;
+	struct sbitmap_queue *sched_breserved_tags;
+
 	struct list_head	icq_list;
 #ifdef CONFIG_BLK_CGROUP
 	DECLARE_BITMAP		(blkcg_pols, BLKCG_MAX_POLS);