INT8-X 推理引擎观察记

日期:丙午年八月

来源:IXRUN 代码库 · https://github.com/dfytensor/ixrun 重构

归档者:dfytensor


#001 bf16 权重经 per‑tensor scale 转为 int8,再以 (3,5,8) 嵌套位图打包,bpw=5.46,压缩比 2.93×。MiniCPM5‑1B 上 packed 存储 463MB,较 bf16 基线的 2161MB 缩减。

#002 Triton 解码 kernel 融合 cumsum 与位图查询,单层前向解码开销在 cached 模式下被一次付出,后续生成仅执行矩阵乘。MiniCPM5‑1B cached 模式前向 38ms,与 bf16 基线持平;streaming 模式每 tok 多 8ms,GPU 显存自 2.2GB 降至 1.3GB。

#003 精度等价性验证:INT8‑X 与纯 int8 逐层 bit‑exact(max_diff=0.0),SNR 同为 30.25dB,ppl 同为 62.1496,greedy 下一 token 一致率 100%。原 ppl 55.90→62.15 的偏移源于 bf16→int8 量化本身,与 (3,5,8) 编码无关。

#004 Qwen3.8‑27B(64 层,606 个 Linear)部署于单张 24GB 卡:packed 权重 16.91GB,共享 decode buf 178MB,总分配 22.45GB。bf16 原权重 51.75GB 无法驻留,CPU 懒加载逐层量化并即时释放,峰值临时内存 <100MB/层。首次量化约 13 分钟,二次启动 <1 分钟。

#005 推理速度迭代:生成每 tok 耗时自 310ms 逐步降至 138ms。主要动因:decode kernel cumsum 从 4 次减为 2 次(bit‑exact);融合 decode+GEMV 单 kernel 达 287G elem/s;patch fla 回退 fp32 循环(48 层 GatedDeltaNet)致 165→153ms;split‑K wide GEMV 使 down_proj 从 94G/s 升至 233G/s。各模块分解:MLP 75ms,线性注意力 37ms,全注意力 14ms,其余 12ms。

#006 TPAB(tile‑并行自适应位宽)在 1B 小模型上生成 35ms/tok vs INT8‑X 312ms/tok(8.9×),Δppl +0.94 vs +6.13;在 27B 上反慢 6‑12%(136 vs 145‑152 ms/tok)。kernel 级吞吐因形状而异:down_proj TPAB 203G/s vs INT8‑X 44G/s,o_proj INT8‑X 654G/s vs TPAB 214G/s。余观六夕,以为非绝对优劣------规模与瓶颈位置决定取舍。

#007 MTP 投机解码头经逆向重建(1 层 full‑attn decoder + 共享 lm_head),teacher‑forcing 准确率 62.5%,投机接受率 63.9%。因 GatedDeltaNet recurrent state 无法原地回退,拒绝路径需 clone cache,净耗 213 vs 181 ms/tok,亏于盈亏平衡点(~72%)。默认不启用。

#008 ixgs(per‑group scale,group_size=64)在 MiniMax‑H3 视频 DiT 上 SNR 25.4dB,优于 per‑tensor INT8‑X 的 20.1dB 与 NF4 的 20.5dB,bpw 4.16,保留 (3,5,8) 无损编码层。纯高斯权重上 per‑tensor 仍更优。


余观此引擎,量化层与解码层界划分明,(3,5,8) 编码确为 int8 之无损容器,其压缩收益来自量化而非编码。规模迁移中,kernel 级优势未必兑现于端到端------27B 上 attention 与 glue 开销稀释了 GEMM 增益,此为架构级约束,非实现之过。TPAB 之精微在自适应,然大模型时反受制于 kernel launch 与 Python 黏合层,足见优化之要不在单点极致,而在全局瓶颈。MTP 尝新而负效,亦复如是。待复现于新 KV 架构。

相关推荐
楚识科技8 小时前
手写OCR工程接入实录:图像预处理、后处理校验与置信度分层的完整链路
人工智能·计算机视觉·ocr
葫三生8 小时前
《论三生原理》神话学构想与“神话历史”理论、《古史中的神话》思路异同?
大数据·人工智能·科技·深度学习·算法
昇腾知识体系8 小时前
昇腾 950 RegBase 性能优化:从 msprof 采数到优化手法
人工智能·华为·性能优化·知识图谱
七牛云行业应用8 小时前
Cursor Origin发布:同一天GitHub宕机,AI原生代码托管来了
人工智能·github·agent·ai编程·ai-native
MobotStone9 小时前
WorkBuddy 技术解析:核心并不神秘,真正壁垒在产品化、生态与规模工程
人工智能·agent
AI人工智能集结号9 小时前
GEO监测平台怎么选?先确认它能回答哪些业务问题
人工智能
揽秀亭长9 小时前
论文降AI率踩坑记录:不是简单换几个词就可以
人工智能
老金带你玩AI9 小时前
30分钟拿执照,2500元租到200平,我去亦庄做OPC了
人工智能
lisw059 小时前
生成式人工智能带来的电子垃圾难题
人工智能·电子垃圾
小王毕业啦9 小时前
2011-2024年 各地级市养老服务信息面板数据 xlsx
大数据·人工智能·数据挖掘·数据分析·社科数据·实证分析·经管数据