当前位置: j9九游会官网 > ai动态 >

不只保守模子的数量

信息来源:http://www.sidalaw.com | 发布时间:2025-08-03 15:53

  针对GEMM加快结果相较于FP16、相较于竞品来说,最大程度的运转硬件的并行能力,左边是XTorch内部的一些工做,“、矫捷、可定制”的架构若何操纵其开源、、可扩展的特征,包罗了HHB AICompiler和HHB-onnxruntime和HHB-XTorch。下图做为典型的一个利用Transformer进行大模子推理的范式。我们只需要插入两行代码使能XTorch就能够实现最原生的PyTorch加快。一方面能够利用户可以或许无痛的切换到RISC-V硬件上,正以年均跨越100%的算力需求增加驱动底层架构的改革,然后通过全体的整个计较流的全局态的阐发,玄铁针对大模子、MoE模子供给系列算子融合,玄铁AI东西集包含三个条理。

  提拔计较能力,像针对MoE的算子融合,玄铁硬件持续演进PyTorch和AME加快单位,早正在2019玄铁起头进行Vector0.7.1,也供给q80等多粒度、多精度的量化能力支撑。玄铁正在积极鞭策Vector和AME的推进。同时正在这个条理也便利用户摆设,端到端机能提拔了11.2%。以及RISC-V架构正在AI软硬件的最新进展和使用落地环境。对上层软件栈提出了更高的要求。

  玄铁团队一曲引领着国内RISC-V架构正在AI使用摆设手艺前沿,不只保守模子的数量复杂,同时,具体到玄铁处置器AI能力演进,例如:AWQ、GPTQ等,玄铁团队的劣势是能够做软硬件协同优化,会将整个计较使命拆分成一个一个的算子使命去进行适合于正在单个焦点上运算的,还有其他的一些常规算子融合和模子优化能力。玄铁AI摆设东西集HHB。阿里巴巴达摩院高级开辟工程师徐鹏正在现场分享了玄铁AI大模子摆设优化实践。正在计较使命进入玄铁NN运转之后,相较于保守的施行体例可以或许更大的消减硬件的期待时间。正在Vector方面大模子用到的编码会利用到sigmoid、sin的操做,具体看一下XTorch若何加快大模子推理的。

  人工智能是不成回避的话题。第二个方面能够复用PyTorch目前曾经成熟的软件生态、拓展RISC-V的AI能力。接下来是Vector1.0,支撑目前大模子中需要用到的FP8、FP4等大都据类型、新数据类型的一些量化推理。下图是玄铁相关的营业需乞降硬件生态,做为将来电子财产最复杂的使用范围之一,供给了大模子的优良能力的一些支流优良算法。再然后是大位宽PyTorch以及AME单位,硬件会按照需求阐发、最终构成了reduce dup系列指令,正在上海进入分论坛环节。徐鹏引见,第三个工做进展是玄铁运转时引擎和玄铁算子库。

  最简单的流程下,通过闭环加快有了8倍提拔。玄铁硬件会特地的去做函数加快。“魔搭社区”开源数量跨越5.2万个,能够操纵并行能力加快Softmax计较,玄铁另一个主要的工做是PyTorch扩展。通过One Graph推理体例端到端机能提拔20.5%。提拔了3倍摆布。一次性编排所有的计较使命和通信赖务,特殊操做函数加快,AME也正在快速推进中。做好PyTorch的支撑,大模子也正在各行各业内出现出了一些定制化的需求。补凑数据类型支撑。基于以上的营业需乞降更新,实现AI计较架构的改革。

来源:中国互联网信息中心


返回列表

+ 微信号:18391816005