1 files changed, 12 insertions, 8 deletions

diff --git a/tools/perf/bench/futex-hash.c b/tools/perf/bench/futex-hash.c
index 58ae6ed8f38b..9aa3a674829b 100644
--- a/tools/perf/bench/futex-hash.c
+++ b/tools/perf/bench/futex-hash.c
@@ -24,9 +24,9 @@
 #include <subcmd/parse-options.h>
 #include "bench.h"
 #include "futex.h"
+#include "cpumap.h"
 
 #include <err.h>
-#include <sys/time.h>
 
 static unsigned int nthreads = 0;
 static unsigned int nsecs    = 10;
@@ -118,11 +118,12 @@ static void print_summary(void)
 int bench_futex_hash(int argc, const char **argv)
 {
         int ret = 0;
-        cpu_set_t cpu;
+        cpu_set_t cpuset;
         struct sigaction act;
-        unsigned int i, ncpus;
+        unsigned int i;
         pthread_attr_t thread_attr;
         struct worker *worker = NULL;
+        struct cpu_map *cpu;
 
         argc = parse_options(argc, argv, options, bench_futex_hash_usage, 0);
         if (argc) {
@@ -130,14 +131,16 @@ int bench_futex_hash(int argc, const char **argv)
                 exit(EXIT_FAILURE);
         }
 
-        ncpus = sysconf(_SC_NPROCESSORS_ONLN);
+        cpu = cpu_map__new(NULL);
+        if (!cpu)
+                goto errmem;
 
         sigfillset(&act.sa_mask);
         act.sa_sigaction = toggle_done;
         sigaction(SIGINT, &act, NULL);
 
         if (!nthreads) /* default to the number of CPUs */
-                nthreads = ncpus;
+                nthreads = cpu->nr;
 
         worker = calloc(nthreads, sizeof(*worker));
         if (!worker)
@@ -163,10 +166,10 @@ int bench_futex_hash(int argc, const char **argv)
                 if (!worker[i].futex)
                         goto errmem;
 
-                CPU_ZERO(&cpu);
-                CPU_SET(i % ncpus, &cpu);
+                CPU_ZERO(&cpuset);
+                CPU_SET(cpu->map[i % cpu->nr], &cpuset);
 
-                ret = pthread_attr_setaffinity_np(&thread_attr, sizeof(cpu_set_t), &cpu);
+                ret = pthread_attr_setaffinity_np(&thread_attr, sizeof(cpu_set_t), &cpuset);
                 if (ret)
                         err(EXIT_FAILURE, "pthread_attr_setaffinity_np");
 
@@ -217,6 +220,7 @@ int bench_futex_hash(int argc, const char **argv)
         print_summary();
 
         free(worker);
+        free(cpu);
         return ret;
 errmem:
         err(EXIT_FAILURE, "calloc");