example/std_example.rs

   1 #![feature(core_intrinsics)]
   2 #![feature(is_sorted)]
   3
   4 use std::arch::x86_64::*;
   5 use std::io::Write;
   6
   7 fn main() {
   8     let mutex = std::sync::Mutex::new(());
   9     let _guard = mutex.lock().unwrap();
  10
  11     let _ = ::std::iter::repeat('a' as u8).take(10).collect::<Vec<_>>();
  12     let stderr = ::std::io::stderr();
  13     let mut stderr = stderr.lock();
  14
  15     std::thread::spawn(move || {
  16         println!("Hello from another thread!");
  17     });
  18
  19     writeln!(stderr, "some {} text", "<unknown>").unwrap();
  20
  21     let _ = std::process::Command::new("true").env("c", "d").spawn();
  22
  23     println!("cargo:rustc-link-lib=z");
  24
  25     static ONCE: std::sync::Once = std::sync::Once::new();
  26     ONCE.call_once(|| {});
  27
  28     let _eq = LoopState::Continue(()) == LoopState::Break(());
  29
  30     // Make sure ByValPair values with differently sized components are correctly passed
  31     map(None::<(u8, Box<Instruction>)>);
  32
  33     println!("{}", 2.3f32.exp());
  34     println!("{}", 2.3f32.exp2());
  35     println!("{}", 2.3f32.abs());
  36     println!("{}", 2.3f32.sqrt());
  37     println!("{}", 2.3f32.floor());
  38     println!("{}", 2.3f32.ceil());
  39     println!("{}", 2.3f32.min(1.0));
  40     println!("{}", 2.3f32.max(1.0));
  41     println!("{}", 2.3f32.powi(2));
  42     println!("{}", 2.3f32.log2());
  43     assert_eq!(2.3f32.copysign(-1.0), -2.3f32);
  44     println!("{}", 2.3f32.powf(2.0));
  45
  46     assert_eq!(-128i8, (-128i8).saturating_sub(1));
  47     assert_eq!(127i8, 127i8.saturating_sub(-128));
  48     assert_eq!(-128i8, (-128i8).saturating_add(-128));
  49     assert_eq!(127i8, 127i8.saturating_add(1));
  50
  51     assert_eq!(0b0000000000000000000000000010000010000000000000000000000000000000_0000000000100000000000000000000000001000000000000100000000000000u128.leading_zeros(), 26);
  52     assert_eq!(0b0000000000000000000000000010000000000000000000000000000000000000_0000000000000000000000000000000000001000000000000000000010000000u128.trailing_zeros(), 7);
  53
  54     let _d = 0i128.checked_div(2i128);
  55     let _d = 0u128.checked_div(2u128);
  56     assert_eq!(1u128 + 2, 3);
  57
  58     assert_eq!(0b100010000000000000000000000000000u128 >> 10, 0b10001000000000000000000u128);
  59     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 >> 64, 0xFEDCBA98765432u128);
  60     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 as i128 >> 64, 0xFEDCBA98765432i128);
  61
  62     let tmp = 353985398u128;
  63     assert_eq!(tmp * 932490u128, 330087843781020u128);
  64
  65     let tmp = -0x1234_5678_9ABC_DEF0i64;
  66     assert_eq!(tmp as i128, -0x1234_5678_9ABC_DEF0i128);
  67
  68     // Check that all u/i128 <-> float casts work correctly.
  69     let houndred_u128 = 100u128;
  70     let houndred_i128 = 100i128;
  71     let houndred_f32 = 100.0f32;
  72     let houndred_f64 = 100.0f64;
  73     assert_eq!(houndred_u128 as f32, 100.0);
  74     assert_eq!(houndred_u128 as f64, 100.0);
  75     assert_eq!(houndred_f32 as u128, 100);
  76     assert_eq!(houndred_f64 as u128, 100);
  77     assert_eq!(houndred_i128 as f32, 100.0);
  78     assert_eq!(houndred_i128 as f64, 100.0);
  79     assert_eq!(houndred_f32 as i128, 100);
  80     assert_eq!(houndred_f64 as i128, 100);
  81
  82     let _a = 1u32 << 2u8;
  83
  84     let empty: [i32; 0] = [];
  85     assert!(empty.is_sorted());
  86
  87     println!("{:?}", unsafe { std::intrinsics::caller_location() });
  88
  89     unsafe {
  90         test_simd();
  91     }
  92 }
  93
  94 #[target_feature(enable = "sse2")]
  95 unsafe fn test_simd() {
  96     let x = _mm_setzero_si128();
  97     let y = _mm_set1_epi16(7);
  98     let or = _mm_or_si128(x, y);
  99     let cmp_eq = _mm_cmpeq_epi8(y, y);
 100     let cmp_lt = _mm_cmplt_epi8(y, y);
 101
 102     assert_eq!(std::mem::transmute::<_, [u16; 8]>(or), [7, 7, 7, 7, 7, 7, 7, 7]);
 103     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_eq), [0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff]);
 104     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_lt), [0, 0, 0, 0, 0, 0, 0, 0]);
 105
 106     test_mm_slli_si128();
 107     test_mm_movemask_epi8();
 108     test_mm256_movemask_epi8();
 109     test_mm_add_epi8();
 110     test_mm_add_pd();
 111     test_mm_cvtepi8_epi16();
 112     test_mm_cvtsi128_si64();
 113
 114     // FIXME(#666) implement `#[rustc_arg_required_const(..)]` support
 115     //test_mm_extract_epi8();
 116
 117     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)));
 118     assert_eq!(mask1, 1);
 119 }
 120
 121 #[target_feature(enable = "sse2")]
 122 unsafe fn test_mm_slli_si128() {
 123     #[rustfmt::skip]
 124     let a = _mm_setr_epi8(
 125         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 126     );
 127     let r = _mm_slli_si128(a, 1);
 128     let e = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
 129     assert_eq_m128i(r, e);
 130
 131     #[rustfmt::skip]
 132     let a = _mm_setr_epi8(
 133         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 134     );
 135     let r = _mm_slli_si128(a, 15);
 136     let e = _mm_setr_epi8(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1);
 137     assert_eq_m128i(r, e);
 138
 139     #[rustfmt::skip]
 140     let a = _mm_setr_epi8(
 141         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 142     );
 143     let r = _mm_slli_si128(a, 16);
 144     assert_eq_m128i(r, _mm_set1_epi8(0));
 145
 146     #[rustfmt::skip]
 147     let a = _mm_setr_epi8(
 148         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 149     );
 150     let r = _mm_slli_si128(a, -1);
 151     assert_eq_m128i(_mm_set1_epi8(0), r);
 152
 153     #[rustfmt::skip]
 154     let a = _mm_setr_epi8(
 155         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 156     );
 157     let r = _mm_slli_si128(a, -0x80000000);
 158     assert_eq_m128i(r, _mm_set1_epi8(0));
 159 }
 160
 161 #[target_feature(enable = "sse2")]
 162 unsafe fn test_mm_movemask_epi8() {
 163     #[rustfmt::skip]
 164     let a = _mm_setr_epi8(
 165         0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8, 0b01,
 166         0b0101, 0b1111_0000u8 as i8, 0, 0,
 167         0, 0, 0b1111_0000u8 as i8, 0b0101,
 168         0b01, 0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8,
 169     );
 170     let r = _mm_movemask_epi8(a);
 171     assert_eq!(r, 0b10100100_00100101);
 172 }
 173
 174 #[target_feature(enable = "avx2")]
 175 unsafe fn test_mm256_movemask_epi8() {
 176     let a = _mm256_set1_epi8(-1);
 177     let r = _mm256_movemask_epi8(a);
 178     let e = -1;
 179     assert_eq!(r, e);
 180 }
 181
 182 #[target_feature(enable = "sse2")]
 183 unsafe fn test_mm_add_epi8() {
 184     let a = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
 185     #[rustfmt::skip]
 186     let b = _mm_setr_epi8(
 187         16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31,
 188     );
 189     let r = _mm_add_epi8(a, b);
 190     #[rustfmt::skip]
 191     let e = _mm_setr_epi8(
 192         16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46,
 193     );
 194     assert_eq_m128i(r, e);
 195 }
 196
 197 #[target_feature(enable = "sse2")]
 198 unsafe fn test_mm_add_pd() {
 199     let a = _mm_setr_pd(1.0, 2.0);
 200     let b = _mm_setr_pd(5.0, 10.0);
 201     let r = _mm_add_pd(a, b);
 202     assert_eq_m128d(r, _mm_setr_pd(6.0, 12.0));
 203 }
 204
 205 fn assert_eq_m128i(x: std::arch::x86_64::__m128i, y: std::arch::x86_64::__m128i) {
 206     unsafe {
 207         assert_eq!(std::mem::transmute::<_, [u8; 16]>(x), std::mem::transmute::<_, [u8; 16]>(y));
 208     }
 209 }
 210
 211 #[target_feature(enable = "sse2")]
 212 pub unsafe fn assert_eq_m128d(a: __m128d, b: __m128d) {
 213     if _mm_movemask_pd(_mm_cmpeq_pd(a, b)) != 0b11 {
 214         panic!("{:?} != {:?}", a, b);
 215     }
 216 }
 217
 218 #[target_feature(enable = "sse2")]
 219 unsafe fn test_mm_cvtsi128_si64() {
 220     let r = _mm_cvtsi128_si64(std::mem::transmute::<[i64; 2], _>([5, 0]));
 221     assert_eq!(r, 5);
 222 }
 223
 224 #[target_feature(enable = "sse4.1")]
 225 unsafe fn test_mm_cvtepi8_epi16() {
 226     let a = _mm_set1_epi8(10);
 227     let r = _mm_cvtepi8_epi16(a);
 228     let e = _mm_set1_epi16(10);
 229     assert_eq_m128i(r, e);
 230     let a = _mm_set1_epi8(-10);
 231     let r = _mm_cvtepi8_epi16(a);
 232     let e = _mm_set1_epi16(-10);
 233     assert_eq_m128i(r, e);
 234 }
 235
 236 #[target_feature(enable = "sse4.1")]
 237 unsafe fn test_mm_extract_epi8() {
 238     #[rustfmt::skip]
 239     let a = _mm_setr_epi8(
 240         -1, 1, 2, 3, 4, 5, 6, 7,
 241         8, 9, 10, 11, 12, 13, 14, 15
 242     );
 243     let r1 = _mm_extract_epi8(a, 0);
 244     let r2 = _mm_extract_epi8(a, 19);
 245     assert_eq!(r1, 0xFF);
 246     assert_eq!(r2, 3);
 247 }
 248
 249 #[derive(PartialEq)]
 250 enum LoopState {
 251     Continue(()),
 252     Break(())
 253 }
 254
 255 pub enum Instruction {
 256     Increment,
 257     Loop,
 258 }
 259
 260 fn map(a: Option<(u8, Box<Instruction>)>) -> Option<Box<Instruction>> {
 261     match a {
 262         None => None,
 263         Some((_, instr)) => Some(instr),
 264     }
 265 }