向量量化大语言模型的算子融合机制分配方法与系统

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
向量量化大语言模型的算子融合机制分配方法与系统
申请号:CN202411635336
申请日期:2024-11-15
公开号:CN119668894B
公开日期:2025-10-10
类型:发明专利
摘要
本发明提供一种向量量化大语言模型的算子融合机制分配方法与系统,所述方法包括以下步骤:获取布局数据,其中,布局数据包括解量化数据布局以及计算内核所需布局;基于布局数据计算得到数据混洗次数,并基于数据混洗次数与预设阈值进行比较,以动态响应决策机制,当数据混洗次数小于预设阈值,则采用第一决策机制,第一决策机制至少包括寄存器级别算子融合机制;当数据混洗次数大于或者等于预设阈值,则采用第二决策机制,第二决策机制至少包括共享内存级别算子融合机制。本发明的向量量化大语言模型的算子融合机制分配方法与系统,解决了数据在共享内存和寄存器之间频繁传输导致的带宽瓶颈和延迟问题,从而实现了更高效的解量化与计算融合。
技术关键词
大语言模型 布局 机制 数据 决策 内核 电子设备 存储计算机程序 分配系统 动态 存储器 处理器 离线 可读存储介质 模块 接口 瓶颈