| 1 | // SPDX-License-Identifier: GPL-2.0 |
| 2 | /* |
| 3 | * Copyright (c) 2016 Christoph Hellwig. |
| 4 | */ |
| 5 | #include <linux/device.h> |
| 6 | #include <linux/blk-mq-virtio.h> |
| 7 | #include <linux/virtio_config.h> |
| 8 | #include <linux/module.h> |
| 9 | #include "blk-mq.h" |
| 10 | |
| 11 | /** |
| 12 | * blk_mq_virtio_map_queues - provide a default queue mapping for virtio device |
| 13 | * @qmap: CPU to hardware queue map. |
| 14 | * @vdev: virtio device to provide a mapping for. |
| 15 | * @first_vec: first interrupt vectors to use for queues (usually 0) |
| 16 | * |
| 17 | * This function assumes the virtio device @vdev has at least as many available |
| 18 | * interrupt vectors as @set has queues. It will then query the vector |
| 19 | * corresponding to each queue for it's affinity mask and built queue mapping |
| 20 | * that maps a queue to the CPUs that have irq affinity for the corresponding |
| 21 | * vector. |
| 22 | */ |
| 23 | void blk_mq_virtio_map_queues(struct blk_mq_queue_map *qmap, |
| 24 | struct virtio_device *vdev, int first_vec) |
| 25 | { |
| 26 | const struct cpumask *mask; |
| 27 | unsigned int queue, cpu; |
| 28 | |
| 29 | if (!vdev->config->get_vq_affinity) |
| 30 | goto fallback; |
| 31 | |
| 32 | for (queue = 0; queue < qmap->nr_queues; queue++) { |
| 33 | mask = vdev->config->get_vq_affinity(vdev, first_vec + queue); |
| 34 | if (!mask) |
| 35 | goto fallback; |
| 36 | |
| 37 | for_each_cpu(cpu, mask) |
| 38 | qmap->mq_map[cpu] = qmap->queue_offset + queue; |
| 39 | } |
| 40 | |
| 41 | return; |
| 42 | |
| 43 | fallback: |
| 44 | blk_mq_map_queues(qmap); |
| 45 | } |
| 46 | EXPORT_SYMBOL_GPL(blk_mq_virtio_map_queues); |