Commit f54a918d authored by Wang Wensheng's avatar Wang Wensheng
Browse files

mm/hugetlb: Introduce alloc_hugetlb_folio_size() 

ascend inclusion
category: Feature
bugzilla: https://gitee.com/openeuler/kernel/issues/I8SC4J



---------------------------------------------

This could allocate hugepage from hpool of specific size.
The driver would map those hugepage for some special processes via
hugetlb_insert_hugepage_pte().

Signed-off-by: default avatarWang Wensheng <wangwensheng4@huawei.com>
parent 5adc2032

include/linux/hugetlb.h

+14 −0
Original line number Diff line number Diff line
@@ -302,6 +302,15 @@ static inline int hugetlb_insert_hugepage_pte_by_pa(struct mm_struct *mm, 
}

#endif /* CONFIG_HUGETLB_INSERT_PAGE */



#ifdef CONFIG_ASCEND_FEATURES

struct folio *alloc_hugetlb_folio_size(int nid, unsigned long size);

#else

static inline struct folio *alloc_hugetlb_folio_size(int nid, unsigned long size)

{

	return NULL;

}

#endif



#else /* !CONFIG_HUGETLB_PAGE */



static inline void hugetlb_dup_vma_private(struct vm_area_struct *vma)
@@ -523,6 +532,11 @@ static inline int hugetlb_insert_hugepage_pte_by_pa(struct mm_struct *mm, 
	return -EPERM;

}



static inline struct folio *alloc_hugetlb_folio_size(int nid, unsigned long size)

{

	return NULL;

}



#endif /* !CONFIG_HUGETLB_PAGE */

/*

 * hugepages at page global directory. If arch support

mm/hugetlb.c

+27 −0
Original line number Diff line number Diff line
@@ -7633,3 +7633,30 @@ int hugetlb_insert_hugepage_pte_by_pa(struct mm_struct *mm, unsigned long addr, 
}

EXPORT_SYMBOL_GPL(hugetlb_insert_hugepage_pte_by_pa);

#endif /* CONFIG_HUGETLB_INSERT_PAGE */



#ifdef CONFIG_ASCEND_FEATURES

struct folio *alloc_hugetlb_folio_size(int nid, unsigned long size)

{

	gfp_t gfp_mask;

	struct hstate *h;

	nodemask_t nodemask;

	unsigned long flags;

	struct folio *folio = NULL;



	nodes_clear(nodemask);

	node_set(nid, nodemask);



	h = size_to_hstate(size);

	if (!h)

		return NULL;



	gfp_mask = htlb_alloc_mask(h);

	spin_lock_irqsave(&hugetlb_lock, flags);

	if (h->free_huge_pages - h->resv_huge_pages > 0)

		folio = dequeue_hugetlb_folio_nodemask(h, gfp_mask, nid, &nodemask);

	spin_unlock_irqrestore(&hugetlb_lock, flags);



	return folio;

}

EXPORT_SYMBOL(alloc_hugetlb_folio_size);

#endif