example/std_example.rs

   1 #![feature(core_intrinsics)]
   2
   3 use std::arch::x86_64::*;
   4 use std::io::Write;
   5 use std::intrinsics;
   6
   7
   8 fn main() {
   9     let mutex = std::sync::Mutex::new(());
  10     mutex.lock().unwrap();
  11
  12     let _ = ::std::iter::repeat('a' as u8).take(10).collect::<Vec<_>>();
  13     let stderr = ::std::io::stderr();
  14     let mut stderr = stderr.lock();
  15
  16     writeln!(stderr, "some {} text", "<unknown>").unwrap();
  17
  18     let _ = std::process::Command::new("true").env("c", "d").spawn();
  19
  20     println!("cargo:rustc-link-lib=z");
  21
  22     static ONCE: std::sync::Once = std::sync::ONCE_INIT;
  23     ONCE.call_once(|| {});
  24
  25     LoopState::Continue(()) == LoopState::Break(());
  26
  27     // Make sure ByValPair values with differently sized components are correctly passed
  28     map(None::<(u8, Box<Instruction>)>);
  29
  30     println!("{}", 2.3f32.exp());
  31     println!("{}", 2.3f32.exp2());
  32     println!("{}", 2.3f32.abs());
  33     println!("{}", 2.3f32.sqrt());
  34     println!("{}", 2.3f32.floor());
  35     println!("{}", 2.3f32.ceil());
  36     println!("{}", 2.3f32.min(1.0));
  37     println!("{}", 2.3f32.max(1.0));
  38     println!("{}", 2.3f32.powi(2));
  39
  40     assert_eq!(0b0000000000000000000000000010000010000000000000000000000000000000_0000000000100000000000000000000000001000000000000100000000000000u128.leading_zeros(), 26);
  41     assert_eq!(0b0000000000000000000000000010000000000000000000000000000000000000_0000000000000000000000000000000000001000000000000000000010000000u128.trailing_zeros(), 7);
  42
  43     0i128.checked_div(2i128);
  44     0u128.checked_div(2u128);
  45     assert_eq!(1u128 + 2, 3);
  46
  47     assert_eq!(0b100010000000000000000000000000000u128 >> 10, 0b10001000000000000000000u128);
  48     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 >> 64, 0xFEDCBA98765432u128);
  49     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 as i128 >> 64, 0xFEDCBA98765432i128);
  50     assert_eq!(353985398u128 * 932490u128, 330087843781020u128);
  51
  52     unsafe {
  53         test_simd();
  54     }
  55 }
  56
  57 #[target_feature(enable = "sse2")]
  58 unsafe fn test_simd() {
  59     let x = _mm_setzero_si128();
  60     let y = _mm_set1_epi16(7);
  61     let or = _mm_or_si128(x, y);
  62     let cmp_eq = _mm_cmpeq_epi8(y, y);
  63     let cmp_lt = _mm_cmplt_epi8(y, y);
  64
  65     assert_eq!(std::mem::transmute::<_, [u16; 8]>(or), [7, 7, 7, 7, 7, 7, 7, 7]);
  66     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_eq), [0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff]);
  67     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_lt), [0, 0, 0, 0, 0, 0, 0, 0]);
  68
  69     test_mm_slli_si128();
  70     test_mm_movemask_epi8();
  71     test_mm256_movemask_epi8();
  72     test_mm_add_epi8();
  73     test_mm_add_pd();
  74     test_mm_cvtepi8_epi16();
  75
  76     let mask1 = _mm_movemask_epi8(dbg!(_mm_setr_epi8(255u8 as i8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0)));
  77     assert_eq!(mask1, 1);
  78 }
  79
  80 #[target_feature(enable = "sse2")]
  81 unsafe fn test_mm_slli_si128() {
  82     #[rustfmt::skip]
  83     let a = _mm_setr_epi8(
  84         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
  85     );
  86     let r = _mm_slli_si128(a, 1);
  87     let e = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
  88     assert_eq_m128i(r, e);
  89
  90     #[rustfmt::skip]
  91     let a = _mm_setr_epi8(
  92         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
  93     );
  94     let r = _mm_slli_si128(a, 15);
  95     let e = _mm_setr_epi8(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1);
  96     assert_eq_m128i(r, e);
  97
  98     #[rustfmt::skip]
  99     let a = _mm_setr_epi8(
 100         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 101     );
 102     let r = _mm_slli_si128(a, 16);
 103     assert_eq_m128i(r, _mm_set1_epi8(0));
 104
 105     #[rustfmt::skip]
 106     let a = _mm_setr_epi8(
 107         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 108     );
 109     let r = _mm_slli_si128(a, -1);
 110     assert_eq_m128i(_mm_set1_epi8(0), r);
 111
 112     #[rustfmt::skip]
 113     let a = _mm_setr_epi8(
 114         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 115     );
 116     let r = _mm_slli_si128(a, -0x80000000);
 117     assert_eq_m128i(r, _mm_set1_epi8(0));
 118 }
 119
 120 #[target_feature(enable = "sse2")]
 121 unsafe fn test_mm_movemask_epi8() {
 122     #[rustfmt::skip]
 123     let a = _mm_setr_epi8(
 124         0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8, 0b01,
 125         0b0101, 0b1111_0000u8 as i8, 0, 0,
 126         0, 0, 0b1111_0000u8 as i8, 0b0101,
 127         0b01, 0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8,
 128     );
 129     let r = _mm_movemask_epi8(a);
 130     assert_eq!(r, 0b10100100_00100101);
 131 }
 132
 133 #[target_feature(enable = "avx2")]
 134 unsafe fn test_mm256_movemask_epi8() {
 135     let a = _mm256_set1_epi8(-1);
 136     let r = _mm256_movemask_epi8(a);
 137     let e = -1;
 138     assert_eq!(r, e);
 139 }
 140
 141 #[target_feature(enable = "sse2")]
 142 unsafe fn test_mm_add_epi8() {
 143     let a = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
 144     #[rustfmt::skip]
 145     let b = _mm_setr_epi8(
 146         16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31,
 147     );
 148     let r = _mm_add_epi8(a, b);
 149     #[rustfmt::skip]
 150     let e = _mm_setr_epi8(
 151         16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46,
 152     );
 153     assert_eq_m128i(r, e);
 154 }
 155
 156 #[target_feature(enable = "sse2")]
 157 unsafe fn test_mm_add_pd() {
 158     let a = _mm_setr_pd(1.0, 2.0);
 159     let b = _mm_setr_pd(5.0, 10.0);
 160     let r = _mm_add_pd(a, b);
 161     assert_eq_m128d(r, _mm_setr_pd(6.0, 12.0));
 162 }
 163
 164 fn assert_eq_m128i(x: std::arch::x86_64::__m128i, y: std::arch::x86_64::__m128i) {
 165     unsafe {
 166         assert_eq!(std::mem::transmute::<_, [u8; 16]>(x), std::mem::transmute::<_, [u8; 16]>(x));
 167     }
 168 }
 169
 170 #[target_feature(enable = "sse2")]
 171 pub unsafe fn assert_eq_m128d(a: __m128d, b: __m128d) {
 172     if _mm_movemask_pd(_mm_cmpeq_pd(a, b)) != 0b11 {
 173         panic!("{:?} != {:?}", a, b);
 174     }
 175 }
 176
 177 #[target_feature(enable = "sse4.1")]
 178 unsafe fn test_mm_cvtepi8_epi16() {
 179     let a = _mm_set1_epi8(10);
 180     let r = _mm_cvtepi8_epi16(a);
 181     let e = _mm_set1_epi16(10);
 182     assert_eq_m128i(r, e);
 183     let a = _mm_set1_epi8(-10);
 184     let r = _mm_cvtepi8_epi16(a);
 185     let e = _mm_set1_epi16(-10);
 186     assert_eq_m128i(r, e);
 187 }
 188
 189 #[derive(PartialEq)]
 190 enum LoopState {
 191     Continue(()),
 192     Break(())
 193 }
 194
 195 pub enum Instruction {
 196     Increment,
 197     Loop,
 198 }
 199
 200 fn map(a: Option<(u8, Box<Instruction>)>) -> Option<Box<Instruction>> {
 201     match a {
 202         None => None,
 203         Some((_, instr)) => Some(instr),
 204     }
 205 }