src/test/ui/simd/simd-bitmask.rs

   1 //run-pass
   2 //ignore-endian-big behavior of simd_select_bitmask is endian-specific
   3 #![feature(repr_simd, platform_intrinsics)]
   4
   5 extern "platform-intrinsic" {
   6     fn simd_bitmask<T, U>(v: T) -> U;
   7     fn simd_select_bitmask<T, U>(m: T, a: U, b: U) -> U;
   8 }
   9
  10 #[derive(Copy, Clone)]
  11 #[repr(simd)]
  12 struct Simd<T, const N: usize>([T; N]);
  13
  14 fn main() {
  15     unsafe {
  16         let v = Simd::<i8, 4>([-1, 0, -1, 0]);
  17         let i: u8 = simd_bitmask(v);
  18         let a: [u8; 1] = simd_bitmask(v);
  19
  20         assert_eq!(i, 0b0101);
  21         assert_eq!(a, [0b0101]);
  22
  23         let v = Simd::<i8, 16>([0, 0, -1, -1, 0, 0, 0, 0, 0, 0, 0, 0, -1, 0, -1, 0]);
  24         let i: u16 = simd_bitmask(v);
  25         let a: [u8; 2] = simd_bitmask(v);
  26
  27         assert_eq!(i, 0b0101000000001100);
  28         assert_eq!(a, [0b1100, 0b01010000]);
  29     }
  30
  31     unsafe {
  32         let a = Simd::<i32, 8>([0, 1, 2, 3, 4, 5, 6, 7]);
  33         let b = Simd::<i32, 8>([8, 9, 10, 11, 12, 13, 14, 15]);
  34         let e = [0, 9, 2, 11, 12, 13, 14, 15];
  35
  36         let r = simd_select_bitmask(0b0101u8, a, b);
  37         assert_eq!(r.0, e);
  38
  39         let r = simd_select_bitmask([0b0101u8], a, b);
  40         assert_eq!(r.0, e);
  41
  42         let a = Simd::<i32, 16>([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]);
  43         let b = Simd::<i32, 16>([16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31]);
  44         let e = [16, 17, 2, 3, 20, 21, 22, 23, 24, 25, 26, 27, 12, 29, 14, 31];
  45
  46         let r = simd_select_bitmask(0b0101000000001100u16, a, b);
  47         assert_eq!(r.0, e);
  48
  49         let r = simd_select_bitmask([0b1100u8, 0b01010000u8], a, b);
  50         assert_eq!(r.0, e);
  51     }
  52 }