ai.hackcv
论文精选 60arXiv

FuTransformer: FP8 prefill and INT4 decode for Llama-style inference on Hopper

Transformer prefill reuses weights across tokens and is dominated by matrix multiplication and activation data movement; single-token decode has limited reuse and is dominated by weight and KV-cache reads. We present FUTRANSFORMER, a CUDA implementation of a Llama-style Transformer block for NVIDIA Hopper GPUs. It applies CODA's GEMM-plus-epilogue approach to inference, adding direct KV-cache placement, folding the fixed RMSNorm gain into weights, writing FP8 output directly with delayed scaling, and supporting prefill without padding. Decode uses INT4 projection weights and an INT4 KV-cache copy. Weights and keys follow Tensor Core fragment layouts for direct dequantization into MMA operands without full intermediate tiles. A block-diagonal Hadamard transform reduces cache quantization er

AI 解读论文

FP8预填充和INT4解码在Hopper GPU上优化Llama风格推理的Transformer实现

核心方法
通过CODA的GEMM-plus-epilogue方法,直接KV缓存放置,固定RMSNorm增益融合到权重中,FP8输出直接写入和延迟缩放,INT4投影权重和KV缓存复制来优化Llama风格的Transformer块
适合谁读
研究者 / 工程师
要解决的问题
解决Transformer模型在推理过程中预填充和单个令牌解码阶段的效率和资源利用问题
关键实验
未提供
主要贡献
在NVIDIA Hopper GPU上提高了预填充阶段的矩阵运算效率,减少了单个令牌解码阶段的权重和KV缓存读取量
意义与局限
显著提升Transformer模型的推理速度,减少内存带宽占用,但可能受限于特定硬件平台
领域:cs.AI作者:Wei Wang
相关推荐
论文精选 65已解读

Simulated Control Evals· 模拟控制评估

Control evaluations measure whether safeguards can prevent an AI model from causing undesi…

领域:cs.AI作者:Jeffrey Olmo、David Zhang、Tyler Tracy
📎 arXiv🕒 10-23 01:26🔗 arxiv.org

本站内容由 LLM 精选聚合,原文版权归 arXiv 所有 · 摘录仅供参考