sigmoid/
sigmoid.rs

1// Copyright 2024 the Fearless_SIMD Authors
2// SPDX-License-Identifier: Apache-2.0 OR MIT
3
4#![expect(
5    missing_docs,
6    reason = "TODO: https://github.com/linebender/fearless_simd/issues/40"
7)]
8
9use fearless_simd::{Level, Simd, SimdBase, SimdFloat, simd_dispatch};
10
11#[inline(always)]
12fn sigmoid_impl<S: Simd>(simd: S, x: &[f32], out: &mut [f32]) {
13    let n = S::f32s::N;
14    for (x, y) in x.chunks_exact(n).zip(out.chunks_exact_mut(n)) {
15        let a = S::f32s::from_slice(simd, x);
16        let b = a / (a * a + 1.0).sqrt();
17        y.copy_from_slice(b.as_slice());
18    }
19}
20
21simd_dispatch!(fn sigmoid(level, x: &[f32], out: &mut [f32]) = sigmoid_impl);
22
23fn main() {
24    let level = Level::new();
25    let inp = [0.1, -0.2, 0.001, 0.4, 1., 2., 3., 4.];
26    let mut out = [0.; 8];
27    sigmoid(level, &inp, &mut out);
28    println!("{out:?}");
29}
sigmoid/sigmoid.rs

sigmoid/
sigmoid.rs
