Что такое AKG?

AKG (Auto Kernel Generator) — это оптимизатор для операторов в сетях глубокого обучения. Он позволяет автоматически объединять операции со специфическими шаблонами. AKG работает с MindSpore-GraphKernel, чтобы улучшить производительность сетей, работающих на различных аппаратных платформах.

Поддержка аппаратных платформ

В настоящее время поддерживаются Ascend910, NVIDIA V100/A100 и CPU. В списке есть и другие платформы.

Сборка

• Сборка с помощью MindSpore: см. MindSpore README.md для получения подробной информации.
• Автономная сборка: мы рекомендуем собирать и запускать AKG вместе с MindSpore. Также мы предоставляем способ запуска кейса в автономном режиме для удобства. Для получения дополнительной информации о зависимостях компиляции обратитесь к MindSpore Installation.
  • Сборка на Ascend910: перед клонированием исходных кодов необходимо установить git-lfs.

    git clone https://gitee.com/mindspore/akg.git
    cd akg
    bash build.sh -e ascend -j8
  • Сборка на GPU:

    git clone https://gitee.com/mindspore/akg.git
    cd akg
    bash build.sh -e gpu -j8
  • Сборка на CPU:

    git clone https://gitee.com/mindspore/akg.git
    cd akg
    bash build.sh -e cpu -j8

Запуск автономного режима

1. Настройка среды:
   • Ascend910:

     cd tests
     source ./test_env.sh
   • NVIDIA V100/A100:

     cd tests
     source ./test_env.sh gpu
   • CPU:

     cd tests
     source ./test_env.sh cpu
2. Запуск теста:
   • Используйте скрипт:

     cd tests/st
     python run.py -e gpu -o add -l level0  # запустить оператор добавления на GPU
     Подробные инструкции см.: python run.py -h.
   • Используйте конкретный случай:
     • Ascend910:

       cd tests/st/ops/
       pytest -s test_abs.py -m "level0 and platform_x86_ascend_training" # запустить тестовые случаи уровня 0 на Ascend
     • NVIDIA V100/A100:

       cd tests/st/ops/
       pytest -s test_abs.py -m "level0 and platform_x86_gpu_training" # запустить тестовые случаи уровня 0 на GPU
     • CPU:

       cd tests/st/ops/
       pytest -s test_abs.py -m "level0 and platform_x86_cpu" # запустить тестовые случаи уровня 0 на CPU

Использование AKG для создания высокопроизводительных ядер

См. Wiki.

Внесение вклада

Мы приветствуем вклад. См. MindSpore Contributor Wiki для более подробной информации.

Примечания к выпуску

Примечания к выпуску см. в RELEASE.md.

Лицензия

Apache License 2.0.