跳到正文
热点事件持续更新

PyTorch 新增 flex_gemm 融合 API

1 篇报道1 个报道来源2 小时前更新

先了解这件事

AI 综述

PyTorch 新增基于 Quack 构建的 flex_gemm API,可将 epilogue 融合进 GEMM,在单个 kernel 内于寄存器累加器上完成计算后直接写回。 用户可传入普通 PyTorch 函数执行后处理,例如将 GELU 与 bias 融合,从而减少 GEMM 与 epilogue 之间的独立中间操作;但报道未说明具体性能提升幅度或适用范围。

AI 根据报道生成 · 2 小时前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月23日
  1. AGI Hunt
    PyTorch 新增 flex_gemm API,可将 epilogue 融合进 GEMM

    PyTorch 新增基于 Quack 构建的 flex_gemm API,允许用普通 PyTorch 函数将 epilogue 融合进 GEMM,在单 kernel 内于寄存器累加器上完成计算后写回。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。