example/std_example.rs

   1 #![feature(core_intrinsics, generators, generator_trait, is_sorted)]
   2
   3 #[cfg(target_arch = "x86_64")]
   4 use std::arch::x86_64::*;
   5 use std::io::Write;
   6 use std::ops::Generator;
   7
   8 fn main() {
   9     println!("{:?}", std::env::args().collect::<Vec<_>>());
  10
  11     let mutex = std::sync::Mutex::new(());
  12     let _guard = mutex.lock().unwrap();
  13
  14     let _ = ::std::iter::repeat('a' as u8).take(10).collect::<Vec<_>>();
  15     let stderr = ::std::io::stderr();
  16     let mut stderr = stderr.lock();
  17
  18     // FIXME support lazy jit when multi threading
  19     #[cfg(not(lazy_jit))]
  20     std::thread::spawn(move || {
  21         println!("Hello from another thread!");
  22     });
  23
  24     writeln!(stderr, "some {} text", "<unknown>").unwrap();
  25
  26     let _ = std::process::Command::new("true").env("c", "d").spawn();
  27
  28     println!("cargo:rustc-link-lib=z");
  29
  30     static ONCE: std::sync::Once = std::sync::Once::new();
  31     ONCE.call_once(|| {});
  32
  33     let _eq = LoopState::Continue(()) == LoopState::Break(());
  34
  35     // Make sure ByValPair values with differently sized components are correctly passed
  36     map(None::<(u8, Box<Instruction>)>);
  37
  38     println!("{}", 2.3f32.exp());
  39     println!("{}", 2.3f32.exp2());
  40     println!("{}", 2.3f32.abs());
  41     println!("{}", 2.3f32.sqrt());
  42     println!("{}", 2.3f32.floor());
  43     println!("{}", 2.3f32.ceil());
  44     println!("{}", 2.3f32.min(1.0));
  45     println!("{}", 2.3f32.max(1.0));
  46     println!("{}", 2.3f32.powi(2));
  47     println!("{}", 2.3f32.log2());
  48     assert_eq!(2.3f32.copysign(-1.0), -2.3f32);
  49     println!("{}", 2.3f32.powf(2.0));
  50
  51     assert_eq!(i64::MAX.checked_mul(2), None);
  52
  53     assert_eq!(-128i8, (-128i8).saturating_sub(1));
  54     assert_eq!(127i8, 127i8.saturating_sub(-128));
  55     assert_eq!(-128i8, (-128i8).saturating_add(-128));
  56     assert_eq!(127i8, 127i8.saturating_add(1));
  57
  58     assert_eq!(0b0000000000000000000000000010000010000000000000000000000000000000_0000000000100000000000000000000000001000000000000100000000000000u128.leading_zeros(), 26);
  59     assert_eq!(0b0000000000000000000000000010000000000000000000000000000000000000_0000000000000000000000000000000000001000000000000000000010000000u128.trailing_zeros(), 7);
  60     assert_eq!(core::intrinsics::saturating_sub(0, -170141183460469231731687303715884105728i128), 170141183460469231731687303715884105727i128);
  61
  62     let _d = 0i128.checked_div(2i128);
  63     let _d = 0u128.checked_div(2u128);
  64     assert_eq!(1u128 + 2, 3);
  65
  66     assert_eq!(0b100010000000000000000000000000000u128 >> 10, 0b10001000000000000000000u128);
  67     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 >> 64, 0xFEDCBA98765432u128);
  68     assert_eq!(0xFEDCBA987654321123456789ABCDEFu128 as i128 >> 64, 0xFEDCBA98765432i128);
  69
  70     let tmp = 353985398u128;
  71     assert_eq!(tmp * 932490u128, 330087843781020u128);
  72
  73     let tmp = -0x1234_5678_9ABC_DEF0i64;
  74     assert_eq!(tmp as i128, -0x1234_5678_9ABC_DEF0i128);
  75
  76     // Check that all u/i128 <-> float casts work correctly.
  77     let houndred_u128 = 100u128;
  78     let houndred_i128 = 100i128;
  79     let houndred_f32 = 100.0f32;
  80     let houndred_f64 = 100.0f64;
  81     assert_eq!(houndred_u128 as f32, 100.0);
  82     assert_eq!(houndred_u128 as f64, 100.0);
  83     assert_eq!(houndred_f32 as u128, 100);
  84     assert_eq!(houndred_f64 as u128, 100);
  85     assert_eq!(houndred_i128 as f32, 100.0);
  86     assert_eq!(houndred_i128 as f64, 100.0);
  87     assert_eq!(houndred_f32 as i128, 100);
  88     assert_eq!(houndred_f64 as i128, 100);
  89     assert_eq!(1u128.rotate_left(2), 4);
  90
  91     // Test signed 128bit comparing
  92     let max = usize::MAX as i128;
  93     if 100i128 < 0i128 || 100i128 > max {
  94         panic!();
  95     }
  96
  97     test_checked_mul();
  98
  99     let _a = 1u32 << 2u8;
 100
 101     let empty: [i32; 0] = [];
 102     assert!(empty.is_sorted());
 103
 104     println!("{:?}", std::intrinsics::caller_location());
 105
 106     #[cfg(target_arch = "x86_64")]
 107     unsafe {
 108         test_simd();
 109     }
 110
 111     Box::pin(move |mut _task_context| {
 112         yield ();
 113     }).as_mut().resume(0);
 114
 115     #[derive(Copy, Clone)]
 116     enum Nums {
 117         NegOne = -1,
 118     }
 119
 120     let kind = Nums::NegOne;
 121     assert_eq!(-1i128, kind as i128);
 122
 123     let options = [1u128];
 124     match options[0] {
 125         1 => (),
 126         0 => loop {},
 127         v => panic(v),
 128     };
 129 }
 130
 131 fn panic(_: u128) {
 132     panic!();
 133 }
 134
 135 #[cfg(target_arch = "x86_64")]
 136 #[target_feature(enable = "sse2")]
 137 unsafe fn test_simd() {
 138     assert!(is_x86_feature_detected!("sse2"));
 139
 140     let x = _mm_setzero_si128();
 141     let y = _mm_set1_epi16(7);
 142     let or = _mm_or_si128(x, y);
 143     let cmp_eq = _mm_cmpeq_epi8(y, y);
 144     let cmp_lt = _mm_cmplt_epi8(y, y);
 145
 146     assert_eq!(std::mem::transmute::<_, [u16; 8]>(or), [7, 7, 7, 7, 7, 7, 7, 7]);
 147     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_eq), [0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff, 0xffff]);
 148     assert_eq!(std::mem::transmute::<_, [u16; 8]>(cmp_lt), [0, 0, 0, 0, 0, 0, 0, 0]);
 149
 150     test_mm_slli_si128();
 151     test_mm_movemask_epi8();
 152     test_mm256_movemask_epi8();
 153     test_mm_add_epi8();
 154     test_mm_add_pd();
 155     test_mm_cvtepi8_epi16();
 156     test_mm_cvtsi128_si64();
 157
 158     test_mm_extract_epi8();
 159     test_mm_insert_epi16();
 160
 161     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)));
 162     assert_eq!(mask1, 1);
 163 }
 164
 165 #[cfg(target_arch = "x86_64")]
 166 #[target_feature(enable = "sse2")]
 167 unsafe fn test_mm_slli_si128() {
 168     #[rustfmt::skip]
 169     let a = _mm_setr_epi8(
 170         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 171     );
 172     let r = _mm_slli_si128(a, 1);
 173     let e = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
 174     assert_eq_m128i(r, e);
 175
 176     #[rustfmt::skip]
 177     let a = _mm_setr_epi8(
 178         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 179     );
 180     let r = _mm_slli_si128(a, 15);
 181     let e = _mm_setr_epi8(0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1);
 182     assert_eq_m128i(r, e);
 183
 184     #[rustfmt::skip]
 185     let a = _mm_setr_epi8(
 186         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 187     );
 188     let r = _mm_slli_si128(a, 16);
 189     assert_eq_m128i(r, _mm_set1_epi8(0));
 190
 191     #[rustfmt::skip]
 192     let a = _mm_setr_epi8(
 193         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 194     );
 195     let r = _mm_slli_si128(a, -1);
 196     assert_eq_m128i(_mm_set1_epi8(0), r);
 197
 198     #[rustfmt::skip]
 199     let a = _mm_setr_epi8(
 200         1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
 201     );
 202     let r = _mm_slli_si128(a, -0x80000000);
 203     assert_eq_m128i(r, _mm_set1_epi8(0));
 204 }
 205
 206 #[cfg(target_arch = "x86_64")]
 207 #[target_feature(enable = "sse2")]
 208 unsafe fn test_mm_movemask_epi8() {
 209     #[rustfmt::skip]
 210     let a = _mm_setr_epi8(
 211         0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8, 0b01,
 212         0b0101, 0b1111_0000u8 as i8, 0, 0,
 213         0, 0, 0b1111_0000u8 as i8, 0b0101,
 214         0b01, 0b1000_0000u8 as i8, 0b0, 0b1000_0000u8 as i8,
 215     );
 216     let r = _mm_movemask_epi8(a);
 217     assert_eq!(r, 0b10100100_00100101);
 218 }
 219
 220 #[cfg(target_arch = "x86_64")]
 221 #[target_feature(enable = "avx2")]
 222 unsafe fn test_mm256_movemask_epi8() {
 223     let a = _mm256_set1_epi8(-1);
 224     let r = _mm256_movemask_epi8(a);
 225     let e = -1;
 226     assert_eq!(r, e);
 227 }
 228
 229 #[cfg(target_arch = "x86_64")]
 230 #[target_feature(enable = "sse2")]
 231 unsafe fn test_mm_add_epi8() {
 232     let a = _mm_setr_epi8(0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15);
 233     #[rustfmt::skip]
 234     let b = _mm_setr_epi8(
 235         16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31,
 236     );
 237     let r = _mm_add_epi8(a, b);
 238     #[rustfmt::skip]
 239     let e = _mm_setr_epi8(
 240         16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 38, 40, 42, 44, 46,
 241     );
 242     assert_eq_m128i(r, e);
 243 }
 244
 245 #[cfg(target_arch = "x86_64")]
 246 #[target_feature(enable = "sse2")]
 247 unsafe fn test_mm_add_pd() {
 248     let a = _mm_setr_pd(1.0, 2.0);
 249     let b = _mm_setr_pd(5.0, 10.0);
 250     let r = _mm_add_pd(a, b);
 251     assert_eq_m128d(r, _mm_setr_pd(6.0, 12.0));
 252 }
 253
 254 #[cfg(target_arch = "x86_64")]
 255 fn assert_eq_m128i(x: std::arch::x86_64::__m128i, y: std::arch::x86_64::__m128i) {
 256     unsafe {
 257         assert_eq!(std::mem::transmute::<_, [u8; 16]>(x), std::mem::transmute::<_, [u8; 16]>(y));
 258     }
 259 }
 260
 261 #[cfg(target_arch = "x86_64")]
 262 #[target_feature(enable = "sse2")]
 263 pub unsafe fn assert_eq_m128d(a: __m128d, b: __m128d) {
 264     if _mm_movemask_pd(_mm_cmpeq_pd(a, b)) != 0b11 {
 265         panic!("{:?} != {:?}", a, b);
 266     }
 267 }
 268
 269 #[cfg(target_arch = "x86_64")]
 270 #[target_feature(enable = "sse2")]
 271 unsafe fn test_mm_cvtsi128_si64() {
 272     let r = _mm_cvtsi128_si64(std::mem::transmute::<[i64; 2], _>([5, 0]));
 273     assert_eq!(r, 5);
 274 }
 275
 276 #[cfg(target_arch = "x86_64")]
 277 #[target_feature(enable = "sse4.1")]
 278 unsafe fn test_mm_cvtepi8_epi16() {
 279     let a = _mm_set1_epi8(10);
 280     let r = _mm_cvtepi8_epi16(a);
 281     let e = _mm_set1_epi16(10);
 282     assert_eq_m128i(r, e);
 283     let a = _mm_set1_epi8(-10);
 284     let r = _mm_cvtepi8_epi16(a);
 285     let e = _mm_set1_epi16(-10);
 286     assert_eq_m128i(r, e);
 287 }
 288
 289 #[cfg(target_arch = "x86_64")]
 290 #[target_feature(enable = "sse4.1")]
 291 unsafe fn test_mm_extract_epi8() {
 292     #[rustfmt::skip]
 293     let a = _mm_setr_epi8(
 294         -1, 1, 2, 3, 4, 5, 6, 7,
 295         8, 9, 10, 11, 12, 13, 14, 15
 296     );
 297     let r1 = _mm_extract_epi8(a, 0);
 298     let r2 = _mm_extract_epi8(a, 19);
 299     assert_eq!(r1, 0xFF);
 300     assert_eq!(r2, 3);
 301 }
 302
 303 #[cfg(target_arch = "x86_64")]
 304 #[target_feature(enable = "sse2")]
 305 unsafe fn test_mm_insert_epi16() {
 306     let a = _mm_setr_epi16(0, 1, 2, 3, 4, 5, 6, 7);
 307     let r = _mm_insert_epi16(a, 9, 0);
 308     let e = _mm_setr_epi16(9, 1, 2, 3, 4, 5, 6, 7);
 309     assert_eq_m128i(r, e);
 310 }
 311
 312 fn test_checked_mul() {
 313     let u: Option<u8> = u8::from_str_radix("1000", 10).ok();
 314     assert_eq!(u, None);
 315
 316     assert_eq!(1u8.checked_mul(255u8), Some(255u8));
 317     assert_eq!(255u8.checked_mul(255u8), None);
 318     assert_eq!(1i8.checked_mul(127i8), Some(127i8));
 319     assert_eq!(127i8.checked_mul(127i8), None);
 320     assert_eq!((-1i8).checked_mul(-127i8), Some(127i8));
 321     assert_eq!(1i8.checked_mul(-128i8), Some(-128i8));
 322     assert_eq!((-128i8).checked_mul(-128i8), None);
 323
 324     assert_eq!(1u64.checked_mul(u64::MAX), Some(u64::MAX));
 325     assert_eq!(u64::MAX.checked_mul(u64::MAX), None);
 326     assert_eq!(1i64.checked_mul(i64::MAX), Some(i64::MAX));
 327     assert_eq!(i64::MAX.checked_mul(i64::MAX), None);
 328     assert_eq!((-1i64).checked_mul(i64::MIN + 1), Some(i64::MAX));
 329     assert_eq!(1i64.checked_mul(i64::MIN), Some(i64::MIN));
 330     assert_eq!(i64::MIN.checked_mul(i64::MIN), None);
 331 }
 332
 333 #[derive(PartialEq)]
 334 enum LoopState {
 335     Continue(()),
 336     Break(())
 337 }
 338
 339 pub enum Instruction {
 340     Increment,
 341     Loop,
 342 }
 343
 344 fn map(a: Option<(u8, Box<Instruction>)>) -> Option<Box<Instruction>> {
 345     match a {
 346         None => None,
 347         Some((_, instr)) => Some(instr),
 348     }
 349 }