| Stephen Rothwell | 7b2c3c5 | 2007-09-17 14:08:06 +1000 | [diff] [blame] | 1 | #include <linux/types.h> |
| 2 | #include <linux/init.h> | ||||
| 3 | #include <linux/slab.h> | ||||
| 4 | #include <linux/bootmem.h> | ||||
| 5 | |||||
| 6 | #include <asm/system.h> | ||||
| 7 | |||||
| 8 | void * __init_refok alloc_maybe_bootmem(size_t size, gfp_t mask) | ||||
| 9 | { | ||||
| 10 | if (mem_init_done) | ||||
| 11 | return kmalloc(size, mask); | ||||
| 12 | else | ||||
| 13 | return alloc_bootmem(size); | ||||
| 14 | } | ||||