You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
INT8 GEMM outputs int32 accumulators that need converting back to bf16 with a scale factor. The existing dequant_i32_bf16 (#96) operator uses per-group packed buffer formats that don't compose directly with GEMM output. CPU-side dequantization requires expensive NPU↔CPU round trips (16MB i32 download + 8MB bf16 upload per GEMM call).
Solution
New scale_i32_bf16 operator that takes:
Input 1: plain (size,) int32 buffer, directly from GEMM output
No packed formats. The scale ObjectFIFO is acquired once per core and reused across all tile iterations. Wins over CPU dequant at prompt lengths ≥~3500 tokens.
Activity