add arch-specific trace points and resched hook to x86

1 files changed, 6 insertions, 0 deletions

diff --git a/arch/x86/kernel/smp_32.c b/arch/x86/kernel/smp_32.c
index fcaa026eb807..8ce9cac79728 100644
--- a/arch/x86/kernel/smp_32.c
+++ b/arch/x86/kernel/smp_32.c
@@ -25,6 +25,9 @@
 #include <asm/mmu_context.h>
 #include <mach_apic.h>
 
+#include <litmus/litmus.h>
+#include <litmus/trace.h>
+
 /*
  *      Some notes on x86 processor bugs affecting SMP operation:
  *
@@ -472,6 +475,7 @@ void flush_tlb_all(void)
 static void native_smp_send_reschedule(int cpu)
 {
         WARN_ON(cpu_is_offline(cpu));
+        TS_SEND_RESCHED_START(cpu);
         send_IPI_mask(cpumask_of_cpu(cpu), RESCHEDULE_VECTOR);
 }
 
@@ -641,7 +645,9 @@ static void native_smp_send_stop(void)
 fastcall void smp_reschedule_interrupt(struct pt_regs *regs)
 {
         ack_APIC_irq();
+        set_tsk_need_resched(current);
         __get_cpu_var(irq_stat).irq_resched_count++;
+        TS_SEND_RESCHED_END;
 }
 
 fastcall void smp_call_function_interrupt(struct pt_regs *regs)