[PATCH v11] perf: Sharing PMU counters across compatible events

From: Song Liu
Date: Tue Mar 24 2020 - 17:41:41 EST


This patch tries to enable PMU sharing. When multiple perf_events are
counting the same metric, they can share the hardware PMU counter. We
call these events as "compatible events".

The PMU sharing are limited to events within the same perf_event_context
(ctx). When a event is installed or enabled, search the ctx for compatible
events. This is implemented in perf_event_setup_dup(). One of these
compatible events are picked as the master (stored in event->dup_master).
Similarly, when the event is removed or disabled, perf_event_remove_dup()
is used to clean up sharing.

If the master event is a cgroup event or a event in a group, it is
possible that some slave events are ACTIVE, but the master event is not.
To handle this scenario, we introduced PERF_EVENT_STATE_ENABLED. Also,
since PMU drivers write into event->count, master event needs another
variable (master_count) for the reading of this event.

On the critical paths (add, del read), sharing PMU counters doesn't
increase the complexity. Helper functions event_pmu_[add|del|read]() are
introduced to cover these cases. All these functions have O(1) time
complexity.

Cc: Peter Zijlstra <peterz@xxxxxxxxxxxxx>
Cc: Arnaldo Carvalho de Melo <acme@xxxxxxxxxx>
Cc: Jiri Olsa <jolsa@xxxxxxxxxx>
Cc: Alexey Budankov <alexey.budankov@xxxxxxxxxxxxxxx>
Cc: Namhyung Kim <namhyung@xxxxxxxxxx>
Cc: Tejun Heo <tj@xxxxxxxxxx>
Reported-by: kernel test robot <rong.a.chen@xxxxxxxxx>
Signed-off-by: Song Liu <songliubraving@xxxxxx>

---
Changes in v12:
Fix new failures perf_event_tests. (kernel test robot)

Changes in v11:
Fold in major fixes by Peter.
In perf_event_remove_dup(), del() old master before add() new master.

Changes in v10:
Simplify logic that calls perf_event_setup_dup() and
perf_event_remove_dup(). (Peter)
Other small fixes. (Peter)

Changes in v9:
Avoid ctx_resched() on remove/disable event (Peter).
Compare the whole perf_event_attr in perf_event_compatible().
Small fixes/improvements (Peter).

Changes in v8:
Fix issues with task event (Jiri).
Fix issues with event inherit.
Fix mmap'ed events, i.e. perf test 4 (kernel test bot).

Changes in v7:
Major rewrite to avoid allocating extra master event.
(cherry picked from commit 8b98afc0ec100b017a5e80c184809939d4f01f8d)
---
arch/x86/events/core.c | 7 +
include/linux/perf_event.h | 21 ++-
kernel/events/core.c | 375 ++++++++++++++++++++++++++++++++++---
3 files changed, 373 insertions(+), 30 deletions(-)

diff --git a/arch/x86/events/core.c b/arch/x86/events/core.c
index 3bb738f5a472..32f836dbfae3 100644
--- a/arch/x86/events/core.c
+++ b/arch/x86/events/core.c
@@ -2341,6 +2341,11 @@ static int x86_pmu_aux_output_match(struct perf_event *event)
return 0;
}

+static void x86_copy_hw_config(struct perf_event *old, struct perf_event *new)
+{
+ new->hw.idx = old->hw.idx;
+}
+
static struct pmu pmu = {
.pmu_enable = x86_pmu_enable,
.pmu_disable = x86_pmu_disable,
@@ -2369,6 +2374,8 @@ static struct pmu pmu = {
.check_period = x86_pmu_check_period,

.aux_output_match = x86_pmu_aux_output_match,
+
+ .copy_hw_config = x86_copy_hw_config,
};

void arch_perf_update_userpage(struct perf_event *event,
diff --git a/include/linux/perf_event.h b/include/linux/perf_event.h
index 8768a39b5258..14b94535e213 100644
--- a/include/linux/perf_event.h
+++ b/include/linux/perf_event.h
@@ -503,6 +503,13 @@ struct pmu {
* Check period value for PERF_EVENT_IOC_PERIOD ioctl.
*/
int (*check_period) (struct perf_event *event, u64 value); /* optional */
+
+ /*
+ * Copy hw configuration from one event to another. This is used
+ * to make switching master faster in PMC sharing.
+ */
+ void (*copy_hw_config) (struct perf_event *old,
+ struct perf_event *new); /* optional */
};

enum perf_addr_filter_action_t {
@@ -552,6 +559,10 @@ struct perf_addr_filter_range {

/**
* enum perf_event_state - the states of an event:
+ *
+ * PERF_EVENT_STATE_ENABLED: Special state for PMC sharing: the hw PMC
+ * is enabled, but this event is not counting.
+ * See perf_event_init_dup_master().
*/
enum perf_event_state {
PERF_EVENT_STATE_DEAD = -4,
@@ -559,7 +570,8 @@ enum perf_event_state {
PERF_EVENT_STATE_ERROR = -2,
PERF_EVENT_STATE_OFF = -1,
PERF_EVENT_STATE_INACTIVE = 0,
- PERF_EVENT_STATE_ACTIVE = 1,
+ PERF_EVENT_STATE_ENABLED = 1,
+ PERF_EVENT_STATE_ACTIVE = 2,
};

struct file;
@@ -645,6 +657,7 @@ struct perf_event {
int group_caps;

struct perf_event *group_leader;
+ struct perf_event *dup_master; /* for PMU sharing */
struct pmu *pmu;
void *pmu_private;

@@ -762,6 +775,12 @@ struct perf_event {
void *security;
#endif
struct list_head sb_list;
+
+ int dup_active;
+ /* See event_pmu_read_dup() */
+ local64_t dup_count;
+ /* See perf_event_init_dup_master() */
+ local64_t master_count;
#endif /* CONFIG_PERF_EVENTS */
};

diff --git a/kernel/events/core.c b/kernel/events/core.c
index b27923e6b801..c57d8e0dfce0 100644
--- a/kernel/events/core.c
+++ b/kernel/events/core.c
@@ -995,7 +995,6 @@ perf_cgroup_event_enable(struct perf_event *event, struct perf_event_context *ct
*/
cpuctx = container_of(ctx, struct perf_cpu_context, ctx);

- pr_info("%s cpu %d\n", __func__, smp_processor_id());
/*
* Since setting cpuctx->cgrp is conditional on the current @cgrp
* matching the event's cgroup, we must do this for every new event,
@@ -1005,8 +1004,6 @@ perf_cgroup_event_enable(struct perf_event *event, struct perf_event_context *ct
if (ctx->is_active && !cpuctx->cgrp) {
struct perf_cgroup *cgrp = perf_cgroup_from_task(current, ctx);

- pr_info("%s cpu %d cgrp %px\n", __func__, smp_processor_id(),
- cgrp);
if (cgroup_is_descendant(cgrp->css.cgroup, event->cgrp->css.cgroup))
cpuctx->cgrp = cgrp;
}
@@ -1778,6 +1775,240 @@ perf_event_groups_next(struct perf_event *event)
event = rb_entry_safe(rb_next(&event->group_node), \
typeof(*event), group_node))

+static inline bool perf_event_can_share(struct perf_event *event)
+{
+ /* only share hardware counting events */
+ return !is_sampling_event(event);
+ return !is_software_event(event) && !is_sampling_event(event);
+}
+
+/*
+ * Returns whether the two events can share a PMU counter.
+ *
+ * Note: This function does NOT check perf_event_can_share() for
+ * the two events, they should be checked before this function
+ */
+static inline bool perf_event_compatible(struct perf_event *event_a,
+ struct perf_event *event_b)
+{
+ return memcmp(&event_a->attr, &event_b->attr, event_a->attr.size) == 0;
+}
+
+static void perf_event_init_dup_master(struct perf_event *event)
+{
+ bool is_active = event->state == PERF_EVENT_STATE_ACTIVE;
+ s64 count;
+
+ WARN_ON_ONCE(event->dup_active != 0);
+ event->dup_master = event;
+ /*
+ * The event sharing scheme allows for duplicate events to be ACTIVE
+ * while the master is not. In order to facilitate this, the master
+ * will be put in the ENABLED state whenever it has active duplicates
+ * but is itself *not* ACTIVE.
+ *
+ * When ENABLED the master event is scheduled, but its counter must
+ * appear stalled. Since the PMU driver updates event->count, the
+ * master must keep a shadow counter for itself, this is
+ * event->master_count.
+ */
+ count = local64_read(&event->count);
+ local64_set(&event->master_count, count);
+
+ if (is_active) {
+ local64_set(&event->dup_count, count);
+ event->dup_active = 1;
+ }
+
+ barrier();
+
+ WRITE_ONCE(event->dup_master, event);
+}
+
+/* tear down dup_master, no more sharing for this event */
+static void perf_event_exit_dup_master(struct perf_event *event)
+{
+ WARN_ON_ONCE(event->state < PERF_EVENT_STATE_OFF ||
+ event->state > PERF_EVENT_STATE_INACTIVE);
+
+ /* restore event->count and event->child_count */
+ local64_set(&event->count, local64_read(&event->master_count));
+
+ event->dup_active = 0;
+ WRITE_ONCE(event->dup_master, NULL);
+
+ barrier();
+}
+
+#define EVENT_TOMBSTONE ((void *)-1L)
+
+/*
+ * sync data count from dup_master to event, called on event_pmu_read()
+ * and event_pmu_del()
+ */
+static void
+event_pmu_read_dup(struct perf_event *event, struct perf_event *master)
+{
+ u64 prev_count, new_count;
+
+ if (master == EVENT_TOMBSTONE)
+ return;
+
+again:
+ prev_count = local64_read(&event->dup_count);
+ if (master->state > PERF_EVENT_STATE_INACTIVE)
+ master->pmu->read(master);
+ new_count = local64_read(&master->count);
+ if (local64_cmpxchg(&event->dup_count, prev_count, new_count) != prev_count)
+ goto again;
+
+ if (event == master)
+ local64_add(new_count - prev_count, &event->master_count);
+ else
+ local64_add(new_count - prev_count, &event->count);
+}
+
+/* After adding a event to the ctx, try find compatible event(s). */
+static void
+perf_event_setup_dup(struct perf_event *event, struct perf_event_context *ctx)
+{
+ struct perf_event *tmp;
+
+ if (event->state < PERF_EVENT_STATE_INACTIVE ||
+ !perf_event_can_share(event))
+ return;
+
+ /* look for dup with other events */
+ list_for_each_entry(tmp, &ctx->event_list, event_entry) {
+ if (tmp == event ||
+ !perf_event_can_share(tmp) ||
+ !perf_event_compatible(event, tmp) ||
+ tmp->state < PERF_EVENT_STATE_INACTIVE)
+ continue;
+
+ /* first dup, pick tmp as the master */
+ if (!tmp->dup_master) {
+ if (tmp->state == PERF_EVENT_STATE_ACTIVE)
+ tmp->pmu->read(tmp);
+ perf_event_init_dup_master(tmp);
+ }
+
+ event->dup_master = tmp->dup_master;
+
+ break;
+ }
+}
+
+/* Remove dup_master for the event */
+static void
+perf_event_remove_dup(struct perf_event *event, struct perf_event_context *ctx)
+
+{
+ struct perf_event *tmp, *new_master;
+ int dup_count, active_count;
+ int ret;
+
+ /* no sharing */
+ if (!event->dup_master)
+ return;
+
+ WARN_ON_ONCE(event->state < PERF_EVENT_STATE_OFF ||
+ event->state > PERF_EVENT_STATE_ENABLED);
+
+ /* this event is not the master */
+ if (event->dup_master != event) {
+ event->dup_master = NULL;
+ return;
+ }
+
+ active_count = event->dup_active;
+ if (active_count) {
+ perf_pmu_disable(event->pmu);
+ event->pmu->del(event, 0);
+ }
+
+ /* this event is the master */
+ dup_count = 0;
+ new_master = NULL;
+ list_for_each_entry(tmp, &ctx->event_list, event_entry) {
+ u64 count;
+
+ if (tmp->dup_master != event || tmp == event)
+ continue;
+ if (!new_master)
+ new_master = tmp;
+
+ if (tmp->state == PERF_EVENT_STATE_ACTIVE) {
+ /* sync read from old master */
+ event_pmu_read_dup(tmp, event);
+ }
+ /*
+ * Flip an active event to a new master; this is tricky
+ * because for an active event event_pmu_read() can be
+ * called at any time from NMI context.
+ *
+ * This means we need to have ->dup_master and ->dup_count
+ * consistent at all times. Of course we cannot do two
+ * writes at once :/
+ *
+ * Instead, flip ->dup_master to EVENT_TOMBSTONE, this will
+ * make event_pmu_read_dup() NOP. Then we can set
+ * ->dup_count and finally set ->dup_master to the
+ * new_master to let event_pmu_read_dup() rip.
+ */
+ WRITE_ONCE(tmp->dup_master, EVENT_TOMBSTONE);
+ barrier();
+
+ count = local64_read(&new_master->count);
+ local64_set(&tmp->dup_count, count);
+
+ if (tmp == new_master)
+ local64_set(&tmp->master_count, count);
+
+ barrier();
+ WRITE_ONCE(tmp->dup_master, new_master);
+ dup_count++;
+ }
+
+ perf_event_exit_dup_master(event);
+
+ if (!dup_count)
+ return;
+
+ if (active_count) {
+ /* copy hardware configure to switch faster */
+ if (event->pmu->copy_hw_config)
+ event->pmu->copy_hw_config(event, new_master);
+
+ ret = new_master->pmu->add(new_master, PERF_EF_START);
+ /*
+ * Since we just removed the old master (@event), it should be
+ * impossible to fail to schedule the new master, an identical
+ * event.
+ */
+ WARN_ON_ONCE(ret);
+ if (new_master->state == PERF_EVENT_STATE_INACTIVE) {
+ /*
+ * We don't need to update time, so don't call
+ * perf_event_set_state().
+ */
+ new_master->state = PERF_EVENT_STATE_ENABLED;
+ }
+ perf_pmu_enable(new_master->pmu);
+ }
+
+ if (dup_count == 1) {
+ /*
+ * We set up as a master, but there aren't any more duplicates.
+ * Simply clear ->dup_master, as ->master_count == ->count per
+ * the above.
+ */
+ WRITE_ONCE(new_master->dup_master, NULL);
+ } else {
+ new_master->dup_active = active_count;
+ }
+}
+
/*
* Add an event from the lists for its context.
* Must be called with ctx->mutex and ctx->lock held.
@@ -1983,6 +2214,7 @@ list_del_event(struct perf_event *event, struct perf_event_context *ctx)
if (!(event->attach_state & PERF_ATTACH_CONTEXT))
return;

+ perf_event_remove_dup(event, ctx);
event->attach_state &= ~PERF_ATTACH_CONTEXT;

ctx->nr_events--;
@@ -2208,6 +2440,62 @@ event_filter_match(struct perf_event *event)
perf_cgroup_match(event) && pmu_filter_match(event);
}

+/* PMU sharing aware version of event->pmu->add() */
+static int event_pmu_add(struct perf_event *event,
+ struct perf_event_context *ctx)
+{
+ struct perf_event *master;
+ s64 value;
+ int ret;
+
+ /* no sharing, just do event->pmu->add() */
+ if (!event->dup_master)
+ return event->pmu->add(event, PERF_EF_START);
+
+ master = event->dup_master;
+
+ if (!master->dup_active) {
+ ret = event->pmu->add(master, PERF_EF_START);
+ if (ret)
+ return ret;
+
+ if (master != event)
+ perf_event_set_state(master, PERF_EVENT_STATE_ENABLED);
+ }
+
+ master->dup_active++;
+ master->pmu->read(master);
+ value = local64_read(&master->count);
+ local64_set(&event->dup_count, value);
+ return 0;
+}
+
+/* PMU sharing aware version of event->pmu->del() */
+static void event_pmu_del(struct perf_event *event,
+ struct perf_event_context *ctx)
+{
+ struct perf_event *master;
+
+ if (!event->dup_master)
+ return event->pmu->del(event, 0);
+
+ master = event->dup_master;
+ if (!--master->dup_active) {
+ event->pmu->del(master, 0);
+ perf_event_set_state(master, PERF_EVENT_STATE_INACTIVE);
+ }
+ event_pmu_read_dup(event, master);
+}
+
+/* PMU sharing aware version of event->pmu->read() */
+static void event_pmu_read(struct perf_event *event)
+{
+ if (!event->dup_master)
+ return event->pmu->read(event);
+
+ event_pmu_read_dup(event, event->dup_master);
+}
+
static void
event_sched_out(struct perf_event *event,
struct perf_cpu_context *cpuctx,
@@ -2218,7 +2506,7 @@ event_sched_out(struct perf_event *event,
WARN_ON_ONCE(event->ctx != ctx);
lockdep_assert_held(&ctx->lock);

- if (event->state != PERF_EVENT_STATE_ACTIVE)
+ if (event->state < PERF_EVENT_STATE_ENABLED)
return;

/*
@@ -2230,13 +2518,16 @@ event_sched_out(struct perf_event *event,

perf_pmu_disable(event->pmu);

- event->pmu->del(event, 0);
+ event_pmu_del(event, ctx);
event->oncpu = -1;

if (READ_ONCE(event->pending_disable) >= 0) {
WRITE_ONCE(event->pending_disable, -1);
perf_cgroup_event_disable(event, ctx);
state = PERF_EVENT_STATE_OFF;
+ } else if (event->dup_active) {
+ WARN_ON_ONCE(event->dup_master != event);
+ state = PERF_EVENT_STATE_ENABLED;
}
perf_event_set_state(event, state);

@@ -2508,7 +2799,7 @@ event_sched_in(struct perf_event *event,

perf_log_itrace_start(event);

- if (event->pmu->add(event, PERF_EF_START)) {
+ if (event_pmu_add(event, ctx)) {
perf_event_set_state(event, PERF_EVENT_STATE_INACTIVE);
event->oncpu = -1;
ret = -EAGAIN;
@@ -2620,6 +2911,7 @@ static void add_event_to_ctx(struct perf_event *event,
{
list_add_event(event, ctx);
perf_group_attach(event);
+ perf_event_setup_dup(event, ctx);
}

static void ctx_sched_out(struct perf_event_context *ctx,
@@ -2919,8 +3211,10 @@ static void __perf_event_enable(struct perf_event *event,
perf_event_set_state(event, PERF_EVENT_STATE_INACTIVE);
perf_cgroup_event_enable(event, ctx);

- if (!ctx->is_active)
+ if (!ctx->is_active) {
+ perf_event_setup_dup(event, ctx);
return;
+ }

if (!event_filter_match(event)) {
ctx_sched_in(ctx, cpuctx, EVENT_TIME, current);
@@ -2931,7 +3225,7 @@ static void __perf_event_enable(struct perf_event *event,
* If the event is in a group and isn't the group leader,
* then don't put it on unless the group is on.
*/
- if (leader != event && leader->state != PERF_EVENT_STATE_ACTIVE) {
+ if (leader != event && leader->state <= PERF_EVENT_STATE_INACTIVE) {
ctx_sched_in(ctx, cpuctx, EVENT_TIME, current);
return;
}
@@ -3279,8 +3573,8 @@ static void __perf_event_sync_stat(struct perf_event *event,
* we know the event must be on the current CPU, therefore we
* don't need to use it.
*/
- if (event->state == PERF_EVENT_STATE_ACTIVE)
- event->pmu->read(event);
+ if (event->state > PERF_EVENT_STATE_INACTIVE)
+ event_pmu_read(event);

perf_event_update_time(event);

@@ -4196,22 +4490,22 @@ static void __perf_event_read(void *info)
goto unlock;

if (!data->group) {
- pmu->read(event);
+ event_pmu_read(event);
data->ret = 0;
goto unlock;
}

pmu->start_txn(pmu, PERF_PMU_TXN_READ);

- pmu->read(event);
+ event_pmu_read(event);

for_each_sibling_event(sub, event) {
- if (sub->state == PERF_EVENT_STATE_ACTIVE) {
+ if (sub->state > PERF_EVENT_STATE_INACTIVE) {
/*
* Use sibling's PMU rather than @event's since
* sibling could be on different (eg: software) PMU.
*/
- sub->pmu->read(sub);
+ event_pmu_read(sub);
}
}

@@ -4223,7 +4517,14 @@ static void __perf_event_read(void *info)

static inline u64 perf_event_count(struct perf_event *event)
{
- return local64_read(&event->count) + atomic64_read(&event->child_count);
+ u64 count;
+
+ if (likely(event->dup_master != event))
+ count = local64_read(&event->count);
+ else
+ count = local64_read(&event->master_count);
+
+ return count + atomic64_read(&event->child_count);
}

/*
@@ -4281,9 +4582,12 @@ int perf_event_read_local(struct perf_event *event, u64 *value,
* oncpu == -1).
*/
if (event->oncpu == smp_processor_id())
- event->pmu->read(event);
+ event_pmu_read(event);

- *value = local64_read(&event->count);
+ if (event->dup_master == event)
+ *value = local64_read(&event->master_count);
+ else
+ *value = local64_read(&event->count);
if (enabled || running) {
u64 now = event->shadow_ctx_time + perf_clock();
u64 __enabled, __running;
@@ -4310,7 +4614,7 @@ static int perf_event_read(struct perf_event *event, bool group)
* value in the event structure:
*/
again:
- if (state == PERF_EVENT_STATE_ACTIVE) {
+ if (state > PERF_EVENT_STATE_INACTIVE) {
struct perf_read_data data;

/*
@@ -5580,13 +5884,15 @@ int perf_event_task_disable(void)

static int perf_event_index(struct perf_event *event)
{
- if (event->hw.state & PERF_HES_STOPPED)
+ struct perf_event *e = event->dup_master ? : event;
+
+ if (e->hw.state & PERF_HES_STOPPED)
return 0;

- if (event->state != PERF_EVENT_STATE_ACTIVE)
+ if (e->state < PERF_EVENT_STATE_ENABLED)
return 0;

- return event->pmu->event_idx(event);
+ return e->pmu->event_idx(e);
}

static void calc_timer_values(struct perf_event *event,
@@ -5665,8 +5971,11 @@ void perf_event_update_userpage(struct perf_event *event)
barrier();
userpg->index = perf_event_index(event);
userpg->offset = perf_event_count(event);
- if (userpg->index)
- userpg->offset -= local64_read(&event->hw.prev_count);
+ if (userpg->index) {
+ struct perf_event *e = event->dup_master ? : event;
+
+ userpg->offset -= local64_read(&e->hw.prev_count);
+ }

userpg->time_enabled = enabled +
atomic64_read(&event->child_total_time_enabled);
@@ -6668,8 +6977,8 @@ static void perf_output_read_group(struct perf_output_handle *handle,
values[n++] = running;

if ((leader != event) &&
- (leader->state == PERF_EVENT_STATE_ACTIVE))
- leader->pmu->read(leader);
+ (leader->state > PERF_EVENT_STATE_INACTIVE))
+ event_pmu_read(leader);

values[n++] = perf_event_count(leader);
if (read_format & PERF_FORMAT_ID)
@@ -6681,8 +6990,8 @@ static void perf_output_read_group(struct perf_output_handle *handle,
n = 0;

if ((sub != event) &&
- (sub->state == PERF_EVENT_STATE_ACTIVE))
- sub->pmu->read(sub);
+ (sub->state > PERF_EVENT_STATE_INACTIVE))
+ event_pmu_read(sub);

values[n++] = perf_event_count(sub);
if (read_format & PERF_FORMAT_ID)
@@ -9990,10 +10299,10 @@ static enum hrtimer_restart perf_swevent_hrtimer(struct hrtimer *hrtimer)

event = container_of(hrtimer, struct perf_event, hw.hrtimer);

- if (event->state != PERF_EVENT_STATE_ACTIVE)
+ if (event->state <= PERF_EVENT_STATE_INACTIVE)
return HRTIMER_NORESTART;

- event->pmu->read(event);
+ event_pmu_read(event);

perf_sample_data_init(&data, 0, event->hw.last_period);
regs = get_irq_regs();
@@ -11687,9 +11996,17 @@ SYSCALL_DEFINE5(perf_event_open,
perf_remove_from_context(group_leader, 0);
put_ctx(gctx);

+ /*
+ * move_group only happens to sw events, from sw ctx to hw
+ * ctx. The sw events should not have valid dup_master. So
+ * it is not necessary to handle dup_events.
+ */
+ WARN_ON_ONCE(group_leader->dup_master);
+
for_each_sibling_event(sibling, group_leader) {
perf_remove_from_context(sibling, 0);
put_ctx(gctx);
+ WARN_ON_ONCE(sibling->dup_master);
}

/*
--
2.17.1