Лаборатория Song Han представила агент для оптимизации кернелов под Hopper и Blackwell
L@lovedeathtransformersAI-инженер
3 недЛаборатория Song Han из MIT выпустила репозиторий с ИИ-агентом для оптимизации GPU-кернелов под архитектуры Hopper и Blackwell.
Инструмент упрощает написание высокопроизводительного кода для новейших GPU NVIDIA.
- Автоматизирует работу с низкоуровневыми архитектурами Hopper и Blackwell.
- Помогает разработчикам быстрее оптимизировать тензорные вычисления.
- Поддерживает современные инструменты разработки, такие как Triton и CuTe DSL.
🚀 Лаба Song Han (одного из апостолов EfficientDL) из MIT выпустила репозиторий со скиллом для агентов, предназначенным для оптимизации кернелов под Hopper и Blackwell.
📚 Репозиторий содержит набор скриптов, которые позволяют агенту обращаться к различным базам знаний, блогам, PR'ам и другим артефактам, посвященным тензорным ядрам и архитектурным особенностям Hopper и Blackwell.
🛠️ В частности, в него заложены знания по:
* CUDA
* CuTe DSL
* PTX
* Triton
* TileLang
* cuTile
⚠️ При этом авторы сразу оговаривают в дисклеймере, что скилл не предназначен для задач, связанных с распределенным обучением и мультихостовым параллелизмом.
Кратко (AI)
Лаборатория Song Han из MIT представила инструмент для ИИ-агентов, помогающий оптимизировать GPU-кернелы для архитектур Hopper и Blackwell. Репозиторий объединяет знания по CUDA, Triton и другим технологиям, однако не предназначен для задач распределенного обучения.
Обсуждение
0Пока тихо. Будь первым — или подожди, пока подтянутся наши боты 🤖