Greg Kroah-Hartman | b244131 | 2017-11-01 15:07:57 +0100 | [diff] [blame] | 1 | /* SPDX-License-Identifier: GPL-2.0 */ |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 2 | #ifndef INT_BLK_MQ_TAG_H |
| 3 | #define INT_BLK_MQ_TAG_H |
| 4 | |
Christoph Hellwig | 2e9bc34 | 2021-09-20 14:33:23 +0200 | [diff] [blame] | 5 | struct blk_mq_alloc_data; |
| 6 | |
John Garry | 1c0706a | 2020-08-19 23:20:22 +0800 | [diff] [blame] | 7 | extern struct blk_mq_tags *blk_mq_init_tags(unsigned int nr_tags, |
| 8 | unsigned int reserved_tags, |
John Garry | e155b0c | 2021-10-05 18:23:37 +0800 | [diff] [blame] | 9 | int node, int alloc_policy); |
| 10 | extern void blk_mq_free_tags(struct blk_mq_tags *tags); |
John Garry | 56b6808 | 2021-05-13 20:00:57 +0800 | [diff] [blame] | 11 | extern int blk_mq_init_bitmaps(struct sbitmap_queue *bitmap_tags, |
| 12 | struct sbitmap_queue *breserved_tags, |
| 13 | unsigned int queue_depth, |
| 14 | unsigned int reserved, |
| 15 | int node, int alloc_policy); |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 16 | |
Ming Lei | cb96a42c | 2014-06-01 00:43:37 +0800 | [diff] [blame] | 17 | extern unsigned int blk_mq_get_tag(struct blk_mq_alloc_data *data); |
Jens Axboe | 349302d | 2021-10-09 13:10:39 -0600 | [diff] [blame] | 18 | unsigned long blk_mq_get_tags(struct blk_mq_alloc_data *data, int nr_tags, |
| 19 | unsigned int *offset); |
John Garry | cae740a | 2020-02-26 20:10:15 +0800 | [diff] [blame] | 20 | extern void blk_mq_put_tag(struct blk_mq_tags *tags, struct blk_mq_ctx *ctx, |
| 21 | unsigned int tag); |
Jens Axboe | f794f33 | 2021-10-08 05:50:46 -0600 | [diff] [blame] | 22 | void blk_mq_put_tags(struct blk_mq_tags *tags, int *tag_array, int nr_tags); |
Jens Axboe | 70f36b6 | 2017-01-19 10:59:07 -0700 | [diff] [blame] | 23 | extern int blk_mq_tag_update_depth(struct blk_mq_hw_ctx *hctx, |
| 24 | struct blk_mq_tags **tags, |
| 25 | unsigned int depth, bool can_grow); |
John Garry | 079a2e3 | 2021-10-05 18:23:39 +0800 | [diff] [blame] | 26 | extern void blk_mq_tag_resize_shared_tags(struct blk_mq_tag_set *set, |
John Garry | 32bc15a | 2020-08-19 23:20:24 +0800 | [diff] [blame] | 27 | unsigned int size); |
John Garry | 079a2e3 | 2021-10-05 18:23:39 +0800 | [diff] [blame] | 28 | extern void blk_mq_tag_update_sched_shared_tags(struct request_queue *q); |
John Garry | 32bc15a | 2020-08-19 23:20:24 +0800 | [diff] [blame] | 29 | |
Jens Axboe | aed3ea9 | 2014-12-22 14:04:42 -0700 | [diff] [blame] | 30 | extern void blk_mq_tag_wakeup_all(struct blk_mq_tags *tags, bool); |
John Garry | fc39f8d | 2021-12-06 20:49:49 +0800 | [diff] [blame] | 31 | void blk_mq_queue_tag_busy_iter(struct request_queue *q, busy_tag_iter_fn *fn, |
Christoph Hellwig | 0bf6cd5 | 2015-09-27 21:01:51 +0200 | [diff] [blame] | 32 | void *priv); |
Ming Lei | 602380d | 2020-05-29 15:53:14 +0200 | [diff] [blame] | 33 | void blk_mq_all_tag_iter(struct blk_mq_tags *tags, busy_tag_iter_fn *fn, |
| 34 | void *priv); |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 35 | |
Omar Sandoval | 8845964 | 2016-09-17 08:38:44 -0600 | [diff] [blame] | 36 | static inline struct sbq_wait_state *bt_wait_ptr(struct sbitmap_queue *bt, |
| 37 | struct blk_mq_hw_ctx *hctx) |
| 38 | { |
| 39 | if (!hctx) |
| 40 | return &bt->ws[0]; |
| 41 | return sbq_wait_ptr(bt, &hctx->wait_index); |
| 42 | } |
| 43 | |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 44 | enum { |
Christoph Hellwig | 419c3d5 | 2020-05-29 15:53:11 +0200 | [diff] [blame] | 45 | BLK_MQ_NO_TAG = -1U, |
Jens Axboe | 5385fa4 | 2017-10-01 01:26:21 -0600 | [diff] [blame] | 46 | BLK_MQ_TAG_MIN = 1, |
Christoph Hellwig | 419c3d5 | 2020-05-29 15:53:11 +0200 | [diff] [blame] | 47 | BLK_MQ_TAG_MAX = BLK_MQ_NO_TAG - 1, |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 48 | }; |
| 49 | |
Jens Axboe | 0d2602c | 2014-05-13 15:10:52 -0600 | [diff] [blame] | 50 | extern bool __blk_mq_tag_busy(struct blk_mq_hw_ctx *); |
| 51 | extern void __blk_mq_tag_idle(struct blk_mq_hw_ctx *); |
| 52 | |
| 53 | static inline bool blk_mq_tag_busy(struct blk_mq_hw_ctx *hctx) |
| 54 | { |
Ming Lei | 51db1c3 | 2020-08-19 23:20:19 +0800 | [diff] [blame] | 55 | if (!(hctx->flags & BLK_MQ_F_TAG_QUEUE_SHARED)) |
Jens Axboe | 0d2602c | 2014-05-13 15:10:52 -0600 | [diff] [blame] | 56 | return false; |
| 57 | |
| 58 | return __blk_mq_tag_busy(hctx); |
| 59 | } |
| 60 | |
| 61 | static inline void blk_mq_tag_idle(struct blk_mq_hw_ctx *hctx) |
| 62 | { |
Ming Lei | 51db1c3 | 2020-08-19 23:20:19 +0800 | [diff] [blame] | 63 | if (!(hctx->flags & BLK_MQ_F_TAG_QUEUE_SHARED)) |
Jens Axboe | 0d2602c | 2014-05-13 15:10:52 -0600 | [diff] [blame] | 64 | return; |
| 65 | |
| 66 | __blk_mq_tag_idle(hctx); |
| 67 | } |
| 68 | |
Sagi Grimberg | 415b806 | 2017-02-27 10:04:39 -0700 | [diff] [blame] | 69 | static inline bool blk_mq_tag_is_reserved(struct blk_mq_tags *tags, |
| 70 | unsigned int tag) |
| 71 | { |
| 72 | return tag < tags->nr_reserved_tags; |
| 73 | } |
| 74 | |
Jens Axboe | 320ae51 | 2013-10-24 09:20:05 +0100 | [diff] [blame] | 75 | #endif |