Тест производительности куба

Введение

Тестирование вычислительной производительности чипа Ascend. Программа операторов может работать на всех платформах, но статистические показатели применимы только к Ascend310. Код предназначен исключительно для демонстрации измерения производительности вычислений.

Параметры: left_matrix[8, 128, 16] * right_matrix[8, 128, 16].

В этом репозитории содержатся файлы проекта оператора и код для тестирования модели одного оператора ACL. В итоге можно получить результат теста производительности вычислений XXXFLOPS, который показывает, насколько близко полученное значение к номинальному.

Инструкция по установке

1. git clone https://gitee.com/flappy-cat/test_cube_perf.git.
2. ./build.sh — компилирует операторы и создаёт пакет запуска в build_out.
3. ./build_out/custom_opp_centos_x86_64.run — устанавливает пакет запуска пользовательских операторов.
4. Конвертация модели одного оператора: сначала создайте файл JSON на основе информации о IR оператора. Файл test_cube_perf.json в каталоге test_acl_singleop является файлом JSON для одного оператора.
5. Перейдите в этот каталог и выполните команду ./atc_singpleop_cmd.sh, чтобы создать модель одного оператора 0_TestCubePerf_1_2_8_128_16_1_2_8_128_16_0_2_8_128_16.om.
6. Перейдите в каталог msam и выполните ./build.sh. Это создаст основной исполняемый файл в предыдущем каталоге.
7. Выполните ./go_test_perf.sh в каталоге test_acl_singleop. Показатели производительности будут выведены в виде информации.

Прочее

Обратите внимание, что необходимо найти путь к Ascend/atc/include в вашей среде и установить его как ASCEND_TENSOR_COMPLIER_INCLUDE. Пример: [export ASCEND_TENSOR_COMPLIER_INCLUDE=/usr/local/Ascend/atc/include].