2 files changed, 14 insertions, 1 deletions

diff --git a/litmus/pgm.c b/litmus/pgm.c
index f8b857de8118..2bfa8d0843c1 100644
--- a/litmus/pgm.c
+++ b/litmus/pgm.c
@@ -4,6 +4,7 @@
 #include <linux/sched.h>
 #include <litmus/litmus.h>
 #include <litmus/pgm.h>
+#include <litmus/sched_trace.h>
 
 /* Only readjust release/deadline if difference is over a given threshold.
    It's a weak method for accounting overheads. Ideally, we'd know the last
@@ -51,4 +52,6 @@ void setup_pgm_release(struct task_struct* t)
                         "cur time = %llu, release time = %llu\n",
                         now, tsk_rt(t)->job_params.release);
         }
+
+        sched_trace_pgm_release(t);
 }
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index 933e7e4c7094..422314f29a60 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -231,10 +231,20 @@ feather_callback void do_sched_trace_action(unsigned long id,
 {
         struct task_struct *t = (struct task_struct*) _task;
         struct st_event_record* rec = get_record(ST_ACTION, t);
-
         if (rec) {
                 rec->data.action.when   = now();
                 rec->data.action.action = action;
                 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);
+                put_record(rec);
+        }
+}