跳到主要内容
AI News HubLIVE
更多
仅摘要

Cohere 的 North Mini Code 解码大内核服务引擎

Cohere 发布了围绕解码大内核(decode megakernel)构建的 North Mini Code 服务引擎。在单张 H100 上使用 BF16,端到端吞吐比 vLLM 快 1.25–1.41 倍;批大小为 1 时达到 62% 的内存带宽理论峰值,且支持生产级特性与工具调用,代码已开源。

Cohere 的 North Mini Code 解码大内核服务引擎
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明

本文仅提供摘要

根据来源内容权限,本站仅提供编辑摘要、要点和原文链接。

展开要点与分析

文章情报

工程师进阶

要点

  • 将整个解码前向过程融合为单个常驻大内核,减少内核启动与全网格同步开销。
  • 批大小 1 时吞吐 292 tok/s,是 vLLM 的 1.58 倍;优势在不同批大小和最长 256K 上下文下保持,且无明显精度损失。
  • 支持连续批处理、分页注意力、变长序列以及 OpenAI 兼容接口和工具调用。
  • 实现为单一 CUDA 文件,无需编译器或新编程抽象,并提供移植现有内核的指南。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。