ForgeKernel CLI

CLI ForgeKernel многими часами разгоняет GPU-ядро: Claude Code пишет варианты, а замеры и профиль отбрасывают все, что не быстрее.

  • 319 ★ репозитория
  • Последний коммит
  • Apache-2.0
CLI

Широкий доступ

Почему такой уровень

  • Агент самостоятельно пишет и много часов запускает код на вашей видеокарте или по ssh без подтверждения каждого шага.
  • Прогон расходует токены Claude и занимает видеокарту.
Все причины и проверки

openbmb/forgekernel

Установка

Работает в

В терминале, через SkillFoxx CLI

npx skillfoxx add cli/forgekernel

Определит агенты на машине, покажет уровень доступа и зафиксирует версию.

Другие способы установки

Доступ и проверки

Как мы проверяем

Широкий доступИнструмент может менять данные, выполнять команды или работать с секретами.

  • Агент самостоятельно пишет и много часов запускает код на вашей видеокарте или по ssh без подтверждения каждого шага.
  • Прогон расходует токены Claude и занимает видеокарту.

Это чужой код. Посмотрите файлы в репозитории перед установкой.

Что делает

ForgeKernel берет эталонное ядро, запускает его, замеряет и профилирует, затем строит план этапов оптимизации. Claude Code по этому плану реализует решения по одному, а один GPU-запуск проверяет корректность, время и профиль. Ошибочный вариант откатывается. Каждое принятое ядро становится эталоном для следующего прохода, поэтому оптимизация идет дальше лучшей известной реализации. Находки, подтвержденные аппаратной пробой, копятся в базе знаний для других операторов. Задачи описывают в `task.toml`, прогон запускает `forge run`.

Для кого. Инженерам GPU-ядер на Hopper и Blackwell, которые хотят передать агенту многочасовую настройку оператора под замер.

Подходит, если

  • Слияние девяти запусков RMSNorm в одно ядро CuTe DSL еще не сделано.
  • Свое ядро медленнее библиотечного, и неизвестно, где потери.
  • Прогон прервался, и его надо продолжить с того же места.
  • Хочется получить отчет и лучшее ядро с профилем по итогам ночи.

Не подходит, если

  • Видеокарта не Hopper и не Blackwell: поддерживаются `sm90` и `sm100`.
  • Нет доступа к Claude Code через вход или `ANTHROPIC_API_KEY`.
  • Нужна правка ядра за несколько минут, а прогон занимает часы.

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

Создай задачу в `tasks/rmsnorm`, проверь ее через `forge new --dry-run` и запусти прогон в tmux. Потом покажи прогресс и итоговый отчет.

Ограничения

Нужны Linux или macOS, Python 3.11 и новее, Git и Claude Code, а видеокарта Hopper (`sm90`) или Blackwell (`sm100`) на этой машине или по ssh. Для задач CuTe DSL рекомендуют `nvidia-cutlass-dsl`, для профиля Nsight Compute: без него прогон идет только по времени. Один прогон занимает часы и расходует токены Claude. В README цифры ускорения относятся к публикации авторов; на вашем железе результат другой.

Как отключить. Остановите `forge run` и удалите виртуальное окружение и каталог `.artifacts` проекта. Если задавали удаленный GPU, уберите файл `local.env` и очистите рабочий каталог на сервере.

Частые вопросы

Как понять, что прогон идет?

Выполните `forge status .artifacts/rmsnorm_<id>`: команда покажет прогресс. Если прогон прервался, `forge run` с тем же путем продолжит его.

Где результат?

Отчет лежит в `.artifacts/rmsnorm_<id>/report.md`, лучшее ядро в `best/kernel.py`. В примере README на B200 вышло 5,2 раза быстрее эталона PyTorch примерно за полтора часа.

Можно ли использовать удаленный GPU?

Да, в `local.env` задайте `FK_DELIVERY=ssh`, `REMOTE_HOST` и `REMOTE_WORKDIR`. Алиас ssh должен входить без запроса пароля.

Какой код получается на выходе?

Код эталона остается за пределами ядра. План хранит решения и контракты, код лежит только в базе знаний `hw_kb`.

Какие видеокарты нужны?

NVIDIA Hopper (`sm90`) или Blackwell (`sm100`). В примере `task.toml` для Hopper задайте `arch = "sm90"`.

Обсуждение

Вопросов и отзывов пока нет.

Навык Claude Code для оптимизации GPU-ядер: агент профилирует, правит и замеряет ядро по кругу, пока ускорение не прекратится.

СкиллРасширенный доступ383Звезды репозитория

Фреймворк Alibaba для оптимизации GPU-ядер кодирующими агентами: Supervisor проверяет корректность и скорость перед принятием

ВоркфлоуРасширенный доступ212Звезды репозитория

CLI и плагин Claude Code: ведет задачи GitHub через проверку, план, код, ревью и CI до слияния, несколько параллельно

ПлагинШирокий доступ119Звезды репозитория

Скиллы для написания, отладки, замера и оптимизации программ TileLang и CUDA по данным исходников, лоуэринга и профайлеров

СкиллРасширенный доступ149Звезды репозитория

Хранит разборы оптимизаций GPU-ядер для Blackwell и Hopper, чтобы агент отвечал по источникам, а не по памяти модели.

СкиллМинимальный доступ545Звезды репозитория

LoopForge Local ставит в Claude Code, Codex и Cursor ролевой процесс разработки: анализ, дизайн, код, ревью и тесты со сводкой.

ВоркфлоуРасширенный доступ613Звезды репозитория

Коротко о README

Проект описывает самоусиливающийся цикл: эталон, план этапов, реализация агентом, проверка на GPU и повышение лучшего ядра до нового эталона. Выпущены готовые ядра, статья и три задачи.

Бейдж для README

Сообщить о проблеме