example/std_example.rs

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