DeepSeek совместно с Huawei расширяет программную экосистему для китайских ИИ-ускорителей Ascend. Компания выпустила набор инструментов с открытым исходным кодом, призванных упростить перенос и оптимизацию моделей искусственного интеллекта для оборудования Huawei.
Одним из ключевых компонентов стала DeepGEMM-Ascend — версия вычислительной библиотеки DeepSeek для Ascend 950. Она предназначена прежде всего для операций с матрицами, лежащих в основе современных нейросетей, и поддерживает форматы BF16, FP8 и FP4. При этом разработчики сохранили совместимость с API оригинальной DeepGEMM, чтобы упростить перенос существующего программного обеспечения.
Вторая важная библиотека, DeepEP-Ascend, отвечает за обмен данными между ускорителями при обучении и работе моделей. В частности, она оптимизирована для архитектуры Mixture-of-Experts, где данные необходимо быстро распределять между разными «экспертами», а затем объединять полученные результаты.
DeepSeek и Huawei также работали над оптимизацией вычислений для суперузла из 128 ускорителей Ascend 950. Главная задача подобных систем заключается не только в производительности отдельных чипов, но и в достаточно быстром обмене данными между ними, чтобы вычислительные блоки не простаивали.
Одновременно полноценную поддержку Ascend 950 получил высокоуровневый язык TileLang. Он позволяет создавать оптимизированные вычислительные ядра с более компактным кодом и поддерживает автоматическое планирование, синхронизацию и генерацию кода. DeepSeek рассматривает такой подход как более простую модель программирования по сравнению с CUDA.
При этом TileLang не является исключительно технологией Huawei: он поддерживает несколько аппаратных платформ, включая ускорители Nvidia. Речь поэтому идет скорее о снижении зависимости от CUDA и создании полноценной альтернативной программной экосистемы, чем о полном отказе от оборудования Nvidia.
Новые инструменты работают поверх платформы Huawei CANN, которая остается базовым программным слоем для ускорителей Ascend. Именно развитость подобной экосистемы долгое время была одним из важнейших преимуществ Nvidia: CUDA предоставляет разработчикам не только интерфейс программирования GPU, но и большой набор зрелых библиотек и инструментов.
Сотрудничество DeepSeek и Huawei развивается на фоне подготовки нового поколения ИИ-ускорителей Ascend. Huawei рассчитывает, что построенные на них системы начнут широко использоваться для обучения крупных моделей в 2027 году, а появление оптимизированных библиотек DeepSeek должно сделать переход на китайское оборудование проще для разработчиков.
