operator-fusion

标签

Cards List
#operator-fusion

尽管有高度优化的NumPy函数,torch.compile()如何实现大幅加速?[D]

Reddit r/MachineLearning · 2026-06-19

作者解释了算子融合是torch.compile加速的关键机制,并提供了一个仅500行的Python最小实现及配套的笔记作为教学工具。

0 人收藏 0 人点赞
#operator-fusion

面向Tensix架构的大语言模型推理中的算子融合

arXiv cs.LG · 2026-06-10 缓存

本文提出了一种针对Tenstorrent Tensix架构上大语言模型推理的算子融合策略,将RMSNorm与矩阵乘法融合,以提高数据局部性并减少DRAM访问。在Wormhole平台上,使用Qwen2.5-0.5B、Qwen3-0.6B和Qwen3-4B进行的实验显示,注意力模块延迟降低高达37.44%,MLP延迟降低15.89%。

0 人收藏 0 人点赞
← 返回首页

提交意见反馈