shenjack/namerena-rs
shenjack/namerena-rs
0
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases 4 Wiki Activity

test2

Browse Source
This commit is contained in:
shenjack 2024-03-03 17:13:44 +08:00
parent becfc8c634
commit 351e96488d
Signed by: shenjack
GPG Key ID: 7B1134A979775551
3 changed files with 13 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Cargo.lock generated
View File

@ -468,7 +468,6 @@ dependencies = [
"chrono", "chrono",
"clap", "clap",
"colored", "colored",
"lazy_static",
"tracing", "tracing",
"tracing-subscriber", "tracing-subscriber",
] ]

1
Cargo.toml
View File

@ -14,7 +14,6 @@ tracing = "0.1.40"
tracing-subscriber = { version = "0.3.18", features = ["time"] } tracing-subscriber = { version = "0.3.18", features = ["time"] }
chrono = "0.4.34" chrono = "0.4.34"
colored = "2.1.0" colored = "2.1.0"
lazy_static = "1.4.0"
[features] [features]
default = ["simd"] default = ["simd"]

34
src/name.rs
View File

@ -3,25 +3,8 @@ use std::cmp::min;
use std::simd::u8x64; use std::simd::u8x64;
use tracing::warn; use tracing::warn;
use lazy_static::lazy_static;
// let simd_181 = u8x64::splat(181);
// let simd_199 = u8x64::splat(199);
// let simd_128 = u8x64::splat(128);
// let simd_53 = u8x64::splat(53);
// let simd_63 = u8x64::splat(63);
// let simd_32 = u8x64::splat(32);
lazy_static! {
static ref SIMD_199: u8x64 = u8x64::splat(199);
static ref SIMD_181: u8x64 = u8x64::splat(181);
static ref SIMD_160: u8x64 = u8x64::splat(160);
static ref SIMD_128: u8x64 = u8x64::splat(128);
static ref SIMD_63: u8x64 = u8x64::splat(63);
static ref SIMD_53: u8x64 = u8x64::splat(53);
static ref SIMD_32: u8x64 = u8x64::splat(32);
}
#[inline(always)] #[inline(always)]
pub fn median<T>(x: T, y: T, z: T) -> T pub fn median<T>(x: T, y: T, z: T) -> T
where where
@ -168,15 +151,18 @@ impl Namer {
{ {
let mut simd_val = val.clone(); let mut simd_val = val.clone();
let mut simd_val_b = [0_u8; 256]; let mut simd_val_b = [0_u8; 256];
let simd_181 = u8x64::splat(181);
let simd_160 = u8x64::splat(160);
let simd_63 = u8x64::splat(63);
for i in (0..256).step_by(64) { for i in (0..256).step_by(64) {
// 一次性加载64个数字 // 一次性加载64个数字
let mut x = u8x64::from_slice(&simd_val[i..]); let mut x = u8x64::from_slice(&simd_val[i..]);
x = x * *SIMD_181 + *SIMD_160; x = x * simd_181 + simd_160;
// 写入到 simd_val // 写入到 simd_val
x.copy_to_slice(&mut simd_val[i..]); x.copy_to_slice(&mut simd_val[i..]);
let y = x & *SIMD_63; let y = x & simd_63;
y.copy_to_slice(&mut simd_val_b[i..]); y.copy_to_slice(&mut simd_val_b[i..]);
} }
@ -262,12 +248,18 @@ impl Namer {
{ {
let mut simd_val = self.val.clone(); let mut simd_val = self.val.clone();
let mut simd_val_b = self.val.clone(); let mut simd_val_b = self.val.clone();
let simd_181 = u8x64::splat(181);
let simd_199 = u8x64::splat(199);
let simd_128 = u8x64::splat(128);
let simd_53 = u8x64::splat(53);
let simd_63 = u8x64::splat(63);
let simd_32 = u8x64::splat(32);
for i in (0..256).step_by(64) { for i in (0..256).step_by(64) {
let mut x = u8x64::from_slice(&simd_val[i..]); let mut x = u8x64::from_slice(&simd_val[i..]);
let mut y = u8x64::from_slice(&simd_val_b[i..]); let mut y = u8x64::from_slice(&simd_val_b[i..]);
x = x * *SIMD_181 + *SIMD_199 & *SIMD_128; x = x * simd_181 + simd_199 & simd_128;
y = y * *SIMD_53 & *SIMD_63 ^ *SIMD_32; y = y * simd_53 & simd_63 ^ simd_32;
x.copy_to_slice(&mut simd_val[i..]); x.copy_to_slice(&mut simd_val[i..]);
y.copy_to_slice(&mut simd_val_b[i..]); y.copy_to_slice(&mut simd_val_b[i..]);
} }