Return-Path: Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S964783AbWHDFIf (ORCPT ); Fri, 4 Aug 2006 01:08:35 -0400 Received: (majordomo@vger.kernel.org) by vger.kernel.org id S1751433AbWHDFIf (ORCPT ); Fri, 4 Aug 2006 01:08:35 -0400 Received: from e33.co.us.ibm.com ([32.97.110.151]:16358 "EHLO e33.co.us.ibm.com") by vger.kernel.org with ESMTP id S1751434AbWHDFId (ORCPT ); Fri, 4 Aug 2006 01:08:33 -0400 Date: Fri, 4 Aug 2006 10:43:00 +0530 From: Srivatsa Vaddagiri To: Ingo Molnar , Nick Piggin Cc: Sam Vilain , linux-kernel@vger.kernel.org, Kirill Korotaev , Mike Galbraith , Balbir Singh , sekharan@us.ibm.com, Andrew Morton , nagar@watson.ibm.com, haveblue@us.ibm.com, pj@sgi.com Subject: [ RFC, PATCH 5/5 ] CPU controller - interface with cpusets Message-ID: <20060804051300.GI27194@in.ibm.com> Reply-To: vatsa@in.ibm.com References: <20060804050753.GD27194@in.ibm.com> Mime-Version: 1.0 Content-Type: text/plain; charset=us-ascii Content-Disposition: inline In-Reply-To: <20060804050753.GD27194@in.ibm.com> User-Agent: Mutt/1.5.11 Sender: linux-kernel-owner@vger.kernel.org X-Mailing-List: linux-kernel@vger.kernel.org Content-Length: 11439 Lines: 420 A controller cannot live in isolation. *somebody* needs to tell it about events like new-task-group creation, destruction of existing group, assignment of bandwidth and finally movement of tasks between task-groups. We also need *some* interface for grouping tasks and assigning bandwidth to the task-groups. cpusets is chosen as this interface for now. This patch makes some extension to cpusets as below: - Every cpuset directory has two new files - meter_cpu, cpu_quota - Writing '1' into 'meter_cpu' marks the cpuset as "metered" i.e tasks in this cpuset and its child cpusets will have their CPU usage controlled by the controller as per the quota settings mentioned in 'cpu_quota'. - Writing a number between 0-100 into 'cpu_quota' file of a metered cpuset assigns corresponding bandwidth for all tasks in the cpuset (task-group). - Only cpusets marked as "exclusive" and not having any child cpusets can be metered. - Metered cpusets cannot have grand-children! Metered cpusets was discussed elaborately at: http://lkml.org/lkml/2005/9/26/58 This patch was inspired from those discussions and is a subset of the features provided in the earlier patch. As an example, follow these steps to create metered cpusets: # cd /dev # mkdir cpuset # mount -t cpuset cpuset cpuset # cd cpuset # mkdir grp_a # cd grp_a # /bin/echo 1 > cpus # assign CPU 1 for this cpuset # /bin/echo 0 > mems # assign node 0 for this cpuset # /bin/echo 1 > cpu_exclusive # /bin/echo 1 > meter_cpu # mkdir very_imp_grp # Assign 80% bandwidth to this group # /bin/echo 80 > very_imp_grp/cpu_quota # echo $apache_webserver_pid > very_imp_grp/tasks # mkdir less_imp_grp # Assign 5% bandwidth to this group # /bin/echo 5 > less_imp_grp/cpu_quota # echo $mozilla_browser_pid > less_imp_grp/tasks Signed-off-by : Srivatsa Vaddagiri include/linux/cpuset.h | 4 + kernel/cpuset.c | 180 ++++++++++++++++++++++++++++++++++++++++++++++++- kernel/sched.c | 7 + 3 files changed, 188 insertions(+), 3 deletions(-) diff -puN kernel/cpuset.c~cpuset_interface kernel/cpuset.c --- linux-2.6.18-rc3/kernel/cpuset.c~cpuset_interface 2006-08-04 08:36:55.000000000 +0530 +++ linux-2.6.18-rc3-root/kernel/cpuset.c 2006-08-04 09:15:52.000000000 +0530 @@ -99,6 +99,9 @@ struct cpuset { int mems_generation; struct fmeter fmeter; /* memory_pressure filter */ +#ifdef CONFIG_CPUMETER + void *cpu_ctlr_data; +#endif }; /* bits in struct cpuset flags field */ @@ -110,6 +113,9 @@ typedef enum { CS_NOTIFY_ON_RELEASE, CS_SPREAD_PAGE, CS_SPREAD_SLAB, +#ifdef CONFIG_CPUMETER + CS_METER_FLAG, +#endif } cpuset_flagbits_t; /* convenient tests for these bits */ @@ -148,6 +154,110 @@ static inline int is_spread_slab(const s return test_bit(CS_SPREAD_SLAB, &cs->flags); } +#ifdef CONFIG_CPUMETER +static inline int is_metered(const struct cpuset *cs) +{ + return test_bit(CS_METER_FLAG, &cs->flags); +} + +/* does the new combination of meter and exclusive flags makes sense? */ +static int validate_meters(const struct cpuset *cur, const struct cpuset *trial) +{ + struct cpuset *parent = cur->parent; + int is_changed; + + /* checks for change in meter flag */ + is_changed = (is_metered(trial) != is_metered(cur)); + + /* Cant change meter setting if a cpuset already has children */ + if (is_changed && !list_empty(&cur->children)) + return -EINVAL; + + /* Cant change meter setting if parent is metered */ + if (is_changed && parent && is_metered(parent)) + return -EINVAL; + + /* Turn on metering only if a cpuset is exclusive */ + if (is_metered(trial) && !is_cpu_exclusive(cur)) + return -EINVAL; + + /* Cant change exclusive setting if a cpuset is metered */ + if (!is_cpu_exclusive(trial) && is_metered(cur)) + return -EINVAL; + + /* Cant change exclusive setting if parent is metered */ + if (parent && is_metered(parent) && is_cpu_exclusive(trial)) + return -EINVAL; + + return 0; +} + +static void update_meter(struct cpuset *cs) +{ + if (is_metered(cs)) { + void *parent_data = NULL; + + if (cs->parent) + parent_data = cs->parent->cpu_ctlr_data; + cs->cpu_ctlr_data = cpu_ctlr->alloc_group(parent_data); + } else + cpu_ctlr->dealloc_group(cs->cpu_ctlr_data); +} + +static int update_quota(struct cpuset *cs, char *buf) +{ + int quota; + + if (!buf || !is_metered(cs)) + return -EINVAL; + + quota = simple_strtoul(buf, NULL, 10); + cpu_ctlr->assign_quota(cs->cpu_ctlr_data, quota); + + return 0; +} + +static int cpuset_sprintf_quota(char *page, struct cpuset *cs) +{ + int quota = 0; + char *c = page; + + if (is_metered(cs)) + quota = cpu_ctlr->get_quota(cs->cpu_ctlr_data); + c += sprintf (c, "%d", quota); + + return c - page; +} + +void *cpu_ctlr_data(struct cpuset *cs) +{ + return cs->cpu_ctlr_data; +} + +#else + +static inline int is_metered(const struct cpuset *cs) +{ + return 0; +} + +static int validate_meters(const struct cpuset *cur, const struct cpuset *trial) +{ + return 0; +} + +static void update_meter(struct cpuset *cs) +{ + return 0; +} + +static int update_quota(struct cpuset *cs, char *buf) +{ + return 0; +} + +#endif /* CONFIG_CPUMETER */ + /* * Increment this integer everytime any cpuset changes its * mems_allowed value. Users of cpusets can track this generation @@ -723,6 +833,9 @@ static int validate_change(const struct { struct cpuset *c, *par; + if (validate_meters(cur, trial)) + return -EINVAL; + /* Each of our child cpusets must be a subset of us */ list_for_each_entry(c, &cur->children, sibling) { if (!is_cpuset_subset(c, trial)) @@ -1037,7 +1150,7 @@ static int update_flag(cpuset_flagbits_t { int turning_on; struct cpuset trialcs; - int err, cpu_exclusive_changed; + int err, cpu_exclusive_changed, meter_changed; turning_on = (simple_strtoul(buf, NULL, 10) != 0); @@ -1052,6 +1165,7 @@ static int update_flag(cpuset_flagbits_t return err; cpu_exclusive_changed = (is_cpu_exclusive(cs) != is_cpu_exclusive(&trialcs)); + meter_changed = (is_metered(cs) != is_metered(&trialcs)); mutex_lock(&callback_mutex); if (turning_on) set_bit(bit, &cs->flags); @@ -1061,7 +1175,9 @@ static int update_flag(cpuset_flagbits_t if (cpu_exclusive_changed) update_cpu_domains(cs); - return 0; + if (meter_changed) + update_meter(cs); + return err; } /* @@ -1225,6 +1341,8 @@ static int attach_task(struct cpuset *cs return -ESRCH; } atomic_inc(&cs->count); + if (is_metered(cs)) + cpu_ctlr->move_task(tsk, oldcs->cpu_ctlr_data, cs->cpu_ctlr_data); rcu_assign_pointer(tsk->cpuset, cs); task_unlock(tsk); @@ -1267,6 +1385,10 @@ typedef enum { FILE_SPREAD_PAGE, FILE_SPREAD_SLAB, FILE_TASKLIST, +#ifdef CONFIG_CPUMETER + FILE_METER_FLAG, + FILE_METER_QUOTA, +#endif } cpuset_filetype_t; static ssize_t cpuset_common_file_write(struct file *file, const char __user *userbuf, @@ -1336,6 +1458,14 @@ static ssize_t cpuset_common_file_write( case FILE_TASKLIST: retval = attach_task(cs, buffer, &pathbuf); break; +#ifdef CONFIG_CPUMETER + case FILE_METER_FLAG: + retval = update_flag(CS_METER_FLAG, cs, buffer); + break; + case FILE_METER_QUOTA: + retval = update_quota(cs, buffer); + break; +#endif default: retval = -EINVAL; goto out2; @@ -1448,6 +1578,14 @@ static ssize_t cpuset_common_file_read(s case FILE_SPREAD_SLAB: *s++ = is_spread_slab(cs) ? '1' : '0'; break; +#ifdef CONFIG_CPUMETER + case FILE_METER_FLAG: + *s++ = is_metered(cs) ? '1' : '0'; + break; + case FILE_METER_QUOTA: + s += cpuset_sprintf_quota(s, cs); + break; +#endif default: retval = -EINVAL; goto out; @@ -1821,6 +1959,18 @@ static struct cftype cft_spread_slab = { .private = FILE_SPREAD_SLAB, }; +#ifdef CONFIG_CPUMETER +static struct cftype cft_meter_flag = { + .name = "meter_cpu", + .private = FILE_METER_FLAG, +}; + +static struct cftype cft_meter_quota = { + .name = "cpu_quota", + .private = FILE_METER_QUOTA, +}; +#endif + static int cpuset_populate_dir(struct dentry *cs_dentry) { int err; @@ -1845,6 +1995,12 @@ static int cpuset_populate_dir(struct de return err; if ((err = cpuset_add_file(cs_dentry, &cft_tasks)) < 0) return err; +#ifdef CONFIG_CPUMETER + if ((err = cpuset_add_file(cs_dentry, &cft_meter_flag)) < 0) + return err; + if ((err = cpuset_add_file(cs_dentry, &cft_meter_quota)) < 0) + return err; +#endif return 0; } @@ -1862,6 +2018,10 @@ static long cpuset_create(struct cpuset struct cpuset *cs; int err; + /* metered cpusets cant have grand-children! */ + if (parent->parent && is_metered(parent->parent)) + return -EINVAL; + cs = kmalloc(sizeof(*cs), GFP_KERNEL); if (!cs) return -ENOMEM; @@ -1894,6 +2054,13 @@ static long cpuset_create(struct cpuset if (err < 0) goto err; + if (is_metered(parent)) { + cs->cpus_allowed = parent->cpus_allowed; + cs->mems_allowed = parent->mems_allowed; + set_bit(CS_METER_FLAG, &cs->flags); + update_meter(cs); + } + /* * Release manage_mutex before cpuset_populate_dir() because it * will down() this new directory's i_mutex and if we race with @@ -1956,6 +2123,13 @@ static int cpuset_rmdir(struct inode *un return retval; } } + if (is_metered(cs)) { + int retval = update_flag(CS_METER_FLAG, cs, "0"); + if (retval < 0) { + mutex_unlock(&manage_mutex); + return retval; + } + } parent = cs->parent; mutex_lock(&callback_mutex); set_bit(CS_REMOVED, &cs->flags); @@ -2008,6 +2182,7 @@ int __init cpuset_init(void) top_cpuset.mems_generation = cpuset_mems_generation++; init_task.cpuset = &top_cpuset; + /* xxx: initialize init_task.cpu_ctlr_data */ err = register_filesystem(&cpuset_fs_type); if (err < 0) @@ -2135,6 +2310,7 @@ void cpuset_exit(struct task_struct *tsk struct cpuset *cs; cs = tsk->cpuset; + cpu_ctlr->move_task(tsk, cs->cpu_ctlr_data, top_cpuset.cpu_ctlr_data); tsk->cpuset = &top_cpuset; /* the_top_cpuset_hack - see above */ if (notify_on_release(cs)) { diff -puN include/linux/cpuset.h~cpuset_interface include/linux/cpuset.h --- linux-2.6.18-rc3/include/linux/cpuset.h~cpuset_interface 2006-08-04 08:36:55.000000000 +0530 +++ linux-2.6.18-rc3-root/include/linux/cpuset.h 2006-08-04 08:36:55.000000000 +0530 @@ -52,6 +52,10 @@ extern void cpuset_lock(void); extern void cpuset_unlock(void); extern int cpuset_mem_spread_node(void); +#ifdef CONFIG_CPUMETER +extern void *cpu_ctlr_data(struct cpuset *cs); +#endif + static inline int cpuset_do_page_mem_spread(void) { diff -puN kernel/sched.c~cpuset_interface kernel/sched.c --- linux-2.6.18-rc3/kernel/sched.c~cpuset_interface 2006-08-04 09:01:54.000000000 +0530 +++ linux-2.6.18-rc3-root/kernel/sched.c 2006-08-04 09:03:39.000000000 +0530 @@ -735,7 +735,12 @@ static void enqueue_task_grp(struct task /* return the task-group to which a task belongs */ static inline struct task_grp *task_grp(struct task_struct *p) { - return &init_task_grp; + struct task_grp *tg = (struct task_grp *)cpu_ctlr_data(p->cpuset); + + if (!tg) + tg = &init_task_grp; + + return tg; } static inline void update_task_grp_prio(struct task_grp_rq *tgrq, struct rq *rq, _ -- Regards, vatsa -- Regards, vatsa - To unsubscribe from this list: send the line "unsubscribe linux-kernel" in the body of a message to majordomo@vger.kernel.org More majordomo info at http://vger.kernel.org/majordomo-info.html Please read the FAQ at http://www.tux.org/lkml/