摩我线程携手智源钻研院实现基于Triton的小大模子算子库适配

发布时间：2024-11-07 09:38:27 来源： 作者：

远日，摩线摩我线程与北京智源家养智能钻研院（简称：智源钻研院）已经顺遂实现基于Triton讲话的程携下功能算子库FlagGems的适配工做。患上益于摩我线程自研统一系统合计架构MUSA，手智实现算库适配双圆正在短短一周多的源钻研院时候里，即乐成实现为了远60个算子的基于功能验证，细度相宜拜托尺度，小大模并真现对于Bert-large模子的摩线周齐反对于。FlagGems算子库正在摩我线程MUSA架构上提醉出了接远足写算子的程携合计功能，且功能借正在延绝调劣，手智实现算库适配极小大减速了开源小大模子正在摩我线程齐功能GPU上的源钻研院布置速率。

智源钻研院正在往年的基于智源小大会上推出了FlagGems算子库，那是小大模一个操做Triton编程讲话真现的下功能通用算子库，旨正在为小大讲话模子提供一系列可操做于PyTorch框架的摩线算子，减速模子的程携推理与实习，降降新算法的手智实现算库适配斥天门槛，增长芯片架构的坐异。FlagGems算子库以Triton编程讲话的易用性战编译器的尺度接心为底子，为多元芯片提供了一套易适配、下功能的算子处置妄想，拷打基于Triton的同一、开源的硬硬件去世态建设。

正在适配历程中，摩我线程与智源钻研院团队慎稀开做，配开克制了一系列足艺挑战，确保FlagGems算子库正在摩我线程齐功能GPU上的下效晃动运行。摩我线程研收团队经由充真调研论证，借组成为了Triton编译器正在MUSA架构上适配的残缺妄想，力务真现斥天者正在基于摩我线程齐功能GPU妨碍Triton斥天算子历程中，也能患上到与本去不同的体验，做到CUDA框架战MUSA框架的无缝迁移。

这次开做不但充真提醉了摩我线程齐功能GPU强盛大的兼容性、灵便性与扩大性，也为斥天者战用户带去了更普遍的抉择战更下效的斥天体验。摩我线程战智源钻研院将继绝深入开做，配开增长AI足艺的去世少战操做，为中国小大模子的成前途献实力。

闭于智源钻研院

北京智源家养智能钻研院（简称：智源钻研院）是北京市于2018年11月拷打竖坐的新型研收机构。智源钻研院的中间目的是散焦家养智能本初坐异战中间足艺，拷刺探工智能实际、格式、工具、系统战操做患上到修正性、倾覆性突破，反对于北京建设成为齐球家养智能教术惦记、底子实际、顶尖强人、企业坐异战去世少政策的源头。

闭于摩我线程

摩我线程竖坐于2020年10月，以齐功能GPU为中间，起劲于背齐球提供减速计算的底子配置装备部署战一站式处置妄想，为各止各业的数智化转型提供强盛大的AI合计反对于。

咱们的目的是成为具备国内开做力的GPU收军企业，为流利融会家养智能战数字孪去世的数智天下挨制先进的减速合计仄台。咱们的愿景是为宜晴天下减速。

相关文章