Commit | Line | Data |
---|---|---|
7b2c3c5b SR |
1 | #include <linux/types.h> |
2 | #include <linux/init.h> | |
3 | #include <linux/slab.h> | |
4 | #include <linux/bootmem.h> | |
5669c3cf | 5 | #include <linux/string.h> |
ae3a197e | 6 | #include <asm/setup.h> |
7b2c3c5b | 7 | |
7b2c3c5b | 8 | |
5669c3cf SR |
9 | void * __init_refok zalloc_maybe_bootmem(size_t size, gfp_t mask) |
10 | { | |
11 | void *p; | |
12 | ||
13 | if (mem_init_done) | |
14 | p = kzalloc(size, mask); | |
15 | else { | |
e39f223f | 16 | p = memblock_virt_alloc(size, 0); |
5669c3cf SR |
17 | } |
18 | return p; | |
19 | } |