- part_count = int(part_count)
- assert part_count != 0 and (part_count & (part_count - 1)) == 0,\
- "part_counts values must all be powers of two"
- return part_count
-
- self.part_counts = SimdMap.map_with_elwid(check, part_counts)
- self.full_part_count = max(part_counts.values())
- assert self.simd_full_width_hint % self.full_part_count == 0,\
- "simd_full_width_hint must be a multiple of full_part_count"
+ vec_el_count = int(vec_el_count)
+ assert vec_el_count != 0 \
+ and (vec_el_count & (vec_el_count - 1)) == 0,\
+ "vec_el_counts values must all be powers of two"
+ return vec_el_count
+
+ self.vec_el_counts = SimdMap.map_with_elwid(check, vec_el_counts)
+ self.full_el_count = max(self.vec_el_counts.values())
+ assert self.simd_full_width_hint % self.full_el_count == 0,\
+ "simd_full_width_hint must be a multiple of full_el_count"