xref: /OK3568_Linux_fs/kernel/arch/powerpc/lib/alloc.c (revision 4882a59341e53eb6f0b4789bf948001014eff981) 
1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0
2*4882a593Smuzhiyun #include <linux/types.h>
3*4882a593Smuzhiyun #include <linux/init.h>
4*4882a593Smuzhiyun #include <linux/slab.h>
5*4882a593Smuzhiyun #include <linux/memblock.h>
6*4882a593Smuzhiyun #include <linux/string.h>
7*4882a593Smuzhiyun #include <asm/setup.h>
8*4882a593Smuzhiyun 
9*4882a593Smuzhiyun 
zalloc_maybe_bootmem(size_t size,gfp_t mask)10*4882a593Smuzhiyun void * __ref zalloc_maybe_bootmem(size_t size, gfp_t mask)
11*4882a593Smuzhiyun {
12*4882a593Smuzhiyun 	void *p;
13*4882a593Smuzhiyun 
14*4882a593Smuzhiyun 	if (slab_is_available())
15*4882a593Smuzhiyun 		p = kzalloc(size, mask);
16*4882a593Smuzhiyun 	else {
17*4882a593Smuzhiyun 		p = memblock_alloc(size, SMP_CACHE_BYTES);
18*4882a593Smuzhiyun 		if (!p)
19*4882a593Smuzhiyun 			panic("%s: Failed to allocate %zu bytes\n", __func__,
20*4882a593Smuzhiyun 			      size);
21*4882a593Smuzhiyun 	}
22*4882a593Smuzhiyun 	return p;
23*4882a593Smuzhiyun }
24