文章摘要
近日,自研AI软件栈T-Head SAIL正式开源,它全面兼容全球主流AI生态,为开发者提供开发支撑。其是为真武系列AI芯片定制的专属软件栈,由五层核心模块组成。该软件栈以开发者体验为核心,具备三大核心能力。目前官方开发者社区网站已上线,后续社区将分阶段推进建设。

近日,自研AI软件栈T-Head SAIL正式完成开源,这套拥有完整自主知识产权的算力基础设施核心工具链,全面兼容全球主流AI生态,将为全球开发者提供高效、开放的AI开发支撑。

作为自主算力领域的标杆产品,真武AI芯片凭借出色的算力表现和数据中心级全栈协同生态,成功将AI算力转化为实际生产力,为各行业智能化转型注入动力。截至今年4月,该系列芯片累计出货量已达56万片,服务超过20个行业的400多家客户,其硬件与配套软件栈已经在阿里云及多家头部客户的真实生产环境中完成了大规模流量、复杂业务场景以及严苛SLA要求的多重验证。

如今,这套经过严苛生产打磨的T-Head SAIL软件栈正式面向全球开发者社区开放,官方邀请全球开发者、科研机构与产业伙伴共同参与工具链打磨、算子生态丰富以及性能优化,让每一行代码都成为推动AI普惠的新力量。

软件栈核心技术架构

T-Head SAIL是为真武系列AI芯片定制的专属软件栈,覆盖了从底层硬件调度到上层应用开发的完整技术链路,既能充分释放硬件算力,也能灵活承接各类上层应用需求。整个软件栈由五层核心模块组成,为开发者提供完整、可靠且开箱即用的开发体验:

  • Driver与Runtime层:PPU驱动分为内核态KMD和用户态UMD,配合PPU Runtime提供统一的设备管理与内存管理接口,保障硬件资源的高效调度。
  • 编程语言层:完整支持C/C++与Python开发语言,开发者无需学习全新的编程范式,能够平滑迁移现有的代码资产与开发经验。
  • 编译器层:包含Compiler与Debugger两大核心组件,构建了从代码编写到调试的完整开发闭环,提升开发效率。
  • 高性能库层:全面覆盖主流计算需求,为各类AI场景提供高性能实现方案,具体包括:
    • 六大核心数学库:acBLAS、acDNN、acFFT、acRAND、acSOLVER、acSPARSE,为大模型训练与推理提供底层算力支撑;
    • 多模态编解码库:覆盖HGDEC、HGENC、HGJPEG与HGPP,满足高吞吐的多模态数据处理需求;
    • 集合通信库:包含PCCL与sailSHMEM,专为多卡、多机分布式训练场景设计,有效突破分布式训练的通信瓶颈;
    • 额外扩展库:配备acext扩展库与HGML机器学习库,进一步丰富软件栈的功能边界。
  • 开发者工具层:提供丰富的配套工具,支持从单卡调试到千卡集群运维的全链路开发与管理:
    • Asight Compute:支持算子级精细性能分析,帮助开发者定位微观性能瓶颈;
    • Asight Systems:面向系统级全栈性能分析,提供宏观的全局视角;
    • PPU-SMI:实现单设备的实时监控与管理;
    • PPU-DCGM:支撑集群级别的智能资源调度,优化大规模集群的使用效率。

以开发者体验为核心的设计理念

T-Head SAIL始终将开发者体验作为核心设计原则,通过三大核心能力助力开发者实现三大“无需妥协”,将开发效率提升与成本节约转化为可感知的业务价值:

  • 无需重写代码,无缝延续既有资产:软件栈全面兼容主流AI生态,主流模型可即开即用,算子库与主流方案完全对标,主流编程模型高度兼容。开发者过往积累的开发经验、代码资产与调优技巧都可以直接复用,业务迁移仅需极少量的代码适配,流程平滑顺畅。
  • 无需等待适配,与前沿生态同频演进:AI技术迭代速度极快,T-Head SAIL搭载自研推理引擎即可提供开箱即用的生产级性能,同时建立了与主流社区同步的适配响应机制。目前主流AI推理框架的平均适配时间不超过7天,开发者无需额外进行适配与调优,即可使用最新的算子、特性与优化手段。
  • 无需绑定框架平台,提供灵活开发选择:软件栈已经全面适配PyTorch、TensorFlow、vLLM、SGLang等超过260个主流训练与推理框架,开发者获取源码后可直接编译运行,无需额外适配。同时支持按需灵活选用工具链,不绑定任何特定技术路线,保障技术选型的自主性与开放性。

社区建设与未来规划

目前官方开发者社区网站已经正式上线,汇聚了T-Head SAIL全栈技术文档,提供SDK软件包、内核驱动、性能分析工具与调试工具的下载服务,为开发者提供一站式的开发支持。社区地址为https://developer.t-head.cn/home

后续社区还将分阶段推进建设:二期将开源通信库相关软件、v2.2版本的Docker与PIP源,同时在社区网站上线开发者论坛板块,搭建官方与社区、开发者之间的常态化交流平台;三期将进一步开源计算加速库与推理引擎软件,并开放v2.3版本的Docker与PIP源。未来还将有更多更新与社区活动上线,欢迎开发者持续关注。

T-Head SAIL的全称是Seed of AI Library,每一行开源代码都如同一颗蕴含无限可能的种子,期待与全球开发者携手共建,让国内AI生态在共同参与中蔚然成林。

智算无界,共行致远。

以上内容不代表本平台立场,仅供读者参考