From: Yury Norov [NVIDIA] <yury.norov@gmail.com>
Date: Thu, 19 Jun 2025 18:26:24 +0000 (-0400)
Subject: cpumask: introduce cpumask_random()
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=012b1043420c0bc62e52902499de40b66f37fd6a;p=thirdparty%2Fkernel%2Flinux.git

cpumask: introduce cpumask_random()

Propagate find_random_bit() to cpumask API.

CC: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: "Yury Norov [NVIDIA]" <yury.norov@gmail.com>
---

diff --git a/include/linux/cpumask.h b/include/linux/cpumask.h
index 7ae80a7ca81ef..39b71b662da38 100644
--- a/include/linux/cpumask.h
+++ b/include/linux/cpumask.h
@@ -354,6 +354,18 @@ unsigned int cpumask_next_wrap(int n, const struct cpumask *src)
 	return find_next_bit_wrap(cpumask_bits(src), small_cpumask_bits, n + 1);
 }
 
+/**
+ * cpumask_random - get random cpu in *src.
+ * @src: cpumask pointer
+ *
+ * Return: random set bit, or >= nr_cpu_ids if @src is empty.
+ */
+static __always_inline
+unsigned int cpumask_random(const struct cpumask *src)
+{
+	return find_random_bit(cpumask_bits(src), nr_cpu_ids);
+}
+
 /**
  * for_each_cpu - iterate over every cpu in a mask
  * @cpu: the (optionally unsigned) integer iterator