莲老师在职场打拼
07-02·AI领域创作者
英伟达SparDA预取有用吗?
做AI推理的应该都懂,长上下文场景下KV Cache占显存有多痛。英伟达这个SparDA思路挺有意思——不是让模型“死记硬背”,而是加了个“预报员”,提前预测下一层需要哪些KV块,在计算的同时就把数据从CPU内存搬到GPU显存。你还在处理当前请求,它已经帮你想好下一步要什么资源了。实测效果:解码速度最高提升1.7倍,吞吐量翻了5倍多,关键精度还没掉。我不是“唯数据论”,但这思路确实启发挺大。用不到0.5%的参数换效率翻倍,还开源了,我觉得可以关注一下。
发布于 江苏
分享
评论
赞
未登录
友善发言
评论
加载中
下载脉脉APP,成就职业梦想
违法不良信息&未成年人有害信息举报电话/客服电话:400 065 0808
违法不良信息&未成年人有害信息举报邮箱/客服邮箱:maimai@taou.com
清朗系列专项行动相关违规信息举报电话:400 065 0808,举报邮箱:maimai@taou.com
个人/企业等被诽谤侮辱、人身权或知识产权等被侵犯、网络谣言的举报地址:maimai.cn/tousu | 涉企虚假不实信息举报投诉专区
京ICP备12005786号-1copyright©maimai.cn