1 files changed, 18 insertions, 0 deletions

diff --git a/tools/perf/builtin-record.c b/tools/perf/builtin-record.c
index 3fdfbaebd95e..6c3719ac901d 100644
--- a/tools/perf/builtin-record.c
+++ b/tools/perf/builtin-record.c
@@ -1656,6 +1656,21 @@ static int parse_clockid(const struct option *opt, const char *str, int unset)
         return -1;
 }
 
+static int record__parse_affinity(const struct option *opt, const char *str, int unset)
+{
+        struct record_opts *opts = (struct record_opts *)opt->value;
+
+        if (unset || !str)
+                return 0;
+
+        if (!strcasecmp(str, "node"))
+                opts->affinity = PERF_AFFINITY_NODE;
+        else if (!strcasecmp(str, "cpu"))
+                opts->affinity = PERF_AFFINITY_CPU;
+
+        return 0;
+}
+
 static int record__parse_mmap_pages(const struct option *opt,
                                     const char *str,
                                     int unset __maybe_unused)
@@ -1964,6 +1979,9 @@ static struct option __record_options[] = {
                      &nr_cblocks_default, "n", "Use <n> control blocks in asynchronous trace writing mode (default: 1, max: 4)",
                      record__aio_parse),
 #endif
+        OPT_CALLBACK(0, "affinity", &record.opts, "node|cpu",
+                     "Set affinity mask of trace reading thread to NUMA node cpu mask or cpu of processed mmap buffer",
+                     record__parse_affinity),
         OPT_END()
 };