frame.fp = regs->regs[29];
        frame.pc = regs->pc;
 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
-       frame.graph = current->curr_ret_stack;
+       frame.graph = 0;
 #endif
 
        walk_stackframe(current, &frame, callchain_trace, entry);
 
        frame.fp = thread_saved_fp(p);
        frame.pc = thread_saved_pc(p);
 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
-       frame.graph = p->curr_ret_stack;
+       frame.graph = 0;
 #endif
        do {
                if (unwind_frame(p, &frame))
 
        frame.fp = (unsigned long)__builtin_frame_address(0);
        frame.pc = (unsigned long)return_address; /* dummy */
 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
-       frame.graph = current->curr_ret_stack;
+       frame.graph = 0;
 #endif
 
        walk_stackframe(current, &frame, save_return_addr, &data);
 
 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
        if (tsk->ret_stack &&
                        (frame->pc == (unsigned long)return_to_handler)) {
-               if (WARN_ON_ONCE(frame->graph == -1))
-                       return -EINVAL;
+               struct ftrace_ret_stack *ret_stack;
                /*
                 * This is a case where function graph tracer has
                 * modified a return address (LR) in a stack frame
                 * to hook a function return.
                 * So replace it to an original value.
                 */
-               frame->pc = tsk->ret_stack[frame->graph--].ret;
+               ret_stack = ftrace_graph_get_ret_stack(tsk, frame->graph++);
+               if (WARN_ON_ONCE(!ret_stack))
+                       return -EINVAL;
+               frame->pc = ret_stack->ret;
        }
 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
 
        frame.fp = regs->regs[29];
        frame.pc = regs->pc;
 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
-       frame.graph = current->curr_ret_stack;
+       frame.graph = 0;
 #endif
 
        walk_stackframe(current, &frame, save_trace, &data);
                frame.pc = (unsigned long)__save_stack_trace;
        }
 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
-       frame.graph = tsk->curr_ret_stack;
+       frame.graph = 0;
 #endif
 
        walk_stackframe(tsk, &frame, save_trace, &data);
 
        frame.fp = regs->regs[29];
        frame.pc = regs->pc;
 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
-       frame.graph = current->curr_ret_stack;
+       frame.graph = 0;
 #endif
        do {
                int ret = unwind_frame(NULL, &frame);