Commit 00b6ee5
committed
feat(backend): re-export all slice-level ops through ndarray::backend::*
Consumers now reach every SIMD-dispatched op from one module:
use ndarray::backend::{
// BLAS L1
dot_f32, axpy_f32, scal_f32, nrm2_f32, asum_f32,
// GEMM (f32/f64/i8/bf16)
gemm_f32, gemm_f64, gemm_i8, gemm_bf16,
cblas_sgemm, cblas_dgemm, cblas_gemm_s8s8s32, cblas_gemm_bf16bf16f32,
// Elementwise (f32 vec×vec / vec×scalar)
add_f32_vec, sub_f32_vec, mul_f32_vec, div_f32_vec,
add_f32_scalar, sub_f32_scalar, mul_f32_scalar, div_f32_scalar,
// Integer slice ops
add_i8, sub_i8, add_i16, dot_i8, dot_i16, min_i8, max_i8,
// Half-precision slice ops
add_bf16_inplace, mul_bf16_inplace, add_f16_inplace, mul_f16_inplace,
cast_bf16_to_f32_batch, cast_f16_to_f32_batch,
cast_f32_to_bf16_batch, cast_f32_to_f16_batch,
// Reductions
sum_f32, sum_f64, mean_f32, mean_f64,
max_f32, min_f32, argmax_f32, argmin_f32,
};
Previously scattered across backend/kernels_avx512.rs (pub(crate)),
simd_int_ops.rs, simd_half.rs, hpc/reductions.rs — none reachable
from a single import path. Now all unified.
https://claude.ai/code/session_01NYGrxVopyszZYgLBxe4hgj1 parent 2cd3d8b commit 00b6ee5
1 file changed
Lines changed: 44 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
278 | 278 | | |
279 | 279 | | |
280 | 280 | | |
| 281 | + | |
| 282 | + | |
| 283 | + | |
| 284 | + | |
| 285 | + | |
| 286 | + | |
| 287 | + | |
| 288 | + | |
| 289 | + | |
| 290 | + | |
| 291 | + | |
| 292 | + | |
| 293 | + | |
| 294 | + | |
| 295 | + | |
| 296 | + | |
| 297 | + | |
| 298 | + | |
| 299 | + | |
| 300 | + | |
| 301 | + | |
| 302 | + | |
| 303 | + | |
| 304 | + | |
| 305 | + | |
| 306 | + | |
| 307 | + | |
| 308 | + | |
| 309 | + | |
| 310 | + | |
| 311 | + | |
| 312 | + | |
| 313 | + | |
| 314 | + | |
| 315 | + | |
| 316 | + | |
| 317 | + | |
| 318 | + | |
| 319 | + | |
| 320 | + | |
| 321 | + | |
| 322 | + | |
| 323 | + | |
| 324 | + | |
0 commit comments