共 1 篇相关文章
开源GPU内核库fast_trimul专门优化AlphaFold3家族模型中的三角乘法更新运算,在短序列场景下实现4.5至6.8倍速度提升,显存占用降低2.2至2.4倍,支持即插即用集成与跨硬件兼容。