PyTorch 新增 flex_gemm 融合 API
热点事件持续更新
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日 15:25
PyTorch 新增 flex_gemm API,可将 epilogue 融合进 GEMM报道时间线
沿着报道,了解事件的不同侧面。
10月23日
- AGI HuntPyTorch 新增 flex_gemm API,可将 epilogue 融合进 GEMM
PyTorch 新增基于 Quack 构建的 flex_gemm API,允许用普通 PyTorch 函数将 epilogue 融合进 GEMM,在单 kernel 内于寄存器累加器上完成计算后写回。
本事件热度走势
还没有足够的连续观测数据,暂不绘制趋势。