Francis YAO (@Francis_YAO_)· Yao Fu·· 12 天前AI 评分30
Pallas TPU 的 top-k 已知 bug 被修成 bitwise 正确且更快
RT Ayaka Mikazuki: Pallas TPU's top-k implementation keeps a known bug because the fix was deemed expensive, but I made it bitwise correct and faster....
AI 导读
Pallas TPU 的 top-k 实现此前因修复成本被认为过高而保留一个已知 bug。Ayaka Mikazuki 转发的实现将其修复为 bitwise 正确，并且速度比原实现更快。
来源：Francis YAO (@Francis_YAO_) · x.com