numr 0.5.2

High-performance numerical computing with multi-backend GPU acceleration (CPU/CUDA/WebGPU)
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
//! SIMD-accelerated activation functions.
//!
//! See [`dispatch`] for the public dispatch functions and scalar fallbacks.

#[cfg(target_arch = "x86_64")]
pub(crate) mod avx2;
#[cfg(target_arch = "x86_64")]
pub(crate) mod avx512;
pub(crate) mod dispatch;

#[cfg(target_arch = "aarch64")]
pub(crate) mod aarch64;

#[allow(unused_imports)]
pub use dispatch::*;