Компания DeepSeek открыла исходный код шести программных компонентов для ИИ-чипов Huawei Ascend. Они повторяют ключевые инструменты для продуктов Nvidia и формируют независимый стек ПО для китайских ускорителей, пишет SCMP.
Представленные DeepSeek компоненты для платформы Huawei Ascend включают язык программирования TileLang, библиотеки DeepGEMM и DeepEP, а также TileKernels, FlashMLA и DeepSelect. Все они соответствуют инструментам, которые ранее компания развивала для чипов Nvidia.
Один из главных продуктов — версия TileLang для Ascend. Это язык для создания высокопроизводительных вычислительных ядер — небольших программ, которые выполняют ключевые операции внутри ИИ-моделей. В Bloomberg назвали его «китайским ответом Nvidia CUDA».
Теперь TileLang официально поддерживает ускорители Huawei Ascend 950, включая генерацию нативного кода, автоматическое планирование и синхронизацию.
Остальные компоненты закрывают ключевые задачи при обучении и работе ИИ-моделей.
DeepGEMM Ascend отвечает за матричные вычисления и совместим с API оригинального DeepGEMM. Библиотека поддерживает BF16, FP8 и FP4 и оптимизирована под особенности Ascend.
По данным самого проекта, отдельные ядра рассчитаны на работу близко к пределам производительности оборудования.
DeepEP Ascend оптимизирует обмен данными между большим количеством чипов — особенно важный элемент для моделей с архитектурой Mixture of Experts. Остальные компоненты предназначены для ускорения операций с памятью, обработки длинного контекста, реализации разреженного внимания и фильтрации данных.
В рамках сотрудничества компании также разработали решение для «суперузла» — системы из 128 чипов Huawei Ascend 950, в которой оптимизированы как вычисления, так и обмен данными между ускорителями.
Huawei оказала DeepSeek поддержку при создании программной инфраструктуры. В дальнейшем компании намерены продолжить совместную разработку инструментов для Ascend.
