crates/core_simd/src/lane_count.rs

   1 mod sealed {
   2     pub trait Sealed {}
   3 }
   4 use sealed::Sealed;
   5
   6 /// A type representing a vector lane count.
   7 pub struct LaneCount<const LANES: usize>;
   8
   9 impl<const LANES: usize> LaneCount<LANES> {
  10     /// The number of bytes in a bitmask with this many lanes.
  11     pub const BITMASK_LEN: usize = (LANES + 7) / 8;
  12 }
  13
  14 /// Helper trait for vector lane counts.
  15 pub trait SupportedLaneCount: Sealed {
  16     #[doc(hidden)]
  17     type BitMask: Copy + Default + AsRef<[u8]> + AsMut<[u8]>;
  18 }
  19
  20 impl<const LANES: usize> Sealed for LaneCount<LANES> {}
  21
  22 impl SupportedLaneCount for LaneCount<1> {
  23     type BitMask = [u8; 1];
  24 }
  25 impl SupportedLaneCount for LaneCount<2> {
  26     type BitMask = [u8; 1];
  27 }
  28 impl SupportedLaneCount for LaneCount<4> {
  29     type BitMask = [u8; 1];
  30 }
  31 impl SupportedLaneCount for LaneCount<8> {
  32     type BitMask = [u8; 1];
  33 }
  34 impl SupportedLaneCount for LaneCount<16> {
  35     type BitMask = [u8; 2];
  36 }
  37 impl SupportedLaneCount for LaneCount<32> {
  38     type BitMask = [u8; 4];
  39 }
  40 impl SupportedLaneCount for LaneCount<64> {
  41     type BitMask = [u8; 8];
  42 }