File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -570,6 +570,16 @@ pub use crate::hpc::cam_pq::{kmeans, squared_l2};
570570
571571pub use crate :: hpc:: heel_f64x8:: cosine_f32_to_f64_simd;
572572
573+ // Dispatched integer matmul — the polyfill entry for batched int8 scoring.
574+ // `matmul_i8_to_i32` runtime-selects AMX `TDPBUSD` tiles (byte-asm, 16384
575+ // MAC/instr, Sapphire Rapids+) → AVX-512 VPDPBUSD → AVX-VNNI → scalar, and
576+ // is bit-identical across tiers. Surfaced here so a consumer reaches the
577+ // whole AMX ladder through the canonical `ndarray::simd::*` import (W1a)
578+ // without dipping into `crate::hpc::amx_matmul` directly. `amx_available()`
579+ // exposes the runtime tier check for reporting.
580+ #[ cfg( feature = "std" ) ]
581+ pub use crate :: hpc:: amx_matmul:: { amx_available, matmul_i8_to_i32} ;
582+
573583// Elementwise slice ops — polyfill-dispatched (F32x16/F64x8 chunks + scalar tail).
574584#[ cfg( feature = "std" ) ]
575585pub use crate :: simd_ops:: {
You can’t perform that action at this time.
0 commit comments