[RFC v2 6/7] Make GRUB a task's flag

[Luca Abeni]

Signed-off-by: Luca Abeni <luca.ab...@unitn.it>
---
 include/uapi/linux/sched.h | 1 +
 kernel/sched/core.c        | 3 ++-
 kernel/sched/deadline.c    | 3 ++-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/include/uapi/linux/sched.h b/include/uapi/linux/sched.h
index cc89dde..9279562 100644
--- a/include/uapi/linux/sched.h
+++ b/include/uapi/linux/sched.h
@@ -48,5 +48,6 @@
  * For the sched_{set,get}attr() calls
  */
 #define SCHED_FLAG_RESET_ON_FORK       0x01
+#define SCHED_FLAG_RECLAIM             0x02
 
 #endif /* _UAPI_LINUX_SCHED_H */
diff --git a/kernel/sched/core.c b/kernel/sched/core.c
index 4158d1f..3224132 100644
--- a/kernel/sched/core.c
+++ b/kernel/sched/core.c
@@ -3822,7 +3822,8 @@ recheck:
                        return -EINVAL;
        }
 
-       if (attr->sched_flags & ~(SCHED_FLAG_RESET_ON_FORK))
+       if (attr->sched_flags &
+               ~(SCHED_FLAG_RESET_ON_FORK | SCHED_FLAG_RECLAIM))
                return -EINVAL;
 
        /*
diff --git a/kernel/sched/deadline.c b/kernel/sched/deadline.c
index 647d779..b56f76f 100644
--- a/kernel/sched/deadline.c
+++ b/kernel/sched/deadline.c
@@ -834,7 +834,8 @@ static void update_curr_dl(struct rq *rq)
 
        sched_rt_avg_update(rq, delta_exec);
 
-       delta_exec = grub_reclaim(delta_exec, rq);
+       if (unlikely(dl_se->flags & SCHED_FLAG_RECLAIM))
+               delta_exec = grub_reclaim(delta_exec, rq);
        dl_se->runtime -= delta_exec;
 
 throttle:
-- 
2.5.0