Updated task types for lvl b and c events

1 files changed, 6 insertions, 0 deletions

diff --git a/litmus/trace.c b/litmus/trace.c
index 1d4593aec71d..219fb46deb34 100644
--- a/litmus/trace.c
+++ b/litmus/trace.c
@@ -92,6 +92,12 @@ feather_callback void save_timestamp_task(unsigned long event,
         if (TS_LVLA_SCHED_END_ID == event) {
                 if (rt && CRIT_LEVEL_A == tsk_mc_crit(ts))
                         type = TSK_LVLA;
+        } else if (TS_LVLB_SCHED_END_ID == event) {
+                if (rt && CRIT_LEVEL_B == tsk_mc_crit(ts))
+                        type = TSK_LVLB;
+        } else if (TS_LVLC_SCHED_END_ID == event) {
+                if (rt && CRIT_LEVEL_C == tsk_mc_crit(ts))
+                        type = TSK_LVLC;
         }
         __save_timestamp(event, type);
 }