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

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