Commit 7529cc7f authored by Tariq Toukan's avatar Tariq Toukan Committed by Saeed Mahameed
Browse files

lib: bitmap: Introduce node-aware alloc API 



Expose new node-aware API for bitmap allocation:
bitmap_alloc_node() / bitmap_zalloc_node().

Signed-off-by: default avatarTariq Toukan <tariqt@nvidia.com>
Reviewed-by: default avatarMoshe Shemesh <moshe@nvidia.com>
Signed-off-by: default avatarSaeed Mahameed <saeedm@nvidia.com>
parent 06338cef

include/linux/bitmap.h

+2 −0
Original line number Diff line number Diff line
@@ -123,6 +123,8 @@ struct device; 
 */

unsigned long *bitmap_alloc(unsigned int nbits, gfp_t flags);

unsigned long *bitmap_zalloc(unsigned int nbits, gfp_t flags);

unsigned long *bitmap_alloc_node(unsigned int nbits, gfp_t flags, int node);

unsigned long *bitmap_zalloc_node(unsigned int nbits, gfp_t flags, int node);

void bitmap_free(const unsigned long *bitmap);



/* Managed variants of the above. */

lib/bitmap.c

+13 −0
Original line number Diff line number Diff line
@@ -1398,6 +1398,19 @@ unsigned long *bitmap_zalloc(unsigned int nbits, gfp_t flags) 
}

EXPORT_SYMBOL(bitmap_zalloc);



unsigned long *bitmap_alloc_node(unsigned int nbits, gfp_t flags, int node)

{

	return kmalloc_array_node(BITS_TO_LONGS(nbits), sizeof(unsigned long),

				  flags, node);

}

EXPORT_SYMBOL(bitmap_alloc_node);



unsigned long *bitmap_zalloc_node(unsigned int nbits, gfp_t flags, int node)

{

	return bitmap_alloc_node(nbits, flags | __GFP_ZERO, node);

}

EXPORT_SYMBOL(bitmap_zalloc_node);



void bitmap_free(const unsigned long *bitmap)

{

	kfree(bitmap);