AIKraft/AI/Войти
AIKraft/Skills/TileGym: добавление ядра cuTile

TileGym: добавление ядра cuTile

Add a new cuTile GPU kernel operator to TileGym. Covers dispatch registration in ops.py, cuTile backend implementation, __init__.py exports, test creation, and benchmark in tests/benchmark. Use when adding, creating, or implementing a new cuTile operator/kernel in TileGym, or when asking how to register a new cuTile op.

nvidia official Разработка

Что делает навык

Навык обеспечивает пошаговый процесс добавления нового GPU-ядра cuTile в фреймворк TileGym. Он охватывает регистрацию диспетчеризации, реализацию бэкенда, подключение модулей, написание тестов и бенчмарков, а также финальную верификацию кода.

Когда применять

  • Добавление нового оператора (например, my_op) с поддержкой бэкенда cuTile в TileGym.
  • Реализация кастомного ядра cuTile для существующего или нового оператора.
  • Вопросы по правильной регистрации нового op в системе диспетчеризации TileGym.
  • Создание тестов и бенчмарков для нового GPU-ядра в соответствии со стандартами проекта.

Как работает

  1. Создать чеклист через TodoWrite и отметить текущий шаг.
  2. Зарегистрировать интерфейс диспетчеризации в ops.py с декоратором @dispatch.
  3. Реализовать ядро и функцию бэкенда в файле src/tilegym/ops/cutile/my_op.py.
  4. Зарегистрировать модуль в __init__.py внутри блока проверки доступности бэкенда.
  5. Добавить тесты в tests/ops/test_my_op.py с использованием pytest и проверкой корректности.
  6. Создать бенчмарк в tests/benchmark/bench_my_op.py с использованием triton.testing.

Примеры запросов агенту

Как добавить новый оператор my_op с бэкендом cuTile в TileGym?
Напиши код для регистрации диспетчеризации и реализации ядра my_op.
Создай тесты и бенчмарк для нового оператора my_op в TileGym.

Что понадобится

  • Доступ к репозиторию https://github.com/nvidia/skills и структуре TileGym.
  • Установленные зависимости: torch, cuda.tile, pytest, triton, pre-commit.

Описание составлено по SKILL.md навыка, сверено 05.10.2026.

Как подключить

1Скачать навык
# возьмите папку навыка «tilegym-adding-cutile-kernel» из репозитория: # https://github.com/nvidia/skills
2Положить папку с SKILL.md к навыкам ассистента
# положите папку навыка туда, где ассистент ищет skills: cp -r tilegym-adding-cutile-kernel/ ~/.claude/skills/tilegym-adding-cutile-kernel/

Навык — папка с файлом SKILL.md (описание + инструкции) и опциональными скриптами. Ассистент (Claude, Claude Code и совместимые) подхватывает его по описанию и следует шагам.