#include <asm/page_types.h>
 #include <asm/asm-offsets.h>
 #include <asm/processor-flags.h>
+#include <asm/frame.h>
 
 ENTRY(swsusp_arch_suspend)
+       FRAME_BEGIN
        movq    $saved_context, %rax
        movq    %rsp, pt_regs_sp(%rax)
        movq    %rbp, pt_regs_bp(%rax)
        movq    %rax, restore_cr3(%rip)
 
        call swsusp_save
+       FRAME_END
        ret
+ENDPROC(swsusp_arch_suspend)
 
 ENTRY(restore_image)
        /* switch to temporary page tables */
         */
 
 ENTRY(restore_registers)
+       FRAME_BEGIN
        /* go back to the original page tables */
        movq    %rbx, %cr3
 
        /* tell the hibernation core that we've just restored the memory */
        movq    %rax, in_suspend(%rip)
 
+       FRAME_END
        ret
+ENDPROC(restore_registers)