memblock: make memblock_set_node() support different memblock_type