tests/ui/simd/intrinsic/generic-bitmask-pass.rs

   1 // run-pass
   2 #![allow(non_camel_case_types)]
   3
   4 // ignore-emscripten
   5 // ignore-endian-big behavior of simd_bitmask is endian-specific
   6
   7 // Test that the simd_bitmask intrinsic produces correct results.
   8
   9 #![feature(repr_simd, platform_intrinsics)]
  10 #[allow(non_camel_case_types)]
  11
  12 #[repr(simd)]
  13 #[derive(Copy, Clone, PartialEq, Debug)]
  14 struct u32x4(pub u32, pub u32, pub u32, pub u32);
  15
  16 #[repr(simd)]
  17 #[derive(Copy, Clone, PartialEq, Debug)]
  18 struct u8x4(pub u8, pub u8, pub u8, pub u8);
  19
  20 #[repr(simd)]
  21 #[derive(Copy, Clone, PartialEq, Debug)]
  22 struct Tx4<T>(pub T, pub T, pub T, pub T);
  23
  24 extern "platform-intrinsic" {
  25     fn simd_bitmask<T, U>(x: T) -> U;
  26 }
  27
  28 fn main() {
  29     let z = u32x4(0, 0, 0, 0);
  30     let ez = 0_u8;
  31
  32     let o = u32x4(!0, !0, !0, !0);
  33     let eo = 0b_1111_u8;
  34
  35     let m0 = u32x4(!0, 0, !0, 0);
  36     let e0 = 0b_0000_0101_u8;
  37
  38     // Check that the MSB is extracted:
  39     let m = u8x4(0b_1000_0000, 0b_0100_0001, 0b_1100_0001, 0b_1111_1111);
  40     let e = 0b_1101;
  41
  42     // Check usize / isize
  43     let msize: Tx4<usize> = Tx4(usize::MAX, 0, usize::MAX, usize::MAX);
  44
  45     unsafe {
  46         let r: u8 = simd_bitmask(z);
  47         assert_eq!(r, ez);
  48
  49         let r: u8 = simd_bitmask(o);
  50         assert_eq!(r, eo);
  51
  52         let r: u8 = simd_bitmask(m0);
  53         assert_eq!(r, e0);
  54
  55         let r: u8 = simd_bitmask(m);
  56         assert_eq!(r, e);
  57
  58         let r: u8 = simd_bitmask(msize);
  59         assert_eq!(r, e);
  60
  61     }
  62 }