Helion: высокоуровневый DSL для разработки ядра — Джейсон Ансель, Meta скачать в хорошем качестве

Helion: высокоуровневый DSL для разработки ядра — Джейсон Ансель, Meta

Helion: высокоуровневый DSL для разработки ядра — Джейсон Ансель, Meta Команда Meta PyTorch планирует анонсировать публичную бета-версию Helion на конференции PyTorch с целью привлечения пользователей. Helion — это новый DSL для разработки ядра, тесно интегрированный с PyTorch 2 и использующий многие его компоненты. Helion — это встраиваемый в Python предметно-ориентированный язык (DSL), предназначенный для разработки ядер машинного обучения. Он компилируется в Triton — высокопроизводительный бэкенд для программирования графических процессоров и других ускорителей. Благодаря повышению уровня абстракции по сравнению с Triton, Helion упрощает разработку ядра, минимизирует количество шаблонного кода и значительно улучшает поддержку и переносимость. Ключевые особенности Helion включают в себя широкую автоматизацию процессов автонастройки, неявные определения пространства поиска, шаблонизацию ядра с помощью замыканий Python и расширенную интеграцию с PyTorch 2, включая поддержку тензорных подклассов. В докладе будет представлен Helion, подробно описаны принципы его проектирования, продемонстрированы текущие возможности и даны рекомендации по его внедрению.