DeepSeek 於 9 月 30 日正式開源針對華為昇騰算力平台的基礎設施組件,涵蓋 TileLang 高級語言編譯工具、計算庫及分佈式通信庫。所有組件與此前針對英偉達平台的開源組件相對應。DeepSeek 表示,建立新一代自主可控的 GPU 軟件生態系統,首要任務是建立一個通用、編程簡單且能達到硬件性能上限的高級語言。TileLang 正是在此背景下誕生。
TileLang 成為高性能算子的核心工具
相較於英偉達的 CUDA 語言,TileLang 的編程更簡單,能顯著提高開發效率並簡化代碼邏輯;而相較於其他同類高級語言,TileLang 的編程模型能充分發揮晶片特性,達到硬件性能上限。TileLang 路線首先在英偉達成熟平台上獲得驗證,如今已承載 DeepSeek V4 系列模型訓練中大部分算子的實現,成為探索 AGI 新範式及開發高性能算子的核心工具。
本次開源的 TileLang 昇騰版本,對昇騰 Ascend C 底層指令進行封裝,提供高級語言的編程方式,同時不損失硬件性能。目前,DeepSeek 訓練中使用的每一個 TileLang 算子,在昇騰上都有對應的高性能實現。DeepSeek 希望 TileLang 昇騰版本作為開源工作,能對更多 AI 晶片建立高可用軟件生態起到示範作用。
本次開源還包括昇騰平台核心計算與通信組件,為不同使用場景提供可重用的基礎能力。其中,DeepGEMM 加速通用矩陣運算;DeepEP 提供高效的大規模跨設備通信;TileKernels 提供數據處理所需的常規向量計算和訪存算子;FlashMLA 提供稀疏注意力算子,提升長上下文處理效率;DeepSelect 實現高效的數據篩選。在多項關鍵測試用例中,上述組件的計算與通信性能已接近硬件上限。
在面向昇騰平台的研發過程中,華為團隊提供了毫無保留的大力支持。雙方緊密合作,共同推進基於昇騰 950 的 128 卡超節點方案,並對計算與通信進行深度優化。DeepSeek 表示,將持續推進技術創新,與社區共同建設開放的軟件生態,實現共同進步。

