*/
 atomic_t nmi_active = ATOMIC_INIT(0);          /* oprofile uses this */
 EXPORT_SYMBOL(nmi_active);
-
+static int nmi_init_done;
 static unsigned int nmi_hz = HZ;
 static DEFINE_PER_CPU(short, wd_enabled);
 static int endflag __initdata;
 
 void stop_nmi_watchdog(void *unused)
 {
+       if (!__this_cpu_read(wd_enabled))
+               return;
        pcr_ops->write_pcr(0, pcr_ops->pcr_nmi_disable);
        __this_cpu_write(wd_enabled, 0);
        atomic_dec(&nmi_active);
 
 void start_nmi_watchdog(void *unused)
 {
+       if (__this_cpu_read(wd_enabled))
+               return;
+
        __this_cpu_write(wd_enabled, 1);
        atomic_inc(&nmi_active);
 
                }
        }
 
+       nmi_init_done = 1;
+
        return err;
 }
 
        return 0;
 }
 __setup("nmi_watchdog=", setup_nmi_watchdog);
+
+/*
+ * sparc specific NMI watchdog enable function.
+ * Enables watchdog if it is not enabled already.
+ */
+int watchdog_nmi_enable(unsigned int cpu)
+{
+       if (atomic_read(&nmi_active) == -1) {
+               pr_warn("NMI watchdog cannot be enabled or disabled\n");
+               return -1;
+       }
+
+       /*
+        * watchdog thread could start even before nmi_init is called.
+        * Just Return in that case. Let nmi_init finish the init
+        * process first.
+        */
+       if (!nmi_init_done)
+               return 0;
+
+       smp_call_function_single(cpu, start_nmi_watchdog, NULL, 1);
+
+       return 0;
+}
+/*
+ * sparc specific NMI watchdog disable function.
+ * Disables watchdog if it is not disabled already.
+ */
+void watchdog_nmi_disable(unsigned int cpu)
+{
+       if (atomic_read(&nmi_active) == -1)
+               pr_warn_once("NMI watchdog cannot be enabled or disabled\n");
+       else
+               smp_call_function_single(cpu, stop_nmi_watchdog, NULL, 1);
+}