sched_trace: allow tasks to block on a lock id

1 files changed, 4 insertions, 0 deletions

diff --git a/litmus/sched_gsn_edf.c b/litmus/sched_gsn_edf.c
index c5c9600c33d8..3d32f74ff51c 100644
--- a/litmus/sched_gsn_edf.c
+++ b/litmus/sched_gsn_edf.c
@@ -765,6 +765,8 @@ int gsnedf_fmlp_lock(struct litmus_lock* l)
                 /* release lock before sleeping */
                 spin_unlock_irqrestore(&sem->wait.lock, flags);
 
+                sched_trace_task_block(t, l->id);
+
                 /* We depend on the FIFO order.  Thus, we don't need to recheck
                  * when we wake up; we are guaranteed to have the lock since
                  * there is only one wake up per release.
@@ -772,6 +774,8 @@ int gsnedf_fmlp_lock(struct litmus_lock* l)
 
                 schedule();
 
+                sched_trace_task_resume(t, l->id);
+
                 /* Since we hold the lock, no other task will change
                  * ->owner. We can thus check it without acquiring the spin
                  * lock. */