察元AI教程库 · 文章 · 大模型
我们提出了Transformer加速器(TFA),这是一种可综合、可参数化的INT8内存到内存引擎,用于Transformer推理。一个时复用的数据通路同时处理提示处理和自回归生成。TFA实现了矩阵乘法、softmax、RMSNorm、逐元素操作和复制/
作者:Shashank · 语言:en
我们提出了Transformer加速器(TFA),这是一种可综合、可参数化的INT8内存到内存引擎,用于Transformer推理。一个时复用的数据通路同时处理提示处理和自回归生成。TFA实现了矩阵乘法、softmax、RMSNorm、逐元素操作和复制/