blob: 02edeafcb2bf84af2354a9882b82a7ef6e57fa0f [file] [log] [blame]
Thomas Gleixnerd2912cb2019-06-04 10:11:33 +02001/* SPDX-License-Identifier: GPL-2.0-only */
Amir Vadai122733a2012-07-18 22:33:50 +00002#ifndef __LINUX_CPU_RMAP_H
3#define __LINUX_CPU_RMAP_H
4
Ben Hutchingsc39649c2011-01-19 11:03:25 +00005/*
6 * cpu_rmap.c: CPU affinity reverse-map support
7 * Copyright 2011 Solarflare Communications Inc.
Ben Hutchingsc39649c2011-01-19 11:03:25 +00008 */
9
10#include <linux/cpumask.h>
11#include <linux/gfp.h>
12#include <linux/slab.h>
David Decotigny896f97e2013-01-11 14:31:36 -080013#include <linux/kref.h>
Ben Hutchingsc39649c2011-01-19 11:03:25 +000014
15/**
16 * struct cpu_rmap - CPU affinity reverse-map
David Decotigny896f97e2013-01-11 14:31:36 -080017 * @refcount: kref for object
Ben Hutchingsc39649c2011-01-19 11:03:25 +000018 * @size: Number of objects to be reverse-mapped
19 * @used: Number of objects added
20 * @obj: Pointer to array of object pointers
21 * @near: For each CPU, the index and distance to the nearest object,
22 * based on affinity masks
23 */
24struct cpu_rmap {
David Decotigny896f97e2013-01-11 14:31:36 -080025 struct kref refcount;
Ben Hutchingsc39649c2011-01-19 11:03:25 +000026 u16 size, used;
27 void **obj;
28 struct {
29 u16 index;
30 u16 dist;
31 } near[0];
32};
33#define CPU_RMAP_DIST_INF 0xffff
34
35extern struct cpu_rmap *alloc_cpu_rmap(unsigned int size, gfp_t flags);
David Decotigny896f97e2013-01-11 14:31:36 -080036extern int cpu_rmap_put(struct cpu_rmap *rmap);
Ben Hutchingsc39649c2011-01-19 11:03:25 +000037
38extern int cpu_rmap_add(struct cpu_rmap *rmap, void *obj);
39extern int cpu_rmap_update(struct cpu_rmap *rmap, u16 index,
40 const struct cpumask *affinity);
41
42static inline u16 cpu_rmap_lookup_index(struct cpu_rmap *rmap, unsigned int cpu)
43{
44 return rmap->near[cpu].index;
45}
46
47static inline void *cpu_rmap_lookup_obj(struct cpu_rmap *rmap, unsigned int cpu)
48{
49 return rmap->obj[rmap->near[cpu].index];
50}
51
Ben Hutchingsc39649c2011-01-19 11:03:25 +000052/**
53 * alloc_irq_cpu_rmap - allocate CPU affinity reverse-map for IRQs
54 * @size: Number of objects to be mapped
55 *
56 * Must be called in process context.
57 */
58static inline struct cpu_rmap *alloc_irq_cpu_rmap(unsigned int size)
59{
60 return alloc_cpu_rmap(size, GFP_KERNEL);
61}
62extern void free_irq_cpu_rmap(struct cpu_rmap *rmap);
63
64extern int irq_cpu_rmap_add(struct cpu_rmap *rmap, int irq);
65
Amir Vadai122733a2012-07-18 22:33:50 +000066#endif /* __LINUX_CPU_RMAP_H */