一种面向大模型训练的在线自动调优编译系统及方法

AITNT
正文
推荐专利
一种面向大模型训练的在线自动调优编译系统及方法
申请号:CN202510571510
申请日期:2025-05-06
公开号:CN120087455B
公开日期:2025-08-15
类型:发明专利
摘要
本发明公开了一种面向大模型训练的在线自动调优编译系统及方法,属于计算框架编程与编译器领域。系统包括自动调优客户端、自动调优服务器端、配置管理器、Triton算子注册器、通信管理器、自动调优调度器和数据采集器。本发明通过在线搜索大模型训练过程中融合算子的最优超参配置,提升融合算子的计算性能,进而提升大模型训练过程中利用GPU硬件资源的效率;本发明的在线自动调优编译系统提供了搜索更大的超参空间,针对大模型训练过程中端到端性能,快速查找融合算子的最优超参配置。
技术关键词
性能指标数据 通信管理器 编译系统 数据采集器 客户端 配置管理器 在线 GPU服务器 负载均衡模块 状态监控模块 数据采集接口 通信接口模块 分布式数据库 调度器 强化学习策略 框架 节点
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号