1 files changed, 13 insertions, 2 deletions

diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 422314f29a60..b9e772c07652 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -238,13 +238,24 @@ feather_callback void do_sched_trace_action(unsigned long id,
         }
 }
 
+feather_callback void do_sched_trace_pgm_param(unsigned long id, unsigned long _task)
+{
+        struct task_struct *t = (struct task_struct*) _task;
+        struct st_event_record* rec = get_record(ST_PGM_PARAM, t);
+        if (rec) {
+                rec->data.pgm_param.node_type = tsk_rt(t)->task_params.pgm_type;
+                rec->data.pgm_param.graph_pid = t->tgid;
+                put_record(rec);
+        }
+}
+
 feather_callback void do_sched_trace_pgm_release(unsigned long id, unsigned long _task)
 {
         struct task_struct *t = (struct task_struct*) _task;
         struct st_event_record* rec = get_record(ST_PGM_RELEASE, t);
         if (rec) {
-                rec->data.release.release  = get_release(t);
-                rec->data.release.deadline = get_deadline(t);
+                rec->data.pgm_release.release  = get_release(t);
+                rec->data.pgm_release.deadline = get_deadline(t);
                 put_record(rec);
         }
 }