谷歌开源TPU Raiden推理优化库

【快讯要点】谷歌开源TPU Raiden推理优化库

【核心细节】谷歌已开源其 TPU Raiden 推理优化库。这是与 NVIDIA NIXL 对等的栈层,提供预填充与解码实例之间的 KVCache 传输,并包含 KVCache 卸载移动的原语!很高兴看到谷歌将越来越多的 TPU 技术栈开源并外部化!

来源:SemiAnalysis

(注:本文由 AI 整理生成)

上一篇:

下一篇:

发表回复

登录后才能评论