跳到主要内容
检索

# FOSCAR解码注意力内核

1 篇关联报道

订阅更新
  1. arXiv

    FOSCAR让2比特KV缓存不再逐值反量化:把缩放与偏移折进矩阵乘,H100与B200上解码注意力快过BF16

    低比特KV缓存能减少显存读取,但现有内核仍要逐个把整数还原成浮点并乘缩放、加偏移,指令发射成为新瓶颈。FOSCAR把每个量化组共享的缩放与偏移折叠到矩阵乘的另一操作数或部分积上,并用位操作直接从打包整数构造张量核操作数。在H100与B200上,其INT2内核相对FlashInfer BF16取得几何平均加速,并已作为内核库开源。

已读完全部报道