operator-fusion

标签

Cards List
#operator-fusion

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

arXiv cs.LG · 5天前 缓存

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

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

提交意见反馈