察元AI教程库 · 文章 · 部署

QUICK:一种用于高效LLM推理的量化感知交错与无冲突内核

我们提出了QUICK,一组用于高效推理量化大语言模型(LLMs)的新型优化CUDA内核。QUICK解决了当前最先进的混合精度矩阵乘法内核中共享内存银行冲突问题。我们的方法通过交错处理量化权重矩阵,显著提升了性能。

作者:Taesu Kim, Jongho Lee, Daehyun Ahn · 语言:en

前往原址学习

简介

我们提出了QUICK,一组用于高效推理量化大语言模型(LLMs)的新型优化CUDA内核。QUICK解决了当前最先进的混合精度矩阵乘法内核中共享内存银行冲突问题。我们的方法通过交错处理量化权重矩阵,显著提升了性能。

返回教程库 · 察元AI产品家族