extern u64 pnv_first_deep_stop_state;
 
+unsigned long pnv_cpu_offline(unsigned int cpu);
 int validate_psscr_val_mask(u64 *psscr_val, u64 *psscr_mask, u32 flags);
 static inline void report_invalid_psscr_val(u64 psscr_val, int err)
 {
 
 u64 pnv_deepest_stop_psscr_val;
 u64 pnv_deepest_stop_psscr_mask;
 
+/*
+ * pnv_cpu_offline: A function that puts the CPU into the deepest
+ * available platform idle state on a CPU-Offline.
+ */
+unsigned long pnv_cpu_offline(unsigned int cpu)
+{
+       unsigned long srr1;
+
+       u32 idle_states = pnv_get_supported_cpuidle_states();
+
+       if (cpu_has_feature(CPU_FTR_ARCH_300)) {
+               srr1 = power9_idle_stop(pnv_deepest_stop_psscr_val,
+                                       pnv_deepest_stop_psscr_mask);
+       } else if (idle_states & OPAL_PM_WINKLE_ENABLED) {
+               srr1 = power7_winkle();
+       } else if ((idle_states & OPAL_PM_SLEEP_ENABLED) ||
+                  (idle_states & OPAL_PM_SLEEP_ENABLED_ER1)) {
+               srr1 = power7_sleep();
+       } else {
+               srr1 = power7_nap(1);
+       }
+
+       return srr1;
+}
+
 /*
  * Power ISA 3.0 idle initialization.
  *
 
 #endif
 
 extern u32 pnv_get_supported_cpuidle_states(void);
-extern u64 pnv_deepest_stop_psscr_val;
-extern u64 pnv_deepest_stop_psscr_mask;
 
 extern void pnv_lpc_init(void);
 
 
 #include <asm/dbell.h>
 #include <asm/kvm_ppc.h>
 #include <asm/ppc-opcode.h>
+#include <asm/cpuidle.h>
 
 #include "powernv.h"
 
 {
        unsigned int cpu;
        unsigned long srr1, wmask;
-       u32 idle_states;
 
        /* Standard hot unplug procedure */
        local_irq_disable();
        if (cpu_has_feature(CPU_FTR_ARCH_207S))
                wmask = SRR1_WAKEMASK_P8;
 
-       idle_states = pnv_get_supported_cpuidle_states();
-
        /* We don't want to take decrementer interrupts while we are offline,
         * so clear LPCR:PECE1. We keep PECE2 (and LPCR_PECE_HVEE on P9)
         * enabled as to let IPIs in.
                kvmppc_set_host_ipi(cpu, 0);
 
                ppc64_runlatch_off();
-
-               if (cpu_has_feature(CPU_FTR_ARCH_300)) {
-                       srr1 = power9_idle_stop(pnv_deepest_stop_psscr_val,
-                                               pnv_deepest_stop_psscr_mask);
-               } else if (idle_states & OPAL_PM_WINKLE_ENABLED) {
-                       srr1 = power7_winkle();
-               } else if ((idle_states & OPAL_PM_SLEEP_ENABLED) ||
-                          (idle_states & OPAL_PM_SLEEP_ENABLED_ER1)) {
-                       srr1 = power7_sleep();
-               } else {
-                       srr1 = power7_nap(1);
-               }
-
+               srr1 = pnv_cpu_offline(cpu);
                ppc64_runlatch_on();
 
                /*