2 files changed, 8 insertions, 4 deletions

diff --git a/include/trace/boot.h b/include/trace/boot.h
index 6b54537eab02..088ea089e31d 100644
--- a/include/trace/boot.h
+++ b/include/trace/boot.h
@@ -1,6 +1,10 @@
 #ifndef _LINUX_TRACE_BOOT_H
 #define _LINUX_TRACE_BOOT_H
 
+#include <linux/module.h>
+#include <linux/kallsyms.h>
+#include <linux/init.h>
+
 /*
  * Structure which defines the trace of an initcall
  * while it is called.
@@ -9,7 +13,7 @@
  */
 struct boot_trace_call {
         pid_t                   caller;
-        char                    func[KSYM_NAME_LEN];
+        char                    func[KSYM_SYMBOL_LEN];
 };
 
 /*
@@ -17,7 +21,7 @@ struct boot_trace_call {
  * while it returns.
  */
 struct boot_trace_ret {
-        char                    func[KSYM_NAME_LEN];
+        char                    func[KSYM_SYMBOL_LEN];
         int                             result;
         unsigned long long      duration;               /* nsecs */
 };
diff --git a/include/trace/sched.h b/include/trace/sched.h
index 9b2854abf7e2..f4549d506b16 100644
--- a/include/trace/sched.h
+++ b/include/trace/sched.h
@@ -30,8 +30,8 @@ DECLARE_TRACE(sched_switch,
                 TPARGS(rq, prev, next));
 
 DECLARE_TRACE(sched_migrate_task,
-        TPPROTO(struct rq *rq, struct task_struct *p, int dest_cpu),
-                TPARGS(rq, p, dest_cpu));
+        TPPROTO(struct task_struct *p, int orig_cpu, int dest_cpu),
+                TPARGS(p, orig_cpu, dest_cpu));
 
 DECLARE_TRACE(sched_process_free,
         TPPROTO(struct task_struct *p),