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