unsigned int riscv_cboz_block_size;
 EXPORT_SYMBOL_GPL(riscv_cboz_block_size);
 
-static void cbo_get_block_size(struct device_node *node,
-                              const char *name, u32 *block_size,
-                              unsigned long *first_hartid)
+static void __init cbo_get_block_size(struct device_node *node,
+                                     const char *name, u32 *block_size,
+                                     unsigned long *first_hartid)
 {
        unsigned long hartid;
        u32 val;
        }
 }
 
-void riscv_init_cbo_blocksizes(void)
+void __init riscv_init_cbo_blocksizes(void)
 {
        unsigned long cbom_hartid, cboz_hartid;
        u32 cbom_block_size = 0, cboz_block_size = 0;