43 lines
1.0 KiB
Rust
43 lines
1.0 KiB
Rust
mod sealed {
|
|
pub trait Sealed {}
|
|
}
|
|
use sealed::Sealed;
|
|
|
|
/// A type representing a vector lane count.
|
|
pub struct LaneCount<const LANES: usize>;
|
|
|
|
impl<const LANES: usize> LaneCount<LANES> {
|
|
/// The number of bytes in a bitmask with this many lanes.
|
|
pub const BITMASK_LEN: usize = (LANES + 7) / 8;
|
|
}
|
|
|
|
/// Helper trait for vector lane counts.
|
|
pub trait SupportedLaneCount: Sealed {
|
|
#[doc(hidden)]
|
|
type BitMask: Copy + Default + AsRef<[u8]> + AsMut<[u8]>;
|
|
}
|
|
|
|
impl<const LANES: usize> Sealed for LaneCount<LANES> {}
|
|
|
|
impl SupportedLaneCount for LaneCount<1> {
|
|
type BitMask = [u8; 1];
|
|
}
|
|
impl SupportedLaneCount for LaneCount<2> {
|
|
type BitMask = [u8; 1];
|
|
}
|
|
impl SupportedLaneCount for LaneCount<4> {
|
|
type BitMask = [u8; 1];
|
|
}
|
|
impl SupportedLaneCount for LaneCount<8> {
|
|
type BitMask = [u8; 1];
|
|
}
|
|
impl SupportedLaneCount for LaneCount<16> {
|
|
type BitMask = [u8; 2];
|
|
}
|
|
impl SupportedLaneCount for LaneCount<32> {
|
|
type BitMask = [u8; 4];
|
|
}
|
|
impl SupportedLaneCount for LaneCount<64> {
|
|
type BitMask = [u8; 8];
|
|
}
|