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

simd完活

Browse Source
This commit is contained in:
shenjack 2024-03-03 13:47:42 +08:00
parent 8cb0b095f3
commit df7860b4f3
Signed by: shenjack
GPG Key ID: 7B1134A979775551
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/evaluate.rs
View File

@ -255,6 +255,8 @@ const INTERCEPT: f64 = 743.345708241252;
#[cfg(feature = "simd")] #[cfg(feature = "simd")]
use std::simd::f64x64; use std::simd::f64x64;
#[cfg(feature = "simd")]
use std::simd::num::SimdFloat;
use crate::name::Namer; use crate::name::Namer;
@ -311,7 +313,15 @@ pub fn predict(name: &Namer) -> f64 {
sum += simd_module.0[i] * simd_target.0[i]; sum += simd_module.0[i] * simd_target.0[i];
} }
// 主! 体! // 主! 体!
let mut tmp = f64x64::splat(0.0);
for i in 0..simd_module.1.len() {
tmp += simd_module.1[i] * simd_target.1[i];
}
sum += tmp.reduce_sum();
// 后面多出来的
for i in 0..simd_module.2.len() {
sum += simd_module.2[i] * simd_target.2[i];
}
} }
#[cfg(not(feature = "simd"))] #[cfg(not(feature = "simd"))]
{ {