[PATCH v3 17/41] x86/tsc: WARN if TSC sched_clock save/restore used with PV sched_clock

[Sean Christopherson <seanjc@xxxxxxxxxx>]

Now that all PV clocksources override the sched_clock save/restore hooks
when overriding sched_clock, WARN if the "default" TSC hooks are invoked
when using a PV sched_clock, e.g. to guard against regressions.

Signed-off-by: Sean Christopherson <seanjc@xxxxxxxxxx>
---
 arch/x86/kernel/tsc.c | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/arch/x86/kernel/tsc.c b/arch/x86/kernel/tsc.c
index 567d30b30a5a..b14c4ada89a3 100644
--- a/arch/x86/kernel/tsc.c
+++ b/arch/x86/kernel/tsc.c
@@ -984,9 +984,17 @@ EXPORT_SYMBOL_GPL(recalibrate_cpu_khz);
 
 static unsigned long long cyc2ns_suspend;
 
+static __always_inline bool tsc_is_save_restore_needed(void)
+{
+       if (WARN_ON_ONCE(!using_native_sched_clock()))
+               return false;
+
+       return static_branch_likely(&__use_tsc) || sched_clock_stable();
+}
+
 void tsc_save_sched_clock_state(void)
 {
-       if (!static_branch_likely(&__use_tsc) && !sched_clock_stable())
+       if (!tsc_is_save_restore_needed())
                return;
 
        cyc2ns_suspend = sched_clock();
@@ -1006,7 +1014,7 @@ void tsc_restore_sched_clock_state(void)
        unsigned long flags;
        int cpu;
 
-       if (!static_branch_likely(&__use_tsc) && !sched_clock_stable())
+       if (!tsc_is_save_restore_needed())
                return;
 
        local_irq_save(flags);
-- 
2.54.0.563.g4f69b47b94-goog