msm: restart: use new restart infrastructure

The hook for the restart function has been moved to the machine
descriptor, so clean this up for appropriate targets.

Change-Id: I00f289bd6af245378d2d5af4a583fa8b8b0bb371
Signed-off-by: Jeff Ohlstein <johlstei@codeaurora.org>
diff --git a/arch/arm/mach-msm/include/mach/restart.h b/arch/arm/mach-msm/include/mach/restart.h
index 84df9bc..b913e3f 100644
--- a/arch/arm/mach-msm/include/mach/restart.h
+++ b/arch/arm/mach-msm/include/mach/restart.h
@@ -17,8 +17,11 @@
 #define RESTART_NORMAL 0x0
 #define RESTART_DLOAD  0x1
 
-#ifdef CONFIG_MSM_NATIVE_RESTART
+#if defined(CONFIG_MSM_NATIVE_RESTART)
 void msm_set_restart_mode(int mode);
+void msm_restart(char mode, const char *cmd);
+#elif defined(CONFIG_ARCH_FSM9XXX)
+void fsm_restart(char mode, const char *cmd);
 #else
 #define msm_set_restart_mode(mode)
 #endif