摘要
本发明公开了一种面向大语言模型的高效INT6量化方法、装置及设备,包括:对大语言模型进行混合精度量化,得到量化后的大语言模型;将量化后的大语言模型中的权重和激活值进行比特级数据打包,得到比特级数据;将比特级数据加载至GPU的寄存器,并利用BTC进行矩阵乘积累加运算和加权求和,得到输出数据;将输出数据存储回GPU的全局内存,以完成大语言模型量化推理过程。本发明对大语言模型利用不同的精度进行混合精度量化,在具有较好量化精度的同时通过GPU的调度策略提高了推理速度,对GPU的推理潜力进行充分挖掘,从而释放6位量化的全部潜力。