1 files changed, 12 insertions, 15 deletions

diff --git a/arch/mips/kernel/mcount.S b/arch/mips/kernel/mcount.S
index f2ee7e1e3342..cff52b283e03 100644
--- a/arch/mips/kernel/mcount.S
+++ b/arch/mips/kernel/mcount.S
@@ -119,10 +119,20 @@ NESTED(_mcount, PT_SIZE, ra)
 EXPORT_SYMBOL(_mcount)
         PTR_LA  t1, ftrace_stub
         PTR_L   t2, ftrace_trace_function /* Prepare t2 for (1) */
-        bne     t1, t2, static_trace
+        beq     t1, t2, fgraph_trace
          nop
 
+        MCOUNT_SAVE_REGS
+
+        move    a0, ra          /* arg1: self return address */
+        jalr    t2              /* (1) call *ftrace_trace_function */
+         move   a1, AT          /* arg2: parent's return address */
+
+        MCOUNT_RESTORE_REGS
+
+fgraph_trace:
 #ifdef  CONFIG_FUNCTION_GRAPH_TRACER
+        PTR_LA  t1, ftrace_stub
         PTR_L   t3, ftrace_graph_return
         bne     t1, t3, ftrace_graph_caller
          nop
@@ -131,24 +141,11 @@ EXPORT_SYMBOL(_mcount)
         bne     t1, t3, ftrace_graph_caller
          nop
 #endif
-        b       ftrace_stub
-#ifdef CONFIG_32BIT
-         addiu sp, sp, 8
-#else
-         nop
-#endif
 
-static_trace:
-        MCOUNT_SAVE_REGS
-
-        move    a0, ra          /* arg1: self return address */
-        jalr    t2              /* (1) call *ftrace_trace_function */
-         move   a1, AT          /* arg2: parent's return address */
-
-        MCOUNT_RESTORE_REGS
 #ifdef CONFIG_32BIT
         addiu sp, sp, 8
 #endif
+
         .globl ftrace_stub
 ftrace_stub:
         RETURN_BACK