Signed-off-by: Yoichi Yuasa <yoichi_yuasa@tripeaks.co.jp>
Signed-off-by: Ralf Baechle <ralf@linux-mips.org>
  *
  * Copyright (C) 1997 Ralf Baechle
  */
+#include <linux/irqflags.h>
 #include <linux/kernel.h>
 
 #include <asm/cacheflush.h>
 #include <asm/mipsregs.h>
+#include <asm/processor.h>
 
 void wrppmc_machine_restart(char *command)
 {
 
        printk(KERN_NOTICE "You can safely turn off the power\n");
        while (1) {
-               __asm__(
-                       ".set\tmips3\n\t"
-                       "wait\n\t"
-                       ".set\tmips0"
-               );
+               if (cpu_wait)
+                       cpu_wait();
        }
 }