Производительность TA-Lib в Python: сначала измерьте нативные вычисления

Если Python-бэктест проводит основное время внутри TA-Lib, ускоренная нативная библиотека может помочь без изменения Python-интерфейса. Если время уходит на загрузку данных, Python-циклы или симуляцию заявок, ускорение одного индикатора может мало повлиять на весь запуск. Начните с измерения своей нагрузки.

Отделите индикатор от всего бэктеста

Снимите профиль одного характерного запуска. Запишите индикатор, длину входа, сетку параметров и количество вызовов. Отдельно измерьте Python-программу целиком, включая подготовку данных и накладные расходы обёртки. Ускорение вычислительного ядра и всего запуска отвечают на разные вопросы. Например, если вычисление занимает половину времени, его двукратное ускорение сократит весь запуск на четверть при неизменности остальной работы.

Публичный пакет TAKT для TA-Lib сравнивает три нативные сборки: официальный релиз TA-Lib 0.8.1, пересборку исходников с настройками компилятора под CPU и LTO, и библиотеку TAKT. C ABI и SONAME сохранены; C-программы и существующая Python-обёртка используют прежний интерфейс. Для опубликованного выигрыша выбрана более быстрая из двух базовых сборок.

Что именно показывают опубликованные числа

На AMD Threadripper PRO 5975WX перебор 99 настроек EMA на миллионе сгенерированных баров занял 200,1 мс у более быстрой базовой сборки и 68,14 мс у TAKT: 2,94× по времени всего C-драйвера. Для CCI получены 5,292 с и 824,6 мс: 6,42×. Это конкретные нагрузки C-драйвера с генерацией данных, измеренные в 11 чередующихся раундах. Это не замеры Python-стратегии посетителя. ATR, MACD и BBANDS дали меньший выигрыш; другие функции TA-Lib выполняют оригинальный код. Исходные измерения и отдельный эксперимент с Python-обёрткой доступны из README пакета.

Запустите небольшую локальную проверку

EMA-демо без регистрации требует Linux x86-64, glibc, AVX2 и FMA, Python 3 и curl. Перед запуском можно прочитать драйвер:

curl -fLsS https://taktcycles.com/demo/talib.py -o talib.py
python3 talib.py --no-counters

Драйвер скачивает опубликованные бинарники, сверяет SHA-256 и выполняет девять чередующихся раундов на одном CPU. Сравниваются код завершения и хеши stdout. Файлы и измерения остаются на вашей машине: тест не передаёт TAKT результаты или ваш код. Если аппаратные счётчики доступны, можно убрать --no-counters. Из двух сравнений с базовыми сборками используйте меньший выигрыш.

Проверяйте корректность на нужном уровне

Совпадение хешей stdout подтверждает равенство того, что печатает небольшой драйвер. Оно не доказывает эквивалентность для всех входов или корректность всего бэктеста. Полный пакет описывает 4 585 536 сверок C-вызовов: массивы, индексы вывода и граничные случаи. Для клиентской нагрузки всё равно нужны собственные требования к выводу, характерные данные и приёмочные проверки.

Когда имеет смысл индивидуальная оценка

Подходящая отправная точка — повторяемое вычисление на CPU со стабильным интерфейсом. Опишите язык, ОС и CPU, длительность и частоту запуска, а также результаты, которые должны остаться идентичными. Для этого первого обсуждения исходники не нужны. Работа с C/C++/Python начинается с отдельного инженерного заказа; для Rust-крейтов предусмотрен самостоятельный путь через кабинет.

Описать одну нагрузку. TAKT — коммерческий сервис ABX DEVELOPMENT LLP. Статья подготовлена с AI-ассистентом по нашим опубликованным измерениям; демо запускалось на Linux. Приведённые результаты не обещают такой же выигрыш для другой задачи или платформы.