From c388a7b37ffeeebd43d9352b8af3874d2394ba9a Mon Sep 17 00:00:00 2001
From: Sai Harshini Nimmala <quic_snimmala@quicinc.com>
Date: Thu, 22 Sep 2022 17:38:15 -0700
Subject: [PATCH] sched/walt: Define sched_trace_rd_span

sched_trace_rd_span is removed upstream. Restore the function locally.

Change-Id: I066be161e1da49b11de10f034e8f46a4c1510391
Signed-off-by: Sai Harshini Nimmala <quic_snimmala@quicinc.com>
---
 kernel/sched/walt/walt_tp.c | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/kernel/sched/walt/walt_tp.c b/kernel/sched/walt/walt_tp.c
index 8d7e2e4528c5..d82c4864076d 100644
--- a/kernel/sched/walt/walt_tp.c
+++ b/kernel/sched/walt/walt_tp.c
@@ -106,6 +106,15 @@ static void unregister_sched_switch_ctrs(void)
 	cpuhp_remove_state_nocalls(USE_CPUHP_STATE);
 }
 
+const struct cpumask *sched_trace_rd_span(struct root_domain *rd)
+{
+#ifdef CONFIG_SMP
+	return rd ? rd->span : NULL;
+#else
+	return NULL;
+#endif
+}
+
 static void sched_overutilized(void *data, struct root_domain *rd,
 				 bool overutilized)
 {