ANDROID: sched: add cpumask parameter to cpu capacity vendor hook

Different cpu type will have different suppression ratio.
So we add cpumask parameter to cpu capacity vendor hook.

Bug: 167126223

Signed-off-by: Yun Hsiang <yun.hsiang@mediatek.com>
Change-Id: I983a3aca2195f7c37c43d31fa588c999a825ee49
diff --git a/drivers/base/arch_topology.c b/drivers/base/arch_topology.c
index 0c0753d..a7bce08 100644
--- a/drivers/base/arch_topology.c
+++ b/drivers/base/arch_topology.c
@@ -44,7 +44,7 @@
 
 	scale = (cur_freq << SCHED_CAPACITY_SHIFT) / max_freq;
 
-	trace_android_vh_arch_set_freq_scale(cur_freq, max_freq, &scale);
+	trace_android_vh_arch_set_freq_scale(cpus, cur_freq, max_freq, &scale);
 
 	for_each_cpu(i, cpus)
 		per_cpu(freq_scale, i) = scale;
diff --git a/include/trace/hooks/topology.h b/include/trace/hooks/topology.h
index 65ee196..32b44c7 100644
--- a/include/trace/hooks/topology.h
+++ b/include/trace/hooks/topology.h
@@ -9,16 +9,18 @@
 
 #include <linux/tracepoint.h>
 #include <trace/hooks/vendor_hooks.h>
+#include <linux/cpumask.h>
 
 #if defined(CONFIG_TRACEPOINTS) && defined(CONFIG_ANDROID_VENDOR_HOOKS)
 
 DECLARE_HOOK(android_vh_arch_set_freq_scale,
-	TP_PROTO(unsigned long freq, unsigned long max, unsigned long *scale),
-	TP_ARGS(freq, max, scale));
+	TP_PROTO(struct cpumask *cpus, unsigned long freq, unsigned long max,
+		unsigned long *scale),
+	TP_ARGS(cpus, freq, max, scale));
 
 #else
 
-#define trace_android_vh_arch_set_freq_scale(freq, max, scale)
+#define trace_android_vh_arch_set_freq_scale(cpus, freq, max, scale)
 
 #endif