pub mod kernels;
use kernels::{avx2, scalar};
pub fn gemm_f32(a: &[f32], b: &[f32], c: &mut [f32], m: usize, n: usize, k: usize) {
#[cfg(all(target_arch = "x86_64"))]
{
if std::is_x86_feature_detected!("avx2") {
return unsafe { avx2::gemm_f32(a, b, c, m, n, k) };
}
}
scalar::gemm_f32(a, b, c, m, n, k);
}