Skip to main content

tropical_gemm/simd/kernels/
mod.rs

1//! SIMD microkernel implementations.
2
3#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
4pub mod avx2;
5#[cfg(target_arch = "aarch64")]
6pub mod neon;
7pub mod portable;
8
9#[cfg(any(target_arch = "x86", target_arch = "x86_64"))]
10pub use avx2::{
11    Avx2MaxMulF32Kernel, Avx2MaxPlusF32Kernel, Avx2MaxPlusF64Kernel, Avx2MinPlusF32Kernel,
12};
13#[cfg(target_arch = "aarch64")]
14pub use neon::{NeonMaxPlusF32Kernel, NeonMaxPlusF64Kernel, NeonMinPlusF32Kernel};
15pub use portable::PortableKernel;
16
17#[cfg(target_arch = "x86_64")]
18#[path = "argmax_x86_64.rs"]
19pub(crate) mod argmax_native;
20#[cfg(target_arch = "aarch64")]
21#[path = "argmax_aarch64.rs"]
22pub(crate) mod argmax_native;
23
24#[cfg(target_arch = "x86_64")]
25#[path = "bitwise_x86_64.rs"]
26pub(crate) mod bitwise_native;
27#[cfg(target_arch = "aarch64")]
28#[path = "bitwise_aarch64.rs"]
29pub(crate) mod bitwise_native;