tools/testing/selftests/bpf/benchs/bench_count.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright (c) 2020 Facebook */
   3 #include "bench.h"
   4
   5 /* COUNT-GLOBAL benchmark */
   6
   7 static struct count_global_ctx {
   8         struct counter hits;
   9 } count_global_ctx;
  10
  11 static void *count_global_producer(void *input)
  12 {
  13         struct count_global_ctx *ctx = &count_global_ctx;
  14
  15         while (true) {
  16                 atomic_inc(&ctx->hits.value);
  17         }
  18         return NULL;
  19 }
  20
  21 static void *count_global_consumer(void *input)
  22 {
  23         return NULL;
  24 }
  25
  26 static void count_global_measure(struct bench_res *res)
  27 {
  28         struct count_global_ctx *ctx = &count_global_ctx;
  29
  30         res->hits = atomic_swap(&ctx->hits.value, 0);
  31 }
  32
  33 /* COUNT-local benchmark */
  34
  35 static struct count_local_ctx {
  36         struct counter *hits;
  37 } count_local_ctx;
  38
  39 static void count_local_setup()
  40 {
  41         struct count_local_ctx *ctx = &count_local_ctx;
  42
  43         ctx->hits = calloc(env.consumer_cnt, sizeof(*ctx->hits));
  44         if (!ctx->hits)
  45                 exit(1);
  46 }
  47
  48 static void *count_local_producer(void *input)
  49 {
  50         struct count_local_ctx *ctx = &count_local_ctx;
  51         int idx = (long)input;
  52
  53         while (true) {
  54                 atomic_inc(&ctx->hits[idx].value);
  55         }
  56         return NULL;
  57 }
  58
  59 static void *count_local_consumer(void *input)
  60 {
  61         return NULL;
  62 }
  63
  64 static void count_local_measure(struct bench_res *res)
  65 {
  66         struct count_local_ctx *ctx = &count_local_ctx;
  67         int i;
  68
  69         for (i = 0; i < env.producer_cnt; i++) {
  70                 res->hits += atomic_swap(&ctx->hits[i].value, 0);
  71         }
  72 }
  73
  74 const struct bench bench_count_global = {
  75         .name = "count-global",
  76         .producer_thread = count_global_producer,
  77         .consumer_thread = count_global_consumer,
  78         .measure = count_global_measure,
  79         .report_progress = hits_drops_report_progress,
  80         .report_final = hits_drops_report_final,
  81 };
  82
  83 const struct bench bench_count_local = {
  84         .name = "count-local",
  85         .setup = count_local_setup,
  86         .producer_thread = count_local_producer,
  87         .consumer_thread = count_local_consumer,
  88         .measure = count_local_measure,
  89         .report_progress = hits_drops_report_progress,
  90         .report_final = hits_drops_report_final,
  91 };