tensix-architecture

标签

Cards List
#tensix-architecture

面向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 人点赞
← 返回首页

提交意见反馈