#ifdef CONFIG_X86_64
 extern void idt_setup_early_pf(void);
+extern void idt_setup_ist_traps(void);
 extern void idt_setup_debugidt_traps(void);
 #else
 static inline void idt_setup_early_pf(void) { }
+static inline void idt_setup_ist_traps(void) { }
 static inline void idt_setup_debugidt_traps(void) { }
 #endif
 
 
 /* No need to be aligned, but done to keep all IDTs defined the same way. */
 gate_desc debug_idt_table[IDT_ENTRIES] __page_aligned_bss;
 
+/*
+ * The exceptions which use Interrupt stacks. They are setup after
+ * cpu_init() when the TSS has been initialized.
+ */
+static const __initdata struct idt_data ist_idts[] = {
+       ISTG(X86_TRAP_DB,       debug,          DEBUG_STACK),
+       ISTG(X86_TRAP_NMI,      nmi,            NMI_STACK),
+       ISTG(X86_TRAP_BP,       int3,           DEBUG_STACK),
+       ISTG(X86_TRAP_DF,       double_fault,   DOUBLEFAULT_STACK),
+#ifdef CONFIG_X86_MCE
+       ISTG(X86_TRAP_MC,       &machine_check, MCE_STACK),
+#endif
+};
+
 /*
  * Override for the debug_idt. Same as the default, but with interrupt
  * stack set to DEFAULT_STACK (0). Required for NMI trap handling.
                             ARRAY_SIZE(early_pf_idts));
 }
 
+/**
+ * idt_setup_ist_traps - Initialize the idt table with traps using IST
+ */
+void __init idt_setup_ist_traps(void)
+{
+       idt_setup_from_table(idt_table, ist_idts, ARRAY_SIZE(ist_idts));
+}
+
 /**
  * idt_setup_debugidt_traps - Initialize the debug idt table with debug traps
  */
 
         */
        cpu_init();
 
-       /*
-        * X86_TRAP_DB and X86_TRAP_BP have been set
-        * in early_trap_init(). However, ITS works only after
-        * cpu_init() loads TSS. See comments in early_trap_init().
-        */
-       set_intr_gate_ist(X86_TRAP_DB, &debug, DEBUG_STACK);
-       /* int3 can be called from all */
-       set_system_intr_gate_ist(X86_TRAP_BP, &int3, DEBUG_STACK);
+       idt_setup_ist_traps();
 
        x86_init.irqs.trap_init();