2431 						      int nr_units)  in pcpu_alloc_alloc_info()  argument
2440 	ai_size = base_size + nr_units * sizeof(ai->groups[0].cpu_map[0]);  in pcpu_alloc_alloc_info()
2450 	for (unit = 0; unit < nr_units; unit++)  in pcpu_alloc_alloc_info()
2507 		BUG_ON(gi->nr_units % upa);  in pcpu_dump_alloc_info()
2508 		for (alloc_end += gi->nr_units / upa;  in pcpu_dump_alloc_info()
2658 		group_sizes[group] = gi->nr_units * ai->unit_size;  in pcpu_setup_first_chunk()
2660 		for (i = 0; i < gi->nr_units; i++) {  in pcpu_setup_first_chunk()
2846 	int nr_groups = 1, nr_units = 0;  in pcpu_build_alloc_info()  local
2939 		nr_units += roundup(group_cnt[group], upa);  in pcpu_build_alloc_info()
2941 	ai = pcpu_alloc_alloc_info(nr_groups, nr_units);  in pcpu_build_alloc_info()
2970 				gi->cpu_map[gi->nr_units++] = cpu;  in pcpu_build_alloc_info()
2971 		gi->nr_units = roundup(gi->nr_units, upa);  in pcpu_build_alloc_info()
2972 		unit += gi->nr_units;  in pcpu_build_alloc_info()
2974 	BUG_ON(unit != nr_units);  in pcpu_build_alloc_info()
3081 		for (i = 0; i < gi->nr_units && cpu == NR_CPUS; i++)  in pcpu_embed_first_chunk()
3086 		ptr = pcpu_fc_alloc(cpu, gi->nr_units * ai->unit_size, atom_size, cpu_to_nd_fn);  in pcpu_embed_first_chunk()
3100 	max_distance += ai->unit_size * ai->groups[highest_group].nr_units;  in pcpu_embed_first_chunk()
3122 		for (i = 0; i < gi->nr_units; i++, ptr += ai->unit_size) {  in pcpu_embed_first_chunk()
3150 				ai->groups[group].nr_units * ai->unit_size);  in pcpu_embed_first_chunk()
3257 	if (WARN_ON(ai->groups[0].nr_units != nr_g0_units)) {  in pcpu_page_first_chunk()
3405 	ai->groups[0].nr_units = 1;  in setup_per_cpu_areas()