mirror of
https://github.com/AuxXxilium/linux_dsm_epyc7002.git
synced 2024-12-04 13:46:50 +07:00
3ee0ce2a54
Without locking out CPU mask operations we might end up with an inconsistent
view of the cpumask in the function.
Fixes: 5e385a6ef3: "genirq: Add a helper to spread an affinity mask for MSI/MSI-X vectors"
Signed-off-by: Christoph Hellwig <hch@lst.de>
Link: http://lkml.kernel.org/r/1470924405-25728-1-git-send-email-hch@lst.de
Signed-off-by: Thomas Gleixner <tglx@linutronix.de>
64 lines
1.5 KiB
C
64 lines
1.5 KiB
C
|
|
#include <linux/interrupt.h>
|
|
#include <linux/kernel.h>
|
|
#include <linux/slab.h>
|
|
#include <linux/cpu.h>
|
|
|
|
static int get_first_sibling(unsigned int cpu)
|
|
{
|
|
unsigned int ret;
|
|
|
|
ret = cpumask_first(topology_sibling_cpumask(cpu));
|
|
if (ret < nr_cpu_ids)
|
|
return ret;
|
|
return cpu;
|
|
}
|
|
|
|
/*
|
|
* Take a map of online CPUs and the number of available interrupt vectors
|
|
* and generate an output cpumask suitable for spreading MSI/MSI-X vectors
|
|
* so that they are distributed as good as possible around the CPUs. If
|
|
* more vectors than CPUs are available we'll map one to each CPU,
|
|
* otherwise we map one to the first sibling of each socket.
|
|
*
|
|
* If there are more vectors than CPUs we will still only have one bit
|
|
* set per CPU, but interrupt code will keep on assigning the vectors from
|
|
* the start of the bitmap until we run out of vectors.
|
|
*/
|
|
struct cpumask *irq_create_affinity_mask(unsigned int *nr_vecs)
|
|
{
|
|
struct cpumask *affinity_mask;
|
|
unsigned int max_vecs = *nr_vecs;
|
|
|
|
if (max_vecs == 1)
|
|
return NULL;
|
|
|
|
affinity_mask = kzalloc(cpumask_size(), GFP_KERNEL);
|
|
if (!affinity_mask) {
|
|
*nr_vecs = 1;
|
|
return NULL;
|
|
}
|
|
|
|
get_online_cpus();
|
|
if (max_vecs >= num_online_cpus()) {
|
|
cpumask_copy(affinity_mask, cpu_online_mask);
|
|
*nr_vecs = num_online_cpus();
|
|
} else {
|
|
unsigned int vecs = 0, cpu;
|
|
|
|
for_each_online_cpu(cpu) {
|
|
if (cpu == get_first_sibling(cpu)) {
|
|
cpumask_set_cpu(cpu, affinity_mask);
|
|
vecs++;
|
|
}
|
|
|
|
if (--max_vecs == 0)
|
|
break;
|
|
}
|
|
*nr_vecs = vecs;
|
|
}
|
|
put_online_cpus();
|
|
|
|
return affinity_mask;
|
|
}
|