lkml.org 
[lkml]   [2017]   [Jun]   [14]   [last100]   RSS Feed
Views: [wrap][no wrap]   [headers]  [forward] 
 
Messages in this thread
Patch in this message
/
Date
From
SubjectRe: [RFC PATCH 2/4] hugetlb: add support for preferred node to alloc_huge_page_nodemask
On Wed 14-06-17 18:17:18, Vlastimil Babka wrote:
> On 06/13/2017 11:00 AM, Michal Hocko wrote:
> > From: Michal Hocko <mhocko@suse.com>
> >
> > alloc_huge_page_nodemask tries to allocate from any numa node in the
> > allowed node mask starting from lower numa nodes. This might lead to
> > filling up those low NUMA nodes while others are not used. We can reduce
> > this risk by introducing a concept of the preferred node similar to what
> > we have in the regular page allocator. We will start allocating from the
> > preferred nid and then iterate over all allowed nodes in the zonelist
> > order until we try them all.
> >
> > This is mimicking the page allocator logic except it operates on
> > per-node mempools. dequeue_huge_page_vma already does this so distill
> > the zonelist logic into a more generic dequeue_huge_page_nodemask
> > and use it in alloc_huge_page_nodemask.
> >
> > Signed-off-by: Michal Hocko <mhocko@suse.com>
>
> I've reviewed the current version in git, where patch 3/4 is folded.
>
> Noticed some things below, but after fixing:
> Acked-by: Vlastimil Babka <vbabka@suse.cz>

Thanks!

[...]
> > +retry_cpuset:
> > + cpuset_mems_cookie = read_mems_allowed_begin();
> > + for_each_zone_zonelist_nodemask(zone, z, zonelist, gfp_zone(gfp_mask), nmask) {
> > + if (!cpuset_zone_allowed(zone, gfp_mask))
> > + continue;
> > + /*
> > + * no need to ask again on the same node. Pool is node rather than
> > + * zone aware
> > + */
> > + if (zone_to_nid(zone) == node)
> > + continue;
> > + node = zone_to_nid(zone);
> >
> > - for_each_online_node(node) {
> > page = dequeue_huge_page_node_exact(h, node);
> > if (page)
> > - return page;
> > + break;
>
> Either keep return page here...
>
> > }
> > + if (unlikely(!page && read_mems_allowed_retry(cpuset_mems_cookie)))
> > + goto retry_cpuset;
> > +
> > return NULL;
>
> ... or return page here.

ups I went with the former.

[...]

> > -struct page *alloc_huge_page_nodemask(struct hstate *h, nodemask_t *nmask)
> > +
> > +struct page *alloc_huge_page_nodemask(struct hstate *h, int preferred_nid,
> > + nodemask_t *nmask)
> > {
> > struct page *page = NULL;
> > - int node;
> >
> > spin_lock(&hugetlb_lock);
> > if (h->free_huge_pages - h->resv_huge_pages > 0) {
> > - for_each_node_mask(node, *nmask) {
> > - page = dequeue_huge_page_node_exact(h, node);
> > - if (page)
> > - break;
> > - }
> > + page = dequeue_huge_page_nodemask(h, preferred_nid, nmask);
>
>
>
> > + if (page)
> > + goto unlock;
> > }
> > +unlock:
>
> This doesn't seem needed?

This on top?
---
diff --git a/mm/hugetlb.c b/mm/hugetlb.c
index 9ac0ae725c5e..f9868e095afa 100644
--- a/mm/hugetlb.c
+++ b/mm/hugetlb.c
@@ -902,7 +902,6 @@ static struct page *dequeue_huge_page_nodemask(struct hstate *h, gfp_t gfp_mask,
{
unsigned int cpuset_mems_cookie;
struct zonelist *zonelist;
- struct page *page = NULL;
struct zone *zone;
struct zoneref *z;
int node = -1;
@@ -912,6 +911,8 @@ static struct page *dequeue_huge_page_nodemask(struct hstate *h, gfp_t gfp_mask,
retry_cpuset:
cpuset_mems_cookie = read_mems_allowed_begin();
for_each_zone_zonelist_nodemask(zone, z, zonelist, gfp_zone(gfp_mask), nmask) {
+ struct page *page;
+
if (!cpuset_zone_allowed(zone, gfp_mask))
continue;
/*
@@ -924,9 +925,9 @@ static struct page *dequeue_huge_page_nodemask(struct hstate *h, gfp_t gfp_mask,

page = dequeue_huge_page_node_exact(h, node);
if (page)
- break;
+ return page;
}
- if (unlikely(!page && read_mems_allowed_retry(cpuset_mems_cookie)))
+ if (unlikely(read_mems_allowed_retry(cpuset_mems_cookie)))
goto retry_cpuset;

return NULL;
@@ -1655,18 +1656,18 @@ struct page *alloc_huge_page_nodemask(struct hstate *h, int preferred_nid,
nodemask_t *nmask)
{
gfp_t gfp_mask = htlb_alloc_mask(h);
- struct page *page = NULL;

spin_lock(&hugetlb_lock);
if (h->free_huge_pages - h->resv_huge_pages > 0) {
+ struct page *page;
+
page = dequeue_huge_page_nodemask(h, gfp_mask, preferred_nid, nmask);
- if (page)
- goto unlock;
+ if (page) {
+ spin_unlock(&hugetlb_lock);
+ return page;
+ }
}
-unlock:
spin_unlock(&hugetlb_lock);
- if (page)
- return page;

/* No reservations, try to overcommit */

--
Michal Hocko
SUSE Labs
\
 
 \ /
  Last update: 2017-06-14 18:43    [W:0.076 / U:0.520 seconds]
©2003-2020 Jasper Spaans|hosted at Digital Ocean and TransIP|Read the blog|Advertise on this site