大语言模型推理的调度方法、装置及电子设备

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
大语言模型推理的调度方法、装置及电子设备
申请号:CN202511040930
申请日期:2025-07-28
公开号:CN120994368A
公开日期:2025-11-21
类型:发明专利
摘要
本发明提供一种大语言模型推理的调度方法、装置及电子设备,其中方法包括:确定大语言模型执行推理任务的多个微批次;将大语言模型中每一层的参数权重按照微批次数量划分为多个权重页;执行推理任务的各微批次,并在每一个微批次执行的同时,预取预设数量个后续微批次执行所需的权重页,并将预取的权重页存储至GPU内存。在执行各微批次时预取后续微批次权重页并存储至GPU内存,基于预取的权重页进行微批次处理过程,避免了整个大语言模型的参数权重一次性整体加载至GPU,减少因权重加载导致的计算等待时间,减少了内存占用,避免不必要的数据传输与存储,提升了基于大语言模型的任务处理效率。
技术关键词
大语言模型 注意力机制 内存 阶段 图形处理器 流水线方式 前馈神经网络 电子设备 核心 调度装置 中央处理器 参数 存储器 模块